修改登录和基础功能样式,目前已经支持钉钉登录和自定义登录注册
This commit is contained in:
27
init_db.py
27
init_db.py
@@ -1,19 +1,42 @@
|
||||
from __init__ import app, db
|
||||
from models import User
|
||||
from sqlalchemy import text
|
||||
|
||||
with app.app_context():
|
||||
db.create_all()
|
||||
|
||||
if not User.query.filter_by(username='管理员').first():
|
||||
# 检查并添加 password_hash 字段(如果不存在)
|
||||
try:
|
||||
db.session.execute(text("ALTER TABLE user ADD COLUMN password_hash VARCHAR(200)"))
|
||||
db.session.commit()
|
||||
print('已添加 password_hash 字段')
|
||||
except Exception as e:
|
||||
if 'duplicate column' in str(e).lower() or 'already exists' in str(e).lower():
|
||||
print('password_hash 字段已存在')
|
||||
else:
|
||||
pass # 字段可能已存在,忽略错误
|
||||
|
||||
# 创建默认管理员账号
|
||||
admin = User.query.filter_by(username='管理员').first()
|
||||
if not admin:
|
||||
admin = User(
|
||||
username='管理员',
|
||||
username='admin',
|
||||
dingtalk_userid='admin',
|
||||
dingtalk_name='管理员',
|
||||
dingtalk_dept='',
|
||||
role='admin'
|
||||
)
|
||||
admin.set_password('admin123') # 设置默认密码
|
||||
db.session.add(admin)
|
||||
db.session.commit()
|
||||
print('数据库初始化完成,已创建管理员账户')
|
||||
print('用户名: admin')
|
||||
print('密码: admin123')
|
||||
print('请及时修改默认密码!')
|
||||
else:
|
||||
# 确保管理员有密码
|
||||
if not admin.password_hash:
|
||||
admin.set_password('admin123')
|
||||
db.session.commit()
|
||||
print('已为管理员设置默认密码: admin123')
|
||||
print('数据库已存在')
|
||||
|
||||
Reference in New Issue
Block a user