31 lines
1.5 KiB
Python
31 lines
1.5 KiB
Python
from main import app, Config
|
|
from .data import data
|
|
from .modules import db
|
|
from .tools.data import save
|
|
from .tools.logs import write
|
|
|
|
from os import mkdir, path, system
|
|
from cryptography.fernet import Fernet
|
|
|
|
from sqlalchemy_utils import database_exists, create_database
|
|
|
|
def install_scripts():
|
|
if not path.isdir(f'./{data}'): mkdir(f'./{data}')
|
|
if not path.isfile(f'./{data}/.gitignore'):
|
|
with open(f'./{data}/.gitignore', 'a+') as file: file.write(f'*')
|
|
if not path.isfile(f'./{data}/config.json'): save({}, 'config.json')
|
|
if not path.isdir(f'./{data}/logs'): mkdir(f'./{data}/logs')
|
|
if not path.isfile(f'./{data}/logs/users.log'): write('users.log', 'Log file created.')
|
|
if not path.isfile(f'./{data}/logs/system.log'): write('system.log', 'Log file created.')
|
|
if not path.isfile(f'./{data}/logs/tests.log'): write('tests.log', 'Log file created.')
|
|
if not database_exists(Config.SQLALCHEMY_DATABASE_URI):
|
|
create_database(Config.SQLALCHEMY_DATABASE_URI)
|
|
write('system.log', 'No database found. Creating a new database.')
|
|
from .models import Entry, Test, User
|
|
with app.app_context(): db.create_all()
|
|
write('system.log', 'Creating database schema.')
|
|
if not path.isfile(f'./{data}/.encryption.key'):
|
|
write('system.log', 'No encryption key found. Generating new encryption key.')
|
|
with open(f'./{data}/.encryption.key', 'wb') as key_file:
|
|
key = Fernet.generate_key()
|
|
key_file.write(key) |