ska-referee-test/ref-test/app/install.py

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.isdir(f'./{data}/questions'): mkdir(f'./{data}/questions')
    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)