def user_register(): form = RegisterForm(request.form) if request.method == 'POST' and form.validate(): user = User() user.set_attrs(form.data) user.updatetime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime( (time.time()))) # 修改数据更新时间 # 判断是否已经存在该公司,如不存在,则创建该公司的公司管理员账户,并创建相应数据表 # 如果存在,则只能创建公司普通人员账户,不创建数据表 if user.authority == 'com_person': db.session.add(user) db.session.commit() else: # 公司不存在,且注册时选择了公司管理员,则允许创建用户,并创建相关数据表 company = Company() company.set_attrs(form.data) company.company_number = "%03d" % (Company.query.count() + 1) db.session.add(company) db.session.add(user) db.session.commit() # 创建报告保存文件夹,根据公司编号创建,如超级管理员为001,则文件夹名为001 os.makedirs( os.path.join(current_app.config['DOCXFILE_DIR'], company.company_number)) return render_template('user/userRegister.html', messages={'message': ['注册成功!']}) else: return render_template('user/userRegister.html', messages=form.errors)
def add_admin_user(): admin_user = User() admin_user.email = '*****@*****.**' admin_user.password = '******' admin_user.company = 'nbtjy' admin_user.company_address = 'JNL1588A' admin_user.linkman = 'sky' admin_user.phone_number = '12345678901' admin_user.authority = 'super_admin' admin_user.updatetime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime( (time.time()))) # 修改数据更新时间 admin_user.status = 1 # =0,该用户失效 # 加入到公司库内 admin_company = Company() admin_company.company = 'nbtjy' admin_company.company_number = '001' admin_company.company_address = 'JNL1588A' admin_company.linkman = 'sky' admin_company.phone_number = '12345678901' admin_company.status = 1 try: db.session.add(admin_user) db.session.add(admin_company) db.session.commit() # 创建报告保存文件夹 os.makedirs(os.path.join(current_app.config['DOCXFILE_DIR'], '001')) except: print('注册超级管理员账户失败!') else: print('注册超级管理员账户成功,请使用email=‘[email protected]’,password=‘123456’登录')