manage.py

# -*- coding: utf-8 -*-

__author__ = "苦叶子"

"""

公众号: 开源优测

Email: lymking@foxmail.com

"""

import os
from app import create_app, db
from app.utils.trigger import Trigger
from app.models import User, Role
from flask_script import Manager, Shell
from flask_migrate import Migrate, MigrateCommand

os.environ["PATH"] = os.environ["PATH"] + ";" + os.getcwd() + "/bin"

app = create_app(os.environ.get('AUTOBEAT_CONFIG') or 'default')
app.config["TRIGGER"] = Trigger(app)
app.config["TRIGGER"].setup()
app.config["TRIGGER"].load_job_list()
#trigger = Trigger(app)
#trigger.setup()
#trigger.load_job_list()

manager = Manager(app)
migrate = Migrate(app, db)


def make_shell_context():
    return dict(app=app, db=db, User=User, Role=Role)


manager.add_command('shell', Shell(make_context=make_shell_context))
manager.add_command('db', MigrateCommand)


@manager.command
def deploy():
    from flask_migrate import upgrade
    from app.models import Role, User

    upgrade()

    Role.insert_roles()

    User.insert_admin()

    User.insert_auto()

    User.insert_user("lymking@foxmail.com", "lyy51", "Lym")


@manager.command
def keyword():
    import subprocess
    libs = ('BuiltIn', 'Collections', 'DateTime',
            'Dialogs', 'OperatingSystem', 'Process',
            'Screenshot', 'String', 'Telnet', 'XML',
            'RequestsLibrary', 'AppiumLibrary', 'SeleniumLibrary')
    print("生成关键字...")
    for lib in libs:
        args = 'python -m robot.libdoc %s doc/%s.xml' % (lib, lib)
        subprocess.call(args, shell=True)


if __name__ == '__main__':

    app.config["TRIGGER"].start()

    manager.run()