-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
67 lines (58 loc) · 1.96 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
"""Boot Script of Client App"""
import os
from dotenv import load_dotenv
dotenv_path = os.path.join(os.path.dirname(__file__), ".env")
if os.path.exists(dotenv_path):
load_dotenv(dotenv_path)
COV = None
if os.environ.get("FLASK_COVERAGE"):
import coverage
COV = coverage.coverage(branch=True, include="client/*")
COV.start()
import sys
import click
from flask_migrate import Migrate, upgrade
from client import create_app, db
app = create_app(os.getenv("FLASK_ENV") or "default")
migrate = Migrate()
with app.app_context():
migrate.init_app(app, db, render_as_batch=True)
@app.cli.command()
@click.option("--coverage/--no-coverage", default=False,
help="Run tests under code coverage.")
def test(coverage):
"""Run the unit tests."""
if coverage and not os.environ.get("FLASK_COVERAGE"):
import subprocess
os.environ["FLASK_COVERAGE"] = "1"
sys.exit(subprocess.call(sys.argv))
import unittest
tests = unittest.TestLoader().discover("tests")
results = unittest.TextTestRunner(verbosity=2).run(tests)
if COV:
COV.stop()
COV.save()
print("Coverage Summary:")
COV.report()
basedir = os.path.abspath(os.path.dirname(__file__))
covdir = os.path.join(basedir, "htmlcov")
COV.html_report(directory=covdir)
print("HTML version: file://%s/index.html" % covdir)
sys.exit(not results.wasSuccessful())
@app.cli.command()
def deploy():
"""Run deployment tasks."""
# migrate database to latest revision
upgrade()
@app.cli.command()
@click.option("-u", "--username", prompt="New Username")
@click.option("-p", "--password", prompt=True, hide_input=True,
confirmation_prompt=True)
def register(username, password):
"""Register an Admin which can access website"""
from client.models import Users
new_user = Users(username, password)
db.session.add(new_user)
db.session.commit()
if __name__ == "__main__":
app.run()