def create_superuser(email, password): admin = user_datastore.find_or_create_role('admin') user = User.query.filter_by(email=email).first() if not user: user = register_user(email=email, password=password) user.confirmed_at = datetime.now() db.session.add(user) db.session.commit() user_datastore.add_role_to_user(user, admin) db.session.commit()
def create_superuser(email, password): admin = user_datastore.find_or_create_role('admin') user = User.query.filter_by(email=email).first() if not user: user = register_user(email=email, password=password) user.confirmed_at = datetime.now() db.session.add(user) db.session.commit() user_datastore.add_role_to_user(user, admin) db.session.commit()
def create_admin(): admin_role = user_datastore.find_or_create_role( name='admin', description='Full Administrator Privliges' ) admin_user = user_datastore.find_user(email='*****@*****.**') if admin_user is None: admin_user = user_datastore.create_user( email='*****@*****.**', password='******' ) user_datastore.add_role_to_user(admin_user, admin_role) db.session.commit()
def init_db(): # try: # roles = ["User", 'Shop', 'Admin'] # for r in roles: # role = Role.query.filter_by(name = r).first() # if role is None: # user_datastore.create_role(name = r) # db.session.commit() # except InterruptedError: # db.session.rollback() user_datastore.find_or_create_role(name="User", description='Normal user') user_datastore.find_or_create_role(name="Shop", description='shop') user_datastore.find_or_create_role(name="Admin", description='admin') user_datastore.find_or_create_role(name="Super_admin", description='super admin') db.session.commit() if not user_datastore.get_user("*****@*****.**"): user_datastore.create_user(email="*****@*****.**", password="******") db.session.commit() if not user_datastore.get_user("*****@*****.**"): user_datastore.create_user(email="*****@*****.**", password="******") if not user_datastore.get_user('*****@*****.**'): user_datastore.create_user(email="*****@*****.**", password='******') # db.session.commit() user_datastore.add_role_to_user('*****@*****.**', 'User') user_datastore.add_role_to_user('*****@*****.**', 'Shop') user_datastore.add_role_to_user('*****@*****.**', 'Admin') # db.session.commit() try: category = { u"生物医药类": [u"生物类药物", u"疫苗", u"诊断试剂", u"医用材料", u"生物治疗"], u"生物能源类": [u"生物液体燃料", u"生物质气化固体燃料"], u"生物化工类": [u"生物基燃料", u"酶制剂", u"生物反应器"], u"生物农业类": [u"生物育种", u"生物农药", u"生物农制品", u"食品及化妆品添加剂"], u"生物环境类": [u"生物环保制品", u"污染检测", u"生物修复"] } for key, v in category.items(): if Product_Category.query.filter_by( category_name=key).first() == None: category_patent = Product_Category(category_name=key, parent_id=0) db.session.add(category_patent) db.session.commit() for item in v: if Product_Category.query.filter_by( category_name=item).first() == None: category_child = Product_Category( category_name=item, parent_id=category_patent.id) db.session.add(category_child) db.session.commit() db.session.commit() except: db.session.rollback() current_app.logger.error(traceback.format_exc()) abort(500, traceback.format_exc())