mirror of
https://github.com/th3r00t/pyShelf.git
synced 2026-04-28 01:59:35 -04:00
54 lines
1.3 KiB
Python
Executable File
54 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import sys
|
|
import json
|
|
from pathlib import Path
|
|
from django.core.management.utils import get_random_secret_key
|
|
from src.backend.lib.pyShelf import Admin
|
|
|
|
|
|
def load_config():
|
|
with open('config.json',"r") as file:
|
|
config = json.load(file)
|
|
file.close()
|
|
return config
|
|
|
|
|
|
def write_config(config):
|
|
with open('config.json',"w") as file:
|
|
json.dump(config, file)
|
|
file.close()
|
|
|
|
|
|
def set_secret(config=load_config()):
|
|
if config["SECRET"] == "":
|
|
config["SECRET"] = get_random_secret_key()
|
|
print(config["SECRET"])
|
|
else:
|
|
print("Secret already set, skipping.")
|
|
|
|
def set_book_directory(config=load_config(), *args):
|
|
if config["BOOKPATH"] == "":
|
|
try: config["BOOKPATH"] = args[0]
|
|
except IndexError: config["BOOKPATH"] = input("Input Book Directory ")
|
|
|
|
def init_django_database():
|
|
cmds = [
|
|
'python3 manage.py makemigrations',
|
|
'python3 manage.py makemigrations interface',
|
|
'python3 manage.py migrate',
|
|
'python3 manage.py migrate interface',
|
|
]
|
|
os.chdir("src")
|
|
for cmd in cmds:
|
|
os.system(cmd)
|
|
os.chdir("../")
|
|
|
|
config = load_config()
|
|
|
|
set_secret(config)
|
|
set_book_directory(config)
|
|
write_config(config)
|
|
init_django_database()
|
|
Admin(Path.cwd()).createsuperuser()
|