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

from .extensions import db
from .tools.data import save
from .tools.logs import write

from sqlalchemy_utils import create_database, database_exists

from cryptography.fernet import Fernet
from os import mkdir, path
from pathlib import Path

def install_app(app):
    with app.app_context():
        data = Path(app.config.get('DATA'))
        database_uri = app.config.get('SQLALCHEMY_DATABASE_URI')
        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(database_uri):       
            create_database(database_uri)
            write('system.log', 'No database found. Creating a new database.')
            from .models import Entry, Dataset, Test, User
            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)