فهرست منبع

improve the readability of the function generate_api_key (#12552)

gakkiyomi 1 سال پیش
والد
کامیت
989fb11fd7
1فایلهای تغییر یافته به همراه2 افزوده شده و 3 حذف شده
  1. 2 3
      api/models/model.py

+ 2 - 3
api/models/model.py

@@ -1405,9 +1405,8 @@ class ApiToken(db.Model):  # type: ignore[name-defined]
     def generate_api_key(prefix, n):
         while True:
             result = prefix + generate_string(n)
-            while db.session.query(ApiToken).filter(ApiToken.token == result).count() > 0:
-                result = prefix + generate_string(n)
-
+            if db.session.query(ApiToken).filter(ApiToken.token == result).count() > 0:
+                continue
             return result