nopaque/app/cli.py
2021-12-07 14:18:05 +01:00

43 lines
1.1 KiB
Python

from . import db
from .models import Corpus, Role
from flask_migrate import upgrade
def register(app):
@app.cli.command()
def deploy():
''' Run deployment tasks. '''
# migrate database to latest revision
upgrade()
# create or update user roles
Role.insert_roles()
@app.cli.group()
def daemon():
''' Daemon commands. '''
pass
@daemon.command('run')
def run_daemon():
''' Run daemon '''
corpus: Corpus
for corpus in Corpus.query.filter(Corpus.num_analysis_sessions > 0):
corpus.num_analysis_sessions = 0
db.session.commit()
from app.daemon import Daemon
daemon: Daemon = Daemon()
daemon.run()
@app.cli.group()
def test():
''' Test commands. '''
pass
@test.command('run')
def run_test():
''' Run unit tests. '''
from unittest import TestLoader, TextTestRunner
from unittest.suite import TestSuite
tests: TestSuite = TestLoader().discover('tests')
TextTestRunner(verbosity=2).run(tests)