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’登录')
