refactor(重构程序&目录结构)

applications/api/users/profile.py

@@ -1,11 +1,10 @@
 # 个人中心
 from flask import request, jsonify
-from flask_login import login_required
 from flask_restful import Resource, reqparse
 
-from applications.common.utils.http import fail_api, success_api
-from applications.extensions import db
-from applications.models import CompanyUser
+from common.utils.http import fail_api, success_api
+from extensions import db
+from models import UserModels
 
 
 class UserStatusResource(Resource):
@@ -19,11 +18,11 @@ class UserStatusResource(Resource):
         res = parser.parse_args()
 
         if res.operate == 1:
-            user = CompanyUser.query.get(user_id)
+            user = UserModels.query.get(user_id)
             user.enable = res.operate
             message = success_api(message="启动成功")
         else:
-            user = CompanyUser.query.filter_by(id=res.user_id).update({"enable": res.operate})
+            user = UserModels.query.filter_by(id=res.user_id).update({"enable": res.operate})
             message = success_api(message="禁用成功")
         if user:
             db.session.commit()
@@ -37,7 +36,7 @@ class UserAvatarResource(Resource):
 
     def put(self, user_id):
         url = request.json.get("avatar").get("src")
-        ret = CompanyUser.query.get(user_id)
+        ret = UserModels.query.get(user_id)
         ret.avatar = url
         db.session.commit()
         if not ret:
@@ -56,7 +55,7 @@ class UserInfoResource(Resource):
 
         res = parser.parse_args()
 
-        ret = CompanyUser.query.get(user_id)
+        ret = UserModels.query.get(user_id)
         ret.username = res.username
         ret.realname = res.real_name
         ret.remark = res.details
@@ -81,7 +80,7 @@ class UserPasswordResource(Resource):
             return fail_api(message='确认密码不一致')
 
         """ 修改当前用户密码 """
-        user = CompanyUser.query.get(user_id)
+        user = UserModels.query.get(user_id)
         is_right = user.validate_password(res.oldPassword)
         if not is_right:
             return jsonify(success=False, message="旧密码错误")