Skip to content

Commit 3fa7a11

Browse files
gakkiyomialexcodelf
authored andcommitted
improve the readability of the function generate_api_key (langgenius#12552)
1 parent 1e6c3da commit 3fa7a11

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

api/models/model.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -1405,9 +1405,8 @@ class ApiToken(db.Model): # type: ignore[name-defined]
14051405
def generate_api_key(prefix, n):
14061406
while True:
14071407
result = prefix + generate_string(n)
1408-
while db.session.query(ApiToken).filter(ApiToken.token == result).count() > 0:
1409-
result = prefix + generate_string(n)
1410-
1408+
if db.session.query(ApiToken).filter(ApiToken.token == result).count() > 0:
1409+
continue
14111410
return result
14121411

14131412

0 commit comments

Comments
 (0)