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()