mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-25 20:13:57 +01:00
revert setup.py
This commit is contained in:
parent
3ef5746b1a
commit
9c41ac77be
56
setup.py
56
setup.py
|
@ -1,8 +1,4 @@
|
||||||
import setuptools
|
import setuptools
|
||||||
import os
|
|
||||||
import sqlite3
|
|
||||||
import argparse
|
|
||||||
import json
|
|
||||||
|
|
||||||
with open('README.md', 'r') as fh:
|
with open('README.md', 'r') as fh:
|
||||||
long_description = fh.read()
|
long_description = fh.read()
|
||||||
|
@ -22,55 +18,3 @@ setuptools.setup(
|
||||||
'Operating System :: OS Independent',
|
'Operating System :: OS Independent',
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
with open('config.json', 'r') as file:
|
|
||||||
config = json.load(file)
|
|
||||||
|
|
||||||
DB_PATH = os.getenv('DB_PATH') or config.get('DB_PATH')
|
|
||||||
|
|
||||||
def create_database():
|
|
||||||
conn = sqlite3.connect(DB_PATH)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
|
|
||||||
cursor.execute('''CREATE TABLE IF NOT EXISTS keys (
|
|
||||||
key_value TEXT NOT NULL
|
|
||||||
);''')
|
|
||||||
conn.commit()
|
|
||||||
print("Database created!")
|
|
||||||
|
|
||||||
def add_key(key):
|
|
||||||
conn = sqlite3.connect(DB_PATH)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
|
|
||||||
cursor.execute("INSERT INTO keys (key_value) VALUES (?)", (key,))
|
|
||||||
conn.commit()
|
|
||||||
print(f"Added key: {key}")
|
|
||||||
|
|
||||||
def remove_key(key_id):
|
|
||||||
conn = sqlite3.connect(DB_PATH)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
|
|
||||||
cursor.execute("DELETE FROM keys WHERE key_value=?", (key_id,))
|
|
||||||
conn.commit()
|
|
||||||
print(f"Removed key with key_id: {key_id}")
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
parser = argparse.ArgumentParser(description="NovaGPT Key Setup Utility")
|
|
||||||
|
|
||||||
parser.add_argument('--create-db', action="store_true", help="Create the keys database")
|
|
||||||
parser.add_argument('--add-key', type=str, help="Add a key to the keys database")
|
|
||||||
parser.add_argument('--remove-key', type=int, help="Remove a key from the keys database by key_id")
|
|
||||||
|
|
||||||
args = parser.parse_args()
|
|
||||||
|
|
||||||
if args.create_db:
|
|
||||||
create_database()
|
|
||||||
elif args.add_key:
|
|
||||||
add_key(args.add_key)
|
|
||||||
elif args.remove_key:
|
|
||||||
remove_key(args.remove_key)
|
|
||||||
else:
|
|
||||||
parser.print_help()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue