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

common/utils/http.py

@@ -0,0 +1,36 @@
+import typing as t
+
+
+def success_api(message: str = "成功", code: int = 200) -> t.Dict:
+    """ 成功响应 默认值”成功“ """
+    return dict(success=True, message=message, code=code)
+
+
+def fail_api(message: str = "失败", code: int = 404) -> t.Dict:
+    """ 失败响应 默认值“失败” """
+    return dict(success=False, message=message, code=code)
+
+
+def table_api(success: bool = True,
+              message: str = "",
+              result: t.Union[dict, list] = None,
+              code: int = 0) -> t.Dict:
+    """
+        动态表格渲染响应
+        此方法返回数据给前端
+        {
+            'success': True,
+            'code': 10002,
+            'message': '提示消息',
+            'result':{'items':[],'total': 100}
+        }
+
+        注：lay_ui 表格数据需要指定 code=0
+    """
+    ret = {
+        'success': success,
+        'message': message,
+        'code': code,
+        'result': result,
+    }
+    return ret

common/utils/rights.py

@@ -0,0 +1,61 @@
+import typing as t
+from functools import wraps
+
+from flask import abort, request, jsonify, session
+from flask_login import login_required
+from flask_login import current_user
+
+from extensions import db
+from models import LogModel
+
+
+def record_logging(success: bool = True) -> None:
+    """
+    记录用户日志数据
+    """
+    info = {
+        'method': request.method,
+        'url': request.path,
+        'ip': request.remote_addr,
+        'user_agent': request.headers.get('User-Agent'),
+        'desc': str(dict(request.values)),
+        'uid': current_user.id,
+        'success': success
+    }
+    log = LogModel()
+    for key, value in info.items():
+        setattr(log, key, value)
+
+    db.session.add(log)
+    db.session.commit()
+
+
+def view_logging_required(func: t.Callable) -> t.Callable:
+    """
+    日志装饰器，用于记录请求
+    """
+
+    @wraps(func)
+    def wrapper(*args, **kwargs) -> t.Callable:
+        record_logging()
+        return func(*args, **kwargs)
+
+    return wrapper
+
+
+def permission_required(permission: str) -> t.Callable:
+    """
+    权限装饰器，用于过滤需要的权限
+    """
+
+    def decorator(func: t.Callable):
+        @wraps(func)
+        def wrapper(*args, **kwargs) -> t.Callable:
+            if permission not in session.get('permissions'):
+                record_logging(success=False)
+                abort(403)
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

common/utils/upload.py

@@ -0,0 +1,30 @@
+import os
+
+from flask import current_app
+
+from common.flask_uploads import UploadSet, IMAGES
+from extensions import db
+from models import PhotoModels
+
+photos = UploadSet('photos', IMAGES)
+
+
+def upload_one(photo, mime):
+    filename = photos.save(photo)
+    file_url = photos.url(filename)
+
+    upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST")
+    size = os.path.getsize(upload_url + '/' + filename)
+    photo = PhotoModels(name=filename, href=file_url, mime=mime, size=size)
+    db.session.add(photo)
+    db.session.commit()
+    return file_url
+
+
+def delete_photo_by_id(_id):
+    photo_name = PhotoModels.query.filter_by(id=_id).first().name
+    photo = PhotoModels.query.filter_by(id=_id).delete()
+    db.session.commit()
+    upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST")
+    os.remove(upload_url + '/' + photo_name)
+    return photo