diff --git a/dislocker_auth.py b/dislocker_auth.py index e329fef..6befeae 100644 --- a/dislocker_auth.py +++ b/dislocker_auth.py @@ -305,12 +305,12 @@ class Auth(): pc_number = int(kwargs["pc_number"]) pc_uuid = str(kwargs["pc_uuid"]) cursor.execute("SELECT pc_uuid FROM pc_list WHERE pc_number = %s", (pc_number,)) - pc_record = cursor.fetchall() - pc_record_uuid = pc_record[0][0] + pc_record_uuid_source = cursor.fetchall() + pc_record_uuid = pc_record_uuid_source[0][0] cursor.execute("SELECT pc_number FROM pc_list WHERE pc_number = %s", (pc_number,)) - pc_record = cursor.fetchall() - pc_record_number = pc_record[0][0] - if pc_record_number == None: + pc_record_number_source = cursor.fetchall() + + if pc_record_number_source == None: pc_token = self.token_generate(36) master_password = self.master_password_generate(16) master_password_hash = self.hash_genarate(master_password)