diff --git a/dislocker_auth.py b/dislocker_auth.py index 0c08fcd..0f1472e 100644 --- a/dislocker_auth.py +++ b/dislocker_auth.py @@ -66,34 +66,41 @@ class Auth(): pc_uuid = str(kwargs["pc_uuid"]) pc_token = str(kwargs["pc_token"]) if "device_list" in kwargs: - device_list = kwargs["device_list"] + if kwargs["device_list"] == []: + device_list = None + else: + device_list = kwargs["device_list"] else: - pass - keyboard_number = "own" - mouse_number = "own" + device_list = None + + keyboard_number = 0 + mouse_number = 0 if "password_hash" in kwargs: password_hash = str(kwargs["password_hash"]) cursor.execute("SELECT * FROM pc_list WHERE pc_number = %s AND password_hash = %s AND pc_uuid = %s AND pc_token = %s", (pc_number, password_hash, pc_uuid, pc_token)) pc_info = cursor.fetchall() if pc_info: - for device in device_list: - cursor.execute("SELECT * FROM keyboard_list WHERE device_instance_path = %s", (device["device_instance_path"],)) - keyboard_record = cursor.fetchall() - if keyboard_record: - keyboard_number = int(keyboard_record[0][0]) - break - else: - pass - - for device in device_list: - cursor.execute("SELECT * FROM mouse_list WHERE device_instance_path = %s", (device["device_instance_path"],)) - mouse_record = cursor.fetchall() - if mouse_record: - mouse_number = int(mouse_record[0][0]) - break - else: - pass + if device_list == None: + pass + else: + for device in device_list: + cursor.execute("SELECT * FROM keyboard_list WHERE device_instance_path = %s", (device["device_instance_path"],)) + keyboard_record = cursor.fetchall() + if keyboard_record: + keyboard_number = int(keyboard_record[0][0]) + break + else: + pass + + for device in device_list: + cursor.execute("SELECT * FROM mouse_list WHERE device_instance_path = %s", (device["device_instance_path"],)) + mouse_record = cursor.fetchall() + if mouse_record: + mouse_number = int(mouse_record[0][0]) + break + else: + pass return {"result": 0, "about": "ok", "output_dict": {"keyboard_number": keyboard_number, "mouse_number": mouse_number}} else: