From 989fb11fd715a96576333207189a12d8594c60f1 Mon Sep 17 00:00:00 2001 From: gakkiyomi Date: Thu, 9 Jan 2025 21:30:17 +0800 Subject: [PATCH] improve the readability of the function generate_api_key (#12552) --- api/models/model.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/api/models/model.py b/api/models/model.py index d2d4d5853f..d6f73c5ede 100644 --- a/api/models/model.py +++ b/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