Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

style: format code with Autopep8, Black, isort and Yapf #31

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

style: format code with Autopep8, Black, isort and Yapf #31

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
102 changes: 60 additions & 42 deletions trustauthx/cli.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,28 +1,33 @@
import argparse
from .llmai import LLMAI_Inter
import os
import subprocess
import sys, os, time
import sys
import time

from dotenv import load_dotenv

from .llmai import LLMAI_Inter


def main():
parser = argparse.ArgumentParser(prog='trustauthx')
load_dotenv(dotenv_path='./.env', override=True, verbose=True)
api_key = os.environ.get('API_KEY')
api_secret = os.environ.get('API_SECRET')
org_id = os.environ.get('ORG_ID')
parser.add_argument('command')
parser.add_argument('framework')

parser.add_argument('-k', required=not api_key, help='API key')
parser.add_argument('-s', required=not api_secret, help='API secret')
parser.add_argument('-o', required=not org_id, help='Organization ID')
parser = argparse.ArgumentParser(prog="trustauthx")
load_dotenv(dotenv_path="./.env", override=True, verbose=True)
api_key = os.environ.get("API_KEY")
api_secret = os.environ.get("API_SECRET")
org_id = os.environ.get("ORG_ID")

parser.add_argument("command")
parser.add_argument("framework")

parser.add_argument("-k", required=not api_key, help="API key")
parser.add_argument("-s", required=not api_secret, help="API secret")
parser.add_argument("-o", required=not org_id, help="Organization ID")

args = parser.parse_args()
# try:
if args.k and args.s and args.o:
if api_key or api_secret or org_id:
file_path = './.env'
file_path = "./.env"
if os.path.isfile(file_path):
os.remove(file_path)
else:
Expand All @@ -36,64 +41,77 @@
"API_KEY": args.k,
"API_SECRET": args.s,
"ORG_ID": args.o
}
with open('.env', 'w') as f:
}
with open(".env", "w") as f:
for key, value in env_vars.items():
f.write(f'{key}={value}\n')
load_dotenv(dotenv_path='./.env', override=True, verbose=True)
api_key = os.environ.get('API_KEY')
api_secret = os.environ.get('API_SECRET')
org_id = os.environ.get('ORG_ID')
f.write(f"{key}={value}\n")
load_dotenv(dotenv_path="./.env", override=True, verbose=True)
api_key = os.environ.get("API_KEY")
api_secret = os.environ.get("API_SECRET")
org_id = os.environ.get("ORG_ID")

if api_key or api_secret or org_id: pass
else: print(f"no .env found, api_key {not bool(api_key)}, api_secret {not bool(api_secret)}, org_id {not bool(org_id)}")
if api_key or api_secret or org_id:
pass
else:
print(
f"no .env found, api_key {not bool(api_key)}, api_secret {not bool(api_secret)}, org_id {not bool(org_id)}"
)

client = LLMAI_Inter(api_key, api_secret, org_id, "")
print("\ngetting auth status ...")
if not client.arb_login():raise ConnectionRefusedError("user not found, invalid credential")
print("\ngetting auth status ...")
if not client.arb_login():
raise ConnectionRefusedError("user not found, invalid credential")
print("\nauthorization success, credentials valid")
if args.command == 'neuroform':
client.framework=args.framework

if args.command == "neuroform":
client.framework = args.framework
sdk = client
print("\ngetting req. dependencies:")
list_depends = sdk.Install_dependancies()
print(list_depends)
print("\nInstalling dependencies...")

def install(packages):
if packages == "pip install fastapi['all']":
subprocess.call("pip install fastapi[all]")
return
if isinstance(packages, list):
for package in packages:subprocess.call(package, shell=True)
else:subprocess.call(packages, shell=True)
for package in packages:
subprocess.call(package, shell=True)

Check failure

Code scanning / Bandit

subprocess call with shell=True identified, security issue. Error

subprocess call with shell=True identified, security issue.
else:
subprocess.call(packages, shell=True)

Check failure

Code scanning / Bandit

subprocess call with shell=True identified, security issue. Error

subprocess call with shell=True identified, security issue.

install(list_depends)
print("\nDependencies installed.")
a = sdk.Create_App(path=os.getcwd())
print(a)
print("\nApp named --> authx.py")
print(f"\napp located at --> {os.path.join(os.getcwd(), 'authx.py')}")
print("\nApp creation Successful...")
print(f"\nyou could start the server with command trustauthx start {args.framework}")

if args.command == 'start':
client.framework=args.framework
print(
f"\nyou could start the server with command trustauthx start {args.framework}"
)

if args.command == "start":
client.framework = args.framework
sdk = client
print("\nTrying to start local server ...")
print("\nthis command might fail in case of few frameworks in such cases consider installing req. lib. and starting server manually")
print(
"\nthis command might fail in case of few frameworks in such cases consider installing req. lib. and starting server manually"
)
b = sdk.Start_server()
process = subprocess.Popen(b, shell=True)
process.wait()

if args.command == 'login':
if args.command == "login":
time.sleep(0.5)
print("\nattempt to Login TrustAuthx Build AI successful")
print("\nExecuting Rate-Limit")
time.sleep(1)
print("\nEverything Done Status 200, Ready To Start")

if args.command == 'logout':
file_path = './.env'
if args.command == "logout":
file_path = "./.env"
if os.path.isfile(file_path):
os.remove(file_path)
else:
Expand All @@ -103,9 +121,9 @@
time.sleep(1)
print("\nEverything Done Status 200, Successfully logged out")

if __name__ == '__main__':
main()

if __name__ == "__main__":
main()

# if args.command == 'fabricate':
# client.framework=args.framework
Expand All @@ -120,4 +138,4 @@
# install(list_depends)
# print("\nDependencies installed.")
# a = sdk.Create_App(out=args.out)
# print(a)
# print(a)
Loading