atoken = token_generator.generate(20)
email = email_generator.generate()
password = password_generator.generate()
key = email
cipher = ZAESIVEmbeddedCipher(ZCipher.AS_HEX)
hex_iv_password = cipher.encrypt(password, key)
print("Auth_token {}".format(atoken))
print("Email {}".format(email))
print("Password {}".format(password))
try:
insert = "INSERT INTO test.User (auth_token, email, password) VALUES(%s, %s, %s)"
values = (atoken, email, hex_iv_password)
db.execute(insert, values)
except:
traceback.print_exc()
db.rollback()
db.commit()
except Exception as ex:
traceback.print_exc()
db.rollback()
finally:
db.connection.close()
auth_token varchar(40) default NULL,
email varchar(255) unique default NULL,
password varchar(512) default NULL, #AES ENCRUPTED Password
validated bool default False,
last_modified datetime default NOW()
);
"""
if main(__name__):
# Commandline parameters user password dabase
db = ZMySQLDB(argv=sys.argv)
try:
#Create test database.
create_db = "CREATE DATABASE IF NOT EXISTS test"
db.execute(create_db)
#Create test.User table
create_tbl = "CREATE TABLE IF NOT EXISTS test.User ( "\
+ " id bigint unsigned not null primary key auto_increment, "\
+ " auth_token varchar(40) default NULL, "\
+ " email varchar(255) unique default NULL, "\
+ " password varchar(512) default NULL, "\
+ " last_modified datetime default NOW() "\
+ " ); "
db.execute(create_tbl)
db.commit()
print("Created test.user Table")
from SOL4Py.crypto.ZAESIVEmbeddedCipher import *
from SOL4Py.generator.ZTokenGenerator import *
from SOL4Py.generator.ZEmailAddressGenerator import *
from SOL4Py.generator.ZPasswordGenerator import *
if main(__name__):
# user password database
# foo password test
db = ZMySQLDB(argv=sys.argv)
try:
delete = "DELETE FROM test.User WHERE email LIKE %s"
values = ('%.biz', )
db.execute(delete, values)
db.commit()
select = "SELECT * FROM test.User WHERE email LIKE %s"
values = ('%.biz', )
rows = db.execute_fetchall(select, values)
if rows != None:
print("Rows len {}".format(len(rows)))
else:
print("Selected None")
except Exception as ex:
traceback.print_exc()
db.rollback()
print("Hit record: id {} email {}".format(id, email))
token_generator = ZTokenGenerator()
password_generator = ZPasswordGenerator()
new_atoken = token_generator.generate(20)
new_password = password_generator.generate()
key = email
cipher = ZAESIVEmbeddedCipher(ZCipher.AS_HEX)
new_hex_iv_password = cipher.encrypt(new_password, key)
print("New Auth_token {}".format(new_atoken))
print("New Password {}".format(new_password))
try:
update = "UPDATE test.User SET auth_token=%s, password =%s WHERE id =%s"
values = (new_atoken, new_hex_iv_password, id)
db.execute(update, values)
except:
traceback.print_exc()
db.commit()
except Exception as ex:
traceback.print_exc()
db.rollback()
finally:
db.connection.close()