mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-28 15:17:53 +01:00
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
import sys
|
|
import uuid
|
|
|
|
from termcolor import colored
|
|
|
|
from database.database import get_app
|
|
|
|
|
|
def get_arguments():
|
|
# The first element in sys.argv is the name of the script itself.
|
|
# Any additional elements are the arguments passed from the command line.
|
|
args = sys.argv[1:]
|
|
|
|
# Create an empty dictionary to store the key-value pairs.
|
|
arguments = {}
|
|
|
|
# Loop through the arguments and parse them as key-value pairs.
|
|
for arg in args:
|
|
if '=' in arg:
|
|
key, value = arg.split('=', 1)
|
|
arguments[key] = value
|
|
else:
|
|
arguments[arg] = True
|
|
|
|
if 'app_id' in arguments:
|
|
try:
|
|
app = get_app(arguments['app_id'])
|
|
arguments['user_id'] = str(app.user.id)
|
|
arguments['app_type'] = app.app_type
|
|
arguments['name'] = app.name
|
|
# Add any other fields from the App model you wish to include
|
|
except ValueError as e:
|
|
print(e)
|
|
# Handle the error as needed, possibly exiting the script
|
|
else:
|
|
arguments['app_id'] = str(uuid.uuid4())
|
|
|
|
if 'user_id' not in arguments:
|
|
arguments['user_id'] = str(uuid.uuid4())
|
|
|
|
if 'email' not in arguments:
|
|
# todo change email so its not uuid4 but make sure to fix storing of development steps where
|
|
# 1 user can have multiple apps. In that case each app should have its own development steps
|
|
arguments['email'] = str(uuid.uuid4())
|
|
|
|
if 'password' not in arguments:
|
|
arguments['password'] = 'password'
|
|
|
|
if 'step' not in arguments:
|
|
arguments['step'] = None
|
|
|
|
print(colored('\n------------------ STARTING NEW PROJECT ----------------------', 'green', attrs=['bold']))
|
|
print(f"If you wish to continue with this project in future run:")
|
|
print(colored(f'python main.py app_id={arguments["app_id"]}', 'green', attrs=['bold']))
|
|
print(colored('--------------------------------------------------------------\n', 'green', attrs=['bold']))
|
|
return arguments
|