-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmanage.py
More file actions
44 lines (32 loc) · 1.07 KB
/
manage.py
File metadata and controls
44 lines (32 loc) · 1.07 KB
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
# import click
# from flask.cli import FlaskGroup
# from flask_migrate import Migrate, MigrateCommand
# from flask_script import Manager
# import os
# from app import app, db
# app.config.from_object(os.environ['APP_SETTINGS'])
# migrate = Migrate(app, db)
# manager = Manager(app)
# manager.add_command('db', MigrateCommand)
# if __name__ == '__main__':
# manager.run()
import os
import click
from flask.cli import FlaskGroup
from flask_migrate import Migrate
import datetime
from app import app, db
from models import User
app.config.from_object(os.environ['APP_SETTINGS'])
migrate = Migrate(app, db)
# Create the FlaskGroup instance
cli = FlaskGroup(app)
@app.cli.command("create_admin")
@click.argument("name", required=False, default="admin")
def create_admin(name):
"""Creates the admin user."""
db.session.add(User(name=name, username="ad@min.com", email="ad@min.com", password="admin", admin=True, is_confirmed=True,
confirmed_on=datetime.datetime.now())) # admin doesn't need email confirmation
db.session.commit()
if __name__ == '__main__':
cli()