From 503559d06c97dc7cd921b7307856cb34c7383284 Mon Sep 17 00:00:00 2001
From: hole-thu <hole_thu@riseup.net>
Date: Tue, 8 Sep 2020 20:21:57 +0800
Subject: [PATCH] =?UTF-8?q?token=E4=BF=A1=E6=81=AF=E8=BD=AC=E7=A7=BB?=
 =?UTF-8?q?=E5=88=B0headers=EF=BC=8C=E5=85=BC=E5=AE=B9url=E5=8F=82?=
 =?UTF-8?q?=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index 7b642de..6c1d0c3 100644
--- a/utils.py
+++ b/utils.py
@@ -9,7 +9,7 @@ def tmp_token():
     return hash_name(str(int(time.time() / 900)) + User.query.get(1).token)[5:21]
 
 def require_token():
-    token = request.args.get('user_token')
+    token = request.headers.get('User-Token') or request.args.get('user_token')
     if not token: abort(401)
 
     if len(token.split('_')) == 2 and get_config('ENABLE_TMP'):