mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-31 00:20:03 +01:00
77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
import os
|
|
import re
|
|
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:
|
|
arguments['email'] = get_email()
|
|
|
|
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
|
|
|
|
|
|
def get_email():
|
|
# Attempt to get email from .gitconfig
|
|
gitconfig_path = os.path.expanduser('~/.gitconfig')
|
|
|
|
if os.path.exists(gitconfig_path):
|
|
with open(gitconfig_path, 'r') as file:
|
|
content = file.read()
|
|
|
|
# Use regex to search for email address
|
|
email_match = re.search(r'email\s*=\s*([\w\.-]+@[\w\.-]+)', content)
|
|
|
|
if email_match:
|
|
return email_match.group(1)
|
|
|
|
# If not found, return a UUID
|
|
# 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
|
|
return str(uuid.uuid4())
|