class Test_2_DB_all_users(unittest.TestCase):
def setUp(self):
self.db = DB()
self.connection = self.db.connect_db()
with self.db.db_cursor(self.connection) as curs:
for i in range(10):
self.u = User()
self.u.username = '******' + ('').join(random.choices(ALPHABET, k=3))
self.u.email = str(self.u.username) + '@email.com'
self.u.hashed_password = {'password': ('').join(random.choices(ALPHABET, k=6)), 'salt': None}
#save
save = self.u.save_to_db(curs)
self.connection.commit()
def test_load_all_users(self):
with self.db.db_cursor(self.connection) as curs:
all_users = User.load_all_users(curs)
self.assertTrue(len(all_users), 10)
self.assertIsInstance(all_users, list)
self.assertIsInstance(all_users[1], User)
self.assertIsInstance(all_users[6], User)
self.assertIsInstance(all_users[9], User)
def tearDown(self):
with self.db.db_cursor(self.connection) as curs:
all_users = User.load_all_users(curs)
for u in all_users:
u.delete(curs)
self.connection.commit()
self.connection.close()
print(self.connection)
def setUp(self):
self.db = DB()
self.connection = self.db.connect_db()
with self.db.db_cursor(self.connection) as curs:
for i in range(10):
self.u = User()
self.u.username = '******' + ('').join(random.choices(ALPHABET, k=3))
self.u.email = str(self.u.username) + '@email.com'
self.u.hashed_password = {'password': ('').join(random.choices(ALPHABET, k=6)), 'salt': None}
#save
save = self.u.save_to_db(curs)
self.connection.commit()
def setUpClass(cls):
cls.db = DB()
cls.connection = cls.db.connect_db()
cls.ids = []
cls.msg_id = []
cls.unames = []
with cls.db.db_cursor(cls.connection) as curs:
for i in range(2):
u = User()
u.username = '******' + ('').join(random.choices(ALPHABET, k=3))
u.email = str(u.username) + '@email.com'
u.hashed_password = {'password': ('').join(random.choices(ALPHABET, k=6)), 'salt': None}
u.save_to_db(curs)
cls.ids.append(u.load_user_by_username(curs, u.username).id) #user ids
cls.unames.append(u.username)
cls.connection.commit()
def setUpClass(cls):
cls.db = DB()
cls.connection = cls.db.connect_db()
cls.u_ids = []
cls.msg_id = []
cls.unames = []
with cls.db.db_cursor(cls.connection) as curs:
#create temporary users
for i in range(5):
u = User()
u.username = '******' + ('').join(random.choices(ALPHABET, k=3))
u.email = str(u.username) + '@email.com'
u.hashed_password = {'password': ('').join(random.choices(ALPHABET, k=6)), 'salt': None}
u.save_to_db(curs)
cls.u_ids.append(u.load_user_by_username(curs, u.username).id) #user ids
cls.unames.append(u.username)
cls.connection.commit()
#create temporary messages
for i in range(5):
message = Message()
message.from_id = random.choice(cls.u_ids)
message.to_id = random.choice(cls.u_ids[1:])
message.text = 'test message ' + ('').join(random.choices(ALPHABET, k=6))
# print(cls.message.id)
# self.message.date !
save = message.save_to_db(curs)
cls.connection.commit()
cls.msg_id.append(message.id) # append zmienia zmienna klasowa cls.msg_id = []
# User 1 messages
for i in range(5):
message = Message()
message.from_id = random.choice(cls.u_ids)
message.to_id = cls.u_ids[0]
message.text = 'test message ' + ('').join(random.choices(ALPHABET, k=6))
# print(cls.message.id)
# self.message.date !
save = message.save_to_db(curs)
cls.connection.commit()
cls.msg_id.append(message.id)
print(cls.msg_id)
#!/usr/bin/python3
from db_run import DB
# create Database and tables users and messages
if __name__ == '__main__':
o = DB()
o.clean()
o.run()
connection = o.connect_db()
with o.db_cursor(connection) as curs:
o.create_tables(curs)
print(curs)
print(curs)
print(connection)
connection.commit()
connection.close()
print(connection)
def setUpClass(cls):
cls.user = User()
cls.user.username = '******'
cls.user.email = '*****@*****.**'
cls.user.hashed_password = {'password': '******', 'salt': 'er445'}
cls.db = DB()
help='change password, use with -u -p -e')
parser.add_argument('-d',
'--delete',
action='store_true',
help='delete user, use wit -u -p')
parser.add_argument('-l',
'--list',
action='store_true',
help='list of users')
options = vars(parser.parse_args())
options = {k: v for k, v in options.items() if v not in (False, None)}
# db connection
cnx = None
ob = DB()
try:
cnx = ob.connect_db()
except Exception:
raise Exception(
'Incorrect connection parameters or database not created!')
with ob.db_cursor(cnx) as curs:
if 'username' in options:
if len(options) > 0 and 'password' not in options:
parser.print_help()
userX = User.load_user_by_username(curs, options['username'])
# no user
if not userX:
if 'password' in options:
if len(options) == 2: