fix(修复模型名)

This commit is contained in:
zhengxinonly
2022-03-11 21:26:36 +08:00
parent 5333a68416
commit 1e34ef339a
18 changed files with 110 additions and 110 deletions
+8 -8
View File
@@ -3,13 +3,13 @@ from flask_restful import Resource, reqparse
from common.utils.http import success_api, fail_api from common.utils.http import success_api, fail_api
from extensions import db from extensions import db
from models import DepartmentModels, UserModels from models import DepartmentModel, UserModel
class DepartmentsResource(Resource): class DepartmentsResource(Resource):
def get(self): def get(self):
dept_data = DepartmentModels.query.order_by(DepartmentModels.sort).all() dept_data = DepartmentModel.query.order_by(DepartmentModel.sort).all()
# TODO dtree 需要返回状态信息 # TODO dtree 需要返回状态信息
res = { res = {
"status": {"code": 200, "message": "默认"}, "status": {"code": 200, "message": "默认"},
@@ -44,7 +44,7 @@ class DepartmentsResource(Resource):
res = parser.parse_args() res = parser.parse_args()
dept = DepartmentModels( dept = DepartmentModel(
parent_id=res.parent_id, parent_id=res.parent_id,
dept_name=res.dept_name, dept_name=res.dept_name,
sort=res.sort, sort=res.sort,
@@ -62,7 +62,7 @@ class DepartmentsResource(Resource):
class DepartmentResource(Resource): class DepartmentResource(Resource):
def get(self, dept_id): def get(self, dept_id):
dept = DepartmentModels.query.filter_by(id=dept_id).first() dept = DepartmentModel.query.filter_by(id=dept_id).first()
dept_data = { dept_data = {
'id': dept.id, 'id': dept.id,
'dept_name': dept.dept_name, 'dept_name': dept.dept_name,
@@ -95,15 +95,15 @@ class DepartmentResource(Resource):
"status": res.status, "status": res.status,
"address": res.address "address": res.address
} }
res = DepartmentModels.query.filter_by(id=dept_id).update(data) res = DepartmentModel.query.filter_by(id=dept_id).update(data)
if not res: if not res:
return fail_api(message="更新失败") return fail_api(message="更新失败")
db.session.commit() db.session.commit()
return success_api(message="更新成功") return success_api(message="更新成功")
def delete(self, dept_id): def delete(self, dept_id):
ret = DepartmentModels.query.filter_by(id=dept_id).delete() ret = DepartmentModel.query.filter_by(id=dept_id).delete()
UserModels.query.filter_by(dept_id=dept_id).update({"dept_id": None}) UserModel.query.filter_by(dept_id=dept_id).update({"dept_id": None})
db.session.commit() db.session.commit()
if ret: if ret:
return success_api(message="删除成功") return success_api(message="删除成功")
@@ -112,7 +112,7 @@ class DepartmentResource(Resource):
class DeptEnableResource(Resource): class DeptEnableResource(Resource):
def put(self, dept_id): def put(self, dept_id):
d = DepartmentModels.query.get(dept_id) d = DepartmentModel.query.get(dept_id)
if d: if d:
d.status = not d.status d.status = not d.status
db.session.commit() db.session.commit()
+12 -12
View File
@@ -7,7 +7,7 @@ from flask_restful import Resource, reqparse
from common.utils.http import success_api, fail_api from common.utils.http import success_api, fail_api
from extensions import db from extensions import db
from models import RightModels, RoleModels from models import RightModel, RoleModel
def get_render_config(): def get_render_config():
@@ -140,15 +140,15 @@ def make_menu_tree():
# 删除权限(目前没有判断父节点自动删除子节点) # 删除权限(目前没有判断父节点自动删除子节点)
def remove_power(power_id): def remove_power(power_id):
power = RightModels.query.filter_by(id=power_id).first() power = RightModel.query.filter_by(id=power_id).first()
role_id_list = [] role_id_list = []
roles = power.role roles = power.role
for role in roles: for role in roles:
role_id_list.append(role.id) role_id_list.append(role.id)
roles = RoleModels.query.filter(RoleModels.id.in_(role_id_list)).all() roles = RoleModel.query.filter(RoleModel.id.in_(role_id_list)).all()
for p in roles: for p in roles:
power.role.remove(p) power.role.remove(p)
r = RightModels.query.filter_by(id=power_id).delete() r = RightModel.query.filter_by(id=power_id).delete()
db.session.commit() db.session.commit()
return r return r
@@ -174,8 +174,8 @@ class RightRightsResource(Resource):
def get(self): def get(self):
"""获取选择父节点""" """获取选择父节点"""
power = RightModels.query.all() power = RightModel.query.all()
# power_data = marshal(power, RightModels.fields()) # power_data = marshal(power, RightModel.fields())
power_data = [ power_data = [
{ {
'powerId': item.id, 'powerId': item.id,
@@ -206,7 +206,7 @@ class RightPowerResource(Resource):
def post(self, power_id): def post(self, power_id):
res = parser_power.parse_args() res = parser_power.parse_args()
power = RightModels( power = RightModel(
icon=res.icon, icon=res.icon,
open_type=res.open_type, open_type=res.open_type,
parent_id=res.parent_id, parent_id=res.parent_id,
@@ -228,15 +228,15 @@ class RightPowerResource(Resource):
def delete(self, power_id): def delete(self, power_id):
# 删除权限(目前没有判断父节点自动删除子节点) # 删除权限(目前没有判断父节点自动删除子节点)
power = RightModels.query.filter_by(id=power_id).first() power = RightModel.query.filter_by(id=power_id).first()
role_id_list = [] role_id_list = []
roles = power.role roles = power.role
for role in roles: for role in roles:
role_id_list.append(role.id) role_id_list.append(role.id)
roles = RoleModels.query.filter(RoleModels.id.in_(role_id_list)).all() roles = RoleModel.query.filter(RoleModel.id.in_(role_id_list)).all()
for p in roles: for p in roles:
power.role.remove(p) power.role.remove(p)
r = RightModels.query.filter_by(id=power_id).delete() r = RightModel.query.filter_by(id=power_id).delete()
db.session.commit() db.session.commit()
if r: if r:
@@ -257,7 +257,7 @@ class RightPowerResource(Resource):
"url": res.power_url, "url": res.power_url,
"sort": res.sort "sort": res.sort
} }
power = RightModels.query.filter_by(id=power_id).update(data) power = RightModel.query.filter_by(id=power_id).update(data)
db.session.commit() db.session.commit()
if not power: if not power:
@@ -268,7 +268,7 @@ class RightPowerResource(Resource):
class RightPowerEnableResource(Resource): class RightPowerEnableResource(Resource):
def put(self, right_id): def put(self, right_id):
power = RightModels.query.get(right_id) power = RightModel.query.get(right_id)
if power: if power:
power.enable = not power.enable power.enable = not power.enable
db.session.commit() db.session.commit()
+16 -16
View File
@@ -2,27 +2,27 @@ from flask_restful import Resource, reqparse, marshal
from common.utils.http import table_api, success_api, fail_api from common.utils.http import table_api, success_api, fail_api
from extensions import db from extensions import db
from models import RightModels, RoleModels, UserModels from models import RightModel, RoleModel, UserModel
def remove_role(role_id): def remove_role(role_id):
""" 删除角色 """ """ 删除角色 """
role = RoleModels.query.filter_by(id=role_id).first() role = RoleModel.query.filter_by(id=role_id).first()
# 删除该角色的权限 # 删除该角色的权限
power_id_list = [] power_id_list = []
for p in role.power: for p in role.power:
power_id_list.append(p.id) power_id_list.append(p.id)
powers = RightModels.query.filter(RightModels.id.in_(power_id_list)).all() powers = RightModel.query.filter(RightModel.id.in_(power_id_list)).all()
for p in powers: for p in powers:
role.power.remove(p) role.power.remove(p)
user_id_list = [] user_id_list = []
for u in role.user: for u in role.user:
user_id_list.append(u.id) user_id_list.append(u.id)
users = UserModels.query.filter(UserModels.id.in_(user_id_list)).all() users = UserModel.query.filter(UserModel.id.in_(user_id_list)).all()
for u in users: for u in users:
role.user.remove(u) role.user.remove(u)
r = RoleModels.query.filter_by(id=role_id).delete() r = RoleModel.query.filter_by(id=role_id).delete()
db.session.commit() db.session.commit()
return r return r
@@ -46,11 +46,11 @@ class RoleRolesResource(Resource):
filters = [] filters = []
if res.role_name: if res.role_name:
filters.append(RoleModels.name.like('%' + res.role_name + '%')) filters.append(RoleModel.name.like('%' + res.role_name + '%'))
if res.role_code: if res.role_code:
filters.append(RoleModels.code.like('%' + res.role_code + '%')) filters.append(RoleModel.code.like('%' + res.role_code + '%'))
paginate = RoleModels.query.filter(*filters).paginate(page=res.page, per_page=res.limit, error_out=False) paginate = RoleModel.query.filter(*filters).paginate(page=res.page, per_page=res.limit, error_out=False)
return table_api(result={'items': [{'id': item.id, return table_api(result={'items': [{'id': item.id,
'roleName': item.name, 'roleName': item.name,
@@ -85,7 +85,7 @@ class RoleRoleResource(Resource):
res = parser.parse_args() res = parser.parse_args()
role = RoleModels( role = RoleModel(
details=res.details, details=res.details,
enable=res.enable, enable=res.enable,
code=res.role_code, code=res.role_code,
@@ -116,7 +116,7 @@ class RoleRoleResource(Resource):
"details": res.details "details": res.details
} }
role = RoleModels.query.filter_by(id=role_id).update(data) role = RoleModel.query.filter_by(id=role_id).update(data)
db.session.commit() db.session.commit()
if not role: if not role:
return fail_api(message="更新角色失败") return fail_api(message="更新角色失败")
@@ -127,7 +127,7 @@ class RoleEnableResource(Resource):
"""启用用户""" """启用用户"""
def put(self, role_id): def put(self, role_id):
ret = RoleModels.query.get(role_id) ret = RoleModel.query.get(role_id)
ret.enable = not ret.enable ret.enable = not ret.enable
db.session.commit() db.session.commit()
@@ -141,10 +141,10 @@ class RolePowerResource(Resource):
def get(self, role_id): def get(self, role_id):
# 获取角色权限 # 获取角色权限
role = RoleModels.query.filter_by(id=role_id).first() role = RoleModel.query.filter_by(id=role_id).first()
# 获取权限列表的 id # 获取权限列表的 id
check_powers_list = [rp.id for rp in role.power] check_powers_list = [rp.id for rp in role.power]
powers = RightModels.query.all() # 获取所有的权限 powers = RightModel.query.all() # 获取所有的权限
powers = [{ powers = [{
'powerId': item.id, 'powerId': item.id,
'powerName': item.name, 'powerName': item.name,
@@ -176,14 +176,14 @@ class RolePowerResource(Resource):
power_list = res.power_ids.split(',') power_list = res.power_ids.split(',')
""" 更新角色权限 """ """ 更新角色权限 """
role = RoleModels.query.filter_by(id=role_id).first() role = RoleModel.query.filter_by(id=role_id).first()
power_id_list = [] power_id_list = []
for p in role.power: for p in role.power:
power_id_list.append(p.id) power_id_list.append(p.id)
powers = RightModels.query.filter(RightModels.id.in_(power_id_list)).all() powers = RightModel.query.filter(RightModel.id.in_(power_id_list)).all()
for p in powers: for p in powers:
role.power.remove(p) role.power.remove(p)
powers = RightModels.query.filter(RightModels.id.in_(power_list)).all() powers = RightModel.query.filter(RightModel.id.in_(power_list)).all()
for p in powers: for p in powers:
role.power.append(p) role.power.append(p)
db.session.commit() db.session.commit()
+4 -4
View File
@@ -7,7 +7,7 @@ from sqlalchemy import desc
from common.utils.http import fail_api, success_api, table_api from common.utils.http import fail_api, success_api, table_api
from common.utils.upload import upload_one, delete_photo_by_id from common.utils.upload import upload_one, delete_photo_by_id
from extensions import db from extensions import db
from models import PhotoModels from models import PhotoModel
class FilePhotosResource(Resource): class FilePhotosResource(Resource):
@@ -15,7 +15,7 @@ class FilePhotosResource(Resource):
def get(self): def get(self):
page = request.args.get('page', type=int) page = request.args.get('page', type=int)
limit = request.args.get('limit', type=int) limit = request.args.get('limit', type=int)
photo_paginate = PhotoModels.query.order_by(desc(PhotoModels.create_at) photo_paginate = PhotoModel.query.order_by(desc(PhotoModel.create_at)
).paginate(page=page, ).paginate(page=page,
per_page=limit, per_page=limit,
error_out=False) error_out=False)
@@ -54,11 +54,11 @@ class FilePhotosResource(Resource):
"""图片批量删除""" """图片批量删除"""
# TODO bugs 图片删除失败 # TODO bugs 图片删除失败
ids = request.form.getlist('ids[]') ids = request.form.getlist('ids[]')
photo_name = PhotoModels.query.filter(PhotoModels.id.in_(ids)).all() photo_name = PhotoModel.query.filter(PhotoModel.id.in_(ids)).all()
upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST") upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST")
for p in photo_name: for p in photo_name:
os.remove(upload_url + '/' + p.name) os.remove(upload_url + '/' + p.name)
photo = PhotoModels.query.filter(PhotoModels.id.in_(ids)).delete(synchronize_session=False) photo = PhotoModel.query.filter(PhotoModel.id.in_(ids)).delete(synchronize_session=False)
db.session.commit() db.session.commit()
if photo: if photo:
return success_api(message="删除成功") return success_api(message="删除成功")
+2 -2
View File
@@ -6,7 +6,7 @@ from flask_restful import Resource, reqparse
from common.gen_captcha import add_auth_session from common.gen_captcha import add_auth_session
from common.utils.http import fail_api, success_api from common.utils.http import fail_api, success_api
from common.utils.rights import record_logging from common.utils.rights import record_logging
from models import UserModels from models import UserModel
class LoginResource(Resource): class LoginResource(Resource):
@@ -28,7 +28,7 @@ class LoginResource(Resource):
if req.captcha != s_code: if req.captcha != s_code:
return fail_api(message="验证码错误") return fail_api(message="验证码错误")
user = UserModels.query.filter_by(username=req.username).first() user = UserModel.query.filter_by(username=req.username).first()
if user is None: if user is None:
return fail_api(message="不存在的用户") return fail_api(message="不存在的用户")
+6 -6
View File
@@ -4,7 +4,7 @@ from flask_restful import Resource, reqparse
from common.utils.http import fail_api, success_api from common.utils.http import fail_api, success_api
from extensions import db from extensions import db
from models import UserModels from models import UserModel
class UserStatusResource(Resource): class UserStatusResource(Resource):
@@ -18,11 +18,11 @@ class UserStatusResource(Resource):
res = parser.parse_args() res = parser.parse_args()
if res.operate == 1: if res.operate == 1:
user = UserModels.query.get(user_id) user = UserModel.query.get(user_id)
user.enable = res.operate user.enable = res.operate
message = success_api(message="启动成功") message = success_api(message="启动成功")
else: else:
user = UserModels.query.filter_by(id=res.user_id).update({"enable": res.operate}) user = UserModel.query.filter_by(id=res.user_id).update({"enable": res.operate})
message = success_api(message="禁用成功") message = success_api(message="禁用成功")
if user: if user:
db.session.commit() db.session.commit()
@@ -36,7 +36,7 @@ class UserAvatarResource(Resource):
def put(self, user_id): def put(self, user_id):
url = request.json.get("avatar").get("src") url = request.json.get("avatar").get("src")
ret = UserModels.query.get(user_id) ret = UserModel.query.get(user_id)
ret.avatar = url ret.avatar = url
db.session.commit() db.session.commit()
if not ret: if not ret:
@@ -55,7 +55,7 @@ class UserInfoResource(Resource):
res = parser.parse_args() res = parser.parse_args()
ret = UserModels.query.get(user_id) ret = UserModel.query.get(user_id)
ret.username = res.username ret.username = res.username
ret.realname = res.real_name ret.realname = res.real_name
ret.remark = res.details ret.remark = res.details
@@ -80,7 +80,7 @@ class UserPasswordResource(Resource):
return fail_api(message='确认密码不一致') return fail_api(message='确认密码不一致')
""" 修改当前用户密码 """ """ 修改当前用户密码 """
user = UserModels.query.get(user_id) user = UserModel.query.get(user_id)
is_right = user.validate_password(res.oldPassword) is_right = user.validate_password(res.oldPassword)
if not is_right: if not is_right:
return jsonify(success=False, message="旧密码错误") return jsonify(success=False, message="旧密码错误")
+20 -20
View File
@@ -5,7 +5,7 @@ from sqlalchemy import desc
from common.utils.http import fail_api, success_api, table_api from common.utils.http import fail_api, success_api, table_api
from extensions import db from extensions import db
from models import UserModels, RoleModels, DepartmentModels from models import UserModel, RoleModel, DepartmentModel
from models import LogModel from models import LogModel
@@ -18,20 +18,20 @@ def get_current_user_logs():
def is_user_exists(username): def is_user_exists(username):
""" 判断用户是否存在 """ """ 判断用户是否存在 """
res = UserModels.query.filter_by(username=username).count() res = UserModel.query.filter_by(username=username).count()
return bool(res) return bool(res)
def delete_by_id(_id): def delete_by_id(_id):
""" 删除用户 """ """ 删除用户 """
user = UserModels.query.filter_by(id=_id).first() user = UserModel.query.filter_by(id=_id).first()
roles_id = [] roles_id = []
for role in user.role: for role in user.role:
roles_id.append(role.id) roles_id.append(role.id)
roles = RoleModels.query.filter(RoleModels.id.in_(roles_id)).all() roles = RoleModel.query.filter(RoleModel.id.in_(roles_id)).all()
for r in roles: for r in roles:
user.role.remove(r) user.role.remove(r)
res = UserModels.query.filter_by(id=_id).delete() res = UserModel.query.filter_by(id=_id).delete()
db.session.commit() db.session.commit()
return res return res
@@ -43,14 +43,14 @@ def batch_remove(ids):
def update_user_role(_id, roles_list): def update_user_role(_id, roles_list):
user = UserModels.query.filter_by(id=_id).first() user = UserModel.query.filter_by(id=_id).first()
roles_id = [] roles_id = []
for role in user.role: for role in user.role:
roles_id.append(role.id) roles_id.append(role.id)
roles = RoleModels.query.filter(RoleModels.id.in_(roles_id)).all() roles = RoleModel.query.filter(RoleModel.id.in_(roles_id)).all()
for r in roles: for r in roles:
user.role.remove(r) user.role.remove(r)
roles = RoleModels.query.filter(RoleModels.id.in_(roles_list)).all() roles = RoleModel.query.filter(RoleModel.id.in_(roles_list)).all()
for r in roles: for r in roles:
user.role.append(r) user.role.append(r)
db.session.commit() db.session.commit()
@@ -72,17 +72,17 @@ class UserUsersResource(Resource):
filters = [] filters = []
if res.real_name: if res.real_name:
filters.append(UserModels.realname.like('%' + res.real_name + '%')) filters.append(UserModel.realname.like('%' + res.real_name + '%'))
if res.username: if res.username:
filters.append(UserModels.username.like('%' + res.username + '%')) filters.append(UserModel.username.like('%' + res.username + '%'))
if res.dept_id: if res.dept_id:
filters.append(UserModels.dept_id == res.dept_id) filters.append(UserModel.dept_id == res.dept_id)
paginate = UserModels.query.filter(*filters).paginate(page=res.page, paginate = UserModel.query.filter(*filters).paginate(page=res.page,
per_page=res.limit, per_page=res.limit,
error_out=False) error_out=False)
dept_name = lambda dept_id: DepartmentModels.query.filter_by(id=dept_id).first().dept_name if dept_id else "" dept_name = lambda dept_id: DepartmentModel.query.filter_by(id=dept_id).first().dept_name if dept_id else ""
user_data = [{ user_data = [{
'id': item.id, 'id': item.id,
'username': item.username, 'username': item.username,
@@ -111,7 +111,7 @@ class UserUsersResource(Resource):
if is_user_exists(res.username): if is_user_exists(res.username):
return fail_api(message="用户已经存在") return fail_api(message="用户已经存在")
user = UserModels() user = UserModel()
user.username = res.username user.username = res.username
user.realname = res.real_name user.realname = res.real_name
user.set_password(res.password) user.set_password(res.password)
@@ -119,8 +119,8 @@ class UserUsersResource(Resource):
db.session.commit() db.session.commit()
""" 增加用户角色 """ """ 增加用户角色 """
user = UserModels.query.filter_by(id=user.id).first() user = UserModel.query.filter_by(id=user.id).first()
roles = RoleModels.query.filter(RoleModels.id.in_(role_ids)).all() roles = RoleModel.query.filter(RoleModel.id.in_(role_ids)).all()
for r in roles: for r in roles:
user.role.append(r) user.role.append(r)
db.session.commit() db.session.commit()
@@ -152,7 +152,7 @@ class UserUserResource(Resource):
if is_user_exists(res.username): if is_user_exists(res.username):
return fail_api(message="用户已经存在") return fail_api(message="用户已经存在")
user = UserModels() user = UserModel()
user.username = res.username user.username = res.username
user.realname = res.real_name user.realname = res.real_name
user.set_password(res.password) user.set_password(res.password)
@@ -160,8 +160,8 @@ class UserUserResource(Resource):
db.session.commit() db.session.commit()
""" 增加用户角色 """ """ 增加用户角色 """
user = UserModels.query.filter_by(id=user.id).first() user = UserModel.query.filter_by(id=user.id).first()
roles = RoleModels.query.filter(RoleModels.id.in_(role_ids)).all() roles = RoleModel.query.filter(RoleModel.id.in_(role_ids)).all()
for r in roles: for r in roles:
user.role.append(r) user.role.append(r)
db.session.commit() db.session.commit()
@@ -189,7 +189,7 @@ class UserRoleResource(Resource):
role_ids = res.role_ids.split(',') role_ids = res.role_ids.split(',')
# 更新用户数据 # 更新用户数据
UserModels.query.filter_by(id=user_id).update({'username': res.username, UserModel.query.filter_by(id=user_id).update({'username': res.username,
'realname': res.real_name, 'realname': res.real_name,
'dept_id': res.dept_id}) 'dept_id': res.dept_id})
db.session.commit() db.session.commit()
+2 -2
View File
@@ -1,6 +1,6 @@
from flask import render_template, request from flask import render_template, request
from models import DepartmentModels from models import DepartmentModel
from common.utils.rights import permission_required, view_logging_required from common.utils.rights import permission_required, view_logging_required
from applications.view import index_bp from applications.view import index_bp
@@ -24,5 +24,5 @@ def add():
@permission_required("admin:dept:edit") @permission_required("admin:dept:edit")
def edit(): def edit():
dept_id = request.args.get("deptId", type=int) dept_id = request.args.get("deptId", type=int)
dept = DepartmentModels.query.get(dept_id) dept = DepartmentModel.query.get(dept_id)
return render_template('admin/department/dept_edit.html', dept=dept) return render_template('admin/department/dept_edit.html', dept=dept)
+2 -2
View File
@@ -1,7 +1,7 @@
from flask import render_template from flask import render_template
from common.utils.rights import permission_required, view_logging_required from common.utils.rights import permission_required, view_logging_required
from models import RightModels from models import RightModel
from applications.view import index_bp from applications.view import index_bp
@@ -16,7 +16,7 @@ def rights_index():
@view_logging_required @view_logging_required
@permission_required("admin:power:edit") @permission_required("admin:power:edit")
def rights_edit(power_id): def rights_edit(power_id):
power = RightModels.query.filter_by(id=power_id).first() power = RightModel.query.filter_by(id=power_id).first()
icon = str(power.icon).split() icon = str(power.icon).split()
if len(icon) == 2: if len(icon) == 2:
icon = icon[1] icon = icon[1]
+2 -2
View File
@@ -2,7 +2,7 @@ from flask import Blueprint, render_template
from common.utils.rights import permission_required, view_logging_required from common.utils.rights import permission_required, view_logging_required
from models import RoleModels from models import RoleModel
role_bp = Blueprint('role', __name__, url_prefix='/admin/role') role_bp = Blueprint('role', __name__, url_prefix='/admin/role')
@@ -28,7 +28,7 @@ def power(role_id):
@view_logging_required @view_logging_required
@permission_required("admin:role:edit") @permission_required("admin:role:edit")
def role_editor(role_id): def role_editor(role_id):
role = RoleModels.query.filter_by(id=role_id).first() role = RoleModel.query.filter_by(id=role_id).first()
return render_template('admin/roles/roles_edit.html', role=role) return render_template('admin/roles/roles_edit.html', role=role)
+4 -4
View File
@@ -3,7 +3,7 @@ from flask_login import login_required, current_user
from sqlalchemy import desc from sqlalchemy import desc
from common.utils.rights import permission_required, view_logging_required from common.utils.rights import permission_required, view_logging_required
from models import LogModel, RoleModels, UserModels from models import LogModel, RoleModel, UserModel
from . import index_bp from . import index_bp
@@ -19,7 +19,7 @@ def users_main():
@view_logging_required @view_logging_required
@permission_required("admin:user:add") @permission_required("admin:user:add")
def users_add_view(): def users_add_view():
roles = RoleModels.query.all() roles = RoleModel.query.all()
return render_template('admin/users/users_add.html', roles=roles) return render_template('admin/users/users_add.html', roles=roles)
@@ -28,8 +28,8 @@ def users_add_view():
@permission_required("admin:user:edit") @permission_required("admin:user:edit")
def users_user_id_view(user_id): def users_user_id_view(user_id):
# 获取编辑用户信息 # 获取编辑用户信息
user = UserModels.query.filter_by(id=user_id).first() user = UserModel.query.filter_by(id=user_id).first()
roles = RoleModels.query.all() roles = RoleModel.query.all()
checked_roles = [] checked_roles = []
for r in user.role: for r in user.role:
checked_roles.append(r.id) checked_roles.append(r.id)
+4 -4
View File
@@ -4,7 +4,7 @@ from flask import current_app
from common.flask_uploads import UploadSet, IMAGES from common.flask_uploads import UploadSet, IMAGES
from extensions import db from extensions import db
from models import PhotoModels from models import PhotoModel
photos = UploadSet('photos', IMAGES) photos = UploadSet('photos', IMAGES)
@@ -15,15 +15,15 @@ def upload_one(photo, mime):
upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST") upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST")
size = os.path.getsize(upload_url + '/' + filename) size = os.path.getsize(upload_url + '/' + filename)
photo = PhotoModels(name=filename, href=file_url, mime=mime, size=size) photo = PhotoModel(name=filename, href=file_url, mime=mime, size=size)
db.session.add(photo) db.session.add(photo)
db.session.commit() db.session.commit()
return file_url return file_url
def delete_photo_by_id(_id): def delete_photo_by_id(_id):
photo_name = PhotoModels.query.filter_by(id=_id).first().name photo_name = PhotoModel.query.filter_by(id=_id).first().name
photo = PhotoModels.query.filter_by(id=_id).delete() photo = PhotoModel.query.filter_by(id=_id).delete()
db.session.commit() db.session.commit()
upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST") upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST")
os.remove(upload_url + '/' + photo_name) os.remove(upload_url + '/' + photo_name)
+10 -10
View File
@@ -39,20 +39,20 @@ def register_script(app: Flask):
"""数据库初始化""" """数据库初始化"""
# 创建化部门数据 # 创建化部门数据
from models import DepartmentModels from models import DepartmentModel
add_data(cp_dept_data_list, DepartmentModels) add_data(cp_dept_data_list, DepartmentModel)
# 图片数据 # 图片数据
from models import PhotoModels from models import PhotoModel
add_data(file_photo_data_list, PhotoModels) add_data(file_photo_data_list, PhotoModel)
# 初始化权限表数据 # 初始化权限表数据
from models import RightModels from models import RightModel
add_data(rt_power_data_list, RightModels) add_data(rt_power_data_list, RightModel)
# 初始化角色表 # 初始化角色表
from models import RoleModels from models import RoleModel
# 角色权限关系表 # 角色权限关系表
from extensions import db from extensions import db
@@ -60,12 +60,12 @@ def register_script(app: Flask):
db.session.execute('insert into rt_role_power VALUES (%s, %s, %s);' % tuple(data)) db.session.execute('insert into rt_role_power VALUES (%s, %s, %s);' % tuple(data))
db.session.commit() db.session.commit()
add_data(rt_role_data_list, RoleModels) add_data(rt_role_data_list, RoleModel)
# 管理员用户 # 管理员用户
from models import UserModels from models import UserModel
add_data(cp_user_data_list, UserModels) add_data(cp_user_data_list, UserModel)
# 用户角色表 # 用户角色表
from extensions import db from extensions import db
+2 -2
View File
@@ -10,6 +10,6 @@ def init_login_manager(app):
@login_manager.user_loader @login_manager.user_loader
def load_user(user_id): def load_user(user_id):
from models import UserModels from models import UserModel
user = UserModels.query.get(int(user_id)) user = UserModel.query.get(int(user_id))
return user return user
+3 -3
View File
@@ -1,4 +1,4 @@
from .file import PhotoModels from .file import PhotoModel
from .rights import RightModels, RoleModels from .rights import RightModel, RoleModel
from .system import LogModel from .system import LogModel
from .users import UserModels, DepartmentModels from .users import UserModel, DepartmentModel
+1 -1
View File
@@ -3,7 +3,7 @@ from datetime import datetime
from extensions import db from extensions import db
class PhotoModels(db.Model): class PhotoModel(db.Model):
__tablename__ = 'file_photo' __tablename__ = 'file_photo'
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(255), nullable=False) name = db.Column(db.String(255), nullable=False)
+4 -4
View File
@@ -17,7 +17,7 @@ role_power = db.Table(
) )
class RightModels(db.Model): class RightModel(db.Model):
__tablename__ = 'rt_power' __tablename__ = 'rt_power'
id = db.Column(db.Integer, primary_key=True, comment='权限编号') id = db.Column(db.Integer, primary_key=True, comment='权限编号')
name = db.Column(db.String(255), comment='权限名称') name = db.Column(db.String(255), comment='权限名称')
@@ -30,10 +30,10 @@ class RightModels(db.Model):
sort = db.Column(db.Integer, comment='排序') sort = db.Column(db.Integer, comment='排序')
enable = db.Column(db.Boolean, comment='是否开启') enable = db.Column(db.Boolean, comment='是否开启')
parent = db.relationship("RightModels", remote_side=[id]) # 自关联 parent = db.relationship("RightModel", remote_side=[id]) # 自关联
class RoleModels(db.Model): class RoleModel(db.Model):
__tablename__ = 'rt_role' __tablename__ = 'rt_role'
id = db.Column(db.Integer, primary_key=True, comment='角色ID') id = db.Column(db.Integer, primary_key=True, comment='角色ID')
name = db.Column(db.String(255), comment='角色名称') name = db.Column(db.String(255), comment='角色名称')
@@ -43,4 +43,4 @@ class RoleModels(db.Model):
details = db.Column(db.String(255), comment='详情') details = db.Column(db.String(255), comment='详情')
sort = db.Column(db.Integer, comment='排序') sort = db.Column(db.Integer, comment='排序')
power = db.relationship('RightModels', secondary="rt_role_power", backref=db.backref('role')) power = db.relationship('RightModel', secondary="rt_role_power", backref=db.backref('role'))
+3 -3
View File
@@ -6,7 +6,7 @@ from werkzeug.security import generate_password_hash, check_password_hash
from extensions import db from extensions import db
class UserModels(db.Model, UserMixin): class UserModel(db.Model, UserMixin):
__tablename__ = 'cp_user' __tablename__ = 'cp_user'
id = db.Column(db.Integer, primary_key=True, autoincrement=True, comment='用户ID') id = db.Column(db.Integer, primary_key=True, autoincrement=True, comment='用户ID')
username = db.Column(db.String(20), comment='用户名') username = db.Column(db.String(20), comment='用户名')
@@ -18,7 +18,7 @@ class UserModels(db.Model, UserMixin):
enable = db.Column(db.Integer, default=0, comment='启用') enable = db.Column(db.Integer, default=0, comment='启用')
dept_id = db.Column(db.Integer, comment='部门id') dept_id = db.Column(db.Integer, comment='部门id')
role = db.relationship('RoleModels', secondary="rt_user_role", backref=db.backref('user'), lazy='dynamic') role = db.relationship('RoleModel', secondary="rt_user_role", backref=db.backref('user'), lazy='dynamic')
def set_password(self, password): def set_password(self, password):
"""设置密码,对密码进行加密存储""" """设置密码,对密码进行加密存储"""
@@ -32,7 +32,7 @@ class UserModels(db.Model, UserMixin):
update_at = db.Column(db.DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间') update_at = db.Column(db.DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
class DepartmentModels(db.Model): class DepartmentModel(db.Model):
__tablename__ = 'cp_dept' __tablename__ = 'cp_dept'
id = db.Column(db.Integer, primary_key=True, comment="部门ID") id = db.Column(db.Integer, primary_key=True, comment="部门ID")
parent_id = db.Column(db.Integer, comment="父级编号") parent_id = db.Column(db.Integer, comment="父级编号")