fix:add a shortcut for authing in development mode without openid

pkuphysu · Oct 22, 2021 · ff9d4bc · ff9d4bc
1 parent 6bfa431
commit ff9d4bc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/pkuphysu_wechat/api/random_draw/views.py b/src/pkuphysu_wechat/api/random_draw/views.py
@@ -24,6 +24,8 @@ def join():
 @bp.route("/all", methods=["GET"])
 def get_all():
     openid = token_required()
-    if openid not in settings.wechat.MASTER_IDS:
+    if openid not in settings.WECHAT.MASTER_IDS and not (
+        openid == "developmentopenid" and not settings.PRODUCTION
+    ):
         abort(404)
     return respond_success(data=[record.name for record in RandomDraw.query.all()])
diff --git a/src/pkuphysu_wechat/auth/utils.py b/src/pkuphysu_wechat/auth/utils.py
@@ -31,7 +31,9 @@ def token_required() -> str:
 
 def master_required():
     openid = token_required()
-    if openid not in settings.WECHAT.MASTER_IDS:
+    if openid not in settings.WECHAT.MASTER_IDS and not (
+        openid == "developmentopenid" and not settings.PRODUCTION
+    ):
         logger.info("%s tried to access admin resouces", openid)
         abort(respond_error(403, "NoHackMaster"))