suti7yk5032/Dislocker
1
0
Fork 0
Code Issues 11 Pull requests Projects 3 Releases Packages Wiki Activity Actions

UUID型をstr(VARCHAR)に変更

Browse source
This commit is contained in:
suti7yk5032 2024-08-26 00:19:53 +09:00
parent c6c602a34b
commit 25f3ae6404
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
dislocker.py
View file

@ -90,7 +90,7 @@ class DL():
find_pc_list_table = cursor.fetchall() find_pc_list_table = cursor.fetchall()
print(find_pc_list_table) print(find_pc_list_table)
if find_pc_list_table[0][0] == False: if find_pc_list_table[0][0] == False:
cursor.execute("CREATE TABLE pc_list (pc_number INTEGER NOT NULL, using_member_id INTEGER, password_hash VARCHAR(32), pc_uuid UUID, pc_token VARCHAR(36), PRIMARY KEY (pc_number), FOREIGN KEY (using_member_id) REFERENCES club_member(member_id))") cursor.execute("CREATE TABLE pc_list (pc_number INTEGER NOT NULL, using_member_id INTEGER, password_hash VARCHAR(32), pc_uuid VARCHAR(36), pc_token VARCHAR(36), PRIMARY KEY (pc_number), FOREIGN KEY (using_member_id) REFERENCES club_member(member_id))")
for i in self.pc_list: for i in self.pc_list:
print(i) print(i)
cursor.execute("INSERT INTO pc_list (pc_number) VALUES (%s)", (i,)) cursor.execute("INSERT INTO pc_list (pc_number) VALUES (%s)", (i,))

4
dislocker_auth.py
View file

@ -53,7 +53,7 @@ class Auth():
try: try:
cursor = self.db.cursor() cursor = self.db.cursor()
pc_number = int(kwargs["pc_number"]) pc_number = int(kwargs["pc_number"])
pc_uuid = uuid.UUID(kwargs["pc_uuid"]) pc_uuid = str(kwargs["pc_uuid"])
pc_token = str(kwargs["pc_token"]) pc_token = str(kwargs["pc_token"])
if "password_hash" in kwargs: if "password_hash" in kwargs:
@ -174,7 +174,7 @@ class Auth():
try: try:
cursor = self.db.cursor() cursor = self.db.cursor()
pc_number = int(kwargs["pc_number"]) pc_number = int(kwargs["pc_number"])
pc_uuid = uuid.UUID(kwargs["pc_uuid"]) pc_uuid = str(kwargs["pc_uuid"])
cursor.execute("SELECT pc_uuid FROM pc_list WHERE pc_number = %s", (pc_number,)) cursor.execute("SELECT pc_uuid FROM pc_list WHERE pc_number = %s", (pc_number,))
pc_record = cursor.fetchall() pc_record = cursor.fetchall()
pc_record_uuid = pc_record[0][0] pc_record_uuid = pc_record[0][0]