def query_user_role(user_id=1):
u = User.query.get(user_id)
for u_r in u.roles:
print(u_r.name)
def add_role_privilege():
admin = Role.query.filter(Role.name.__eq__('系统管理员')).one()
print(admin)
# print(Privilege.query.all())
for p in Privilege.query.all():
admin.privileges.append(p)
db.session.commit()
def query_role_privilege(role_name='系统管理员'):
for r in Role.query.filter(Role.name.__eq__(role_name)).one().privileges:
print(r.name)
if __name__ == '__main__':
app.app_context().push()
db.init_app(app)
add_user()
# add_role()
# add_user_role()
# query_user_role()
# add_privilege()
# add_role_privilege()
# query_role_privilege()
curconfig = 'config.QAConfig'
elif args.env == 'test':
curconfig = 'config.TestConfig'
elif args.env == 'dev':
curconfig = 'config.DevConfig'
elif args.env == 'env':
curconfig = os.getenv('FLASK_ENV_OBJ')
else:
raise Exception('Invalid Config')
if curconfig == None:
raise Exception('Config cannot be null')
return curconfig
# only use for migrate
if __name__ == '__main__':
curconfig = get_config()
import mainapp
mainapp.configure(curconfig)
from mainapp import app
from mainapp import db
migrate = Migrate(app, db)
with app.app_context():
flask.ext.migrate.upgrade()