def insert_init_values():
data_base_manager = DataBaseManager(DB_NAME)
fields = [(1, 'elle', hashlib.sha1('EL').hexdigest(), 1, 101), (2, 'David', hashlib.sha1('2511').hexdigest(), 1, 102),
(3, 'dana123', hashlib.sha1('12345').hexdigest(), 1, 103), (4, 'dana123', hashlib.sha1('12345').hexdigest(), 2, 101),
(5, 'elle', hashlib.sha1('EL').hexdigest(), 2, 102)]
data_base_manager.insert("Users", fields)
# print db tables
data_base_manager.print_table("Users")
data_base_manager.print_table("SPs")
data_base_manager.close_connection()
def add_user(id, username, passw, sp_id, sp_user_id):
"""
Adds A row in UserProfiles table for a new user with this info:
:param id: user ID
:param username: chosen username
:param passw: chosen password
:param sp_id: ID of SP server to which the user is registrating
:param sp_user_id: ID o this user in SP server
:return: None
"""
fields = [(id, username, passw, sp_id, sp_user_id)]
data_base_manager = DataBaseManager(DB_NAME)
data_base_manager.insert("Users", fields)
data_base_manager.print_table("Users")
data_base_manager.close_connection()
def sps_comm():
"""
Server side of socket comm.
Inserts newly connected SPs to DB.
:return: None
"""
public_key, new_key = generate_RSA_key()
while True:
sp_sock, sp_addr = server_sock.accept()
sp_thread = ThreadWithReturnValue(target=register_sp, args=(sp_sock, public_key, new_key))
sp_thread.start()
fields = sp_thread.join()
data_base_manager = DataBaseManager("MainDB.db")
data_base_manager.insert("SPs", fields)
data_base_manager.print_table("SPs")
data_base_manager.close_connection()
