OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-01-20 01:01:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

BaseModel.id is a UUIDField, create UUID from username

Browse Source
This commit is contained in:
Nicholas Albion
2023-09-08 15:14:49 +10:00
parent ee77f1ffac
commit 891d153a2b
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pilot/utils/arguments.py
View File

@@ -1,3 +1,5 @@
import getpass
import hashlib
import sys
import uuid
@@ -23,7 +25,8 @@ def get_arguments():
arguments[arg] = True
if 'user_id' not in arguments:
arguments['user_id'] = str(uuid.uuid4())
# arguments['user_id'] = str(uuid.uuid4())
arguments['user_id'] = username_to_uuid(getpass.getuser())
app = None
if 'workspace' in arguments:
@@ -36,7 +39,7 @@ def get_arguments():
if 'app_id' in arguments:
try:
if app is None:
app = get_app(arguments['app_id'])
app = get_app(arguments['app_id'])
# arguments['user_id'] = str(app.user.id)
arguments['app_type'] = app.app_type
@@ -69,3 +72,10 @@ def get_arguments():
arguments['step'] = None
return arguments
# TODO can we make BaseModel.id a CharField with default=uuid4?
def username_to_uuid(username):
sha1 = hashlib.sha1(username.encode()).hexdigest()
uuid_str = "{}-{}-{}-{}-{}".format(sha1[:8], sha1[8:12], sha1[12:16], sha1[16:20], sha1[20:32])
return str(uuid.UUID(uuid_str))