feat: add api auth

app/util/common.py

@@ -1,6 +1,11 @@
 import os
 from itertools import groupby
 from operator import itemgetter
+from functools import wraps
+
+from flask import request, session
+
+from app import rpc
 
 
 def split_group(dict_list, key):
@@ -13,3 +18,21 @@ def split_group(dict_list, key):
 
 
 basedir = os.getcwd()
+
+
+def login_required(f):
+    """
+    登陆保护，验证用户是否登陆
+    """
+
+    @wraps(f)
+    def wrapper(*args, **kwargs):
+        token = request.headers.get("Authorization", default=None)
+        if not token:
+            return '请登陆'
+        user_id = rpc.admin.identify(token)
+        if not user_id:
+            return '请登陆'
+        session['user_id'] = user_id
+        return f(*args, **kwargs)
+    return wrapper