Total Complexity | 5 |
Total Lines | 43 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | #!/usr/bin/env python |
||
2 | from flask_script import Manager |
||
3 | from api.app import create_app |
||
4 | from api.models import db, User |
||
5 | |||
6 | manager = Manager(create_app) |
||
7 | |||
8 | |||
9 | @manager.command |
||
10 | def createdb(): |
||
11 | app = create_app() |
||
12 | with app.app_context(): |
||
13 | db.drop_all() |
||
14 | db.create_all() |
||
15 | |||
16 | |||
17 | @manager.command |
||
18 | def adduser(username): |
||
19 | """Register a new user.""" |
||
20 | from getpass import getpass |
||
21 | password = getpass() |
||
22 | password2 = getpass(prompt='Confirm: ') |
||
23 | if password != password2: |
||
24 | import sys |
||
25 | sys.exit('Error: passwords do not match.') |
||
26 | db.create_all() |
||
27 | user = User(username=username, password=password) |
||
28 | db.session.add(user) |
||
29 | db.session.commit() |
||
30 | print('User {0} was registered successfully.'.format(username)) |
||
31 | |||
32 | |||
33 | @manager.command |
||
34 | def test(): |
||
35 | from subprocess import call |
||
36 | call(['nosetests', '-v', |
||
37 | '--with-coverage', '--cover-package=api', '--cover-branches', |
||
38 | '--cover-erase', '--cover-html', '--cover-xml', '--cover-html-dir=cover']) |
||
39 | |||
40 | |||
41 | if __name__ == '__main__': |
||
42 | manager.run() |
||
43 |