From 59a08cd645e8fcea56c3b473d69d1d64aeae83c8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 28 Aug 2023 07:01:01 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(crud.py):=20change=20retur?= =?UTF-8?q?n=20type=20of=20check=5Fkey=20function=20from=20bool=20to=20Opt?= =?UTF-8?q?ional[ApiKey]=20to=20provide=20more=20information=20about=20the?= =?UTF-8?q?=20API=20key=20=F0=9F=94=A7=20chore(crud.py):=20rename=20api=5F?= =?UTF-8?q?key=20variable=20to=20api=5Fkey=5Fobject=20in=20check=5Fkey=20f?= =?UTF-8?q?unction=20for=20better=20clarity=20=F0=9F=94=A7=20chore(crud.py?= =?UTF-8?q?):=20change=20return=20value=20of=20check=5Fkey=20function=20fr?= =?UTF-8?q?om=20True=20to=20api=5Fkey=5Fobject.user=20to=20provide=20the?= =?UTF-8?q?=20associated=20user=20with=20the=20API=20key?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../services/database/models/api_key/crud.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/backend/langflow/services/database/models/api_key/crud.py b/src/backend/langflow/services/database/models/api_key/crud.py index 73ac8de7c..adf53d382 100644 --- a/src/backend/langflow/services/database/models/api_key/crud.py +++ b/src/backend/langflow/services/database/models/api_key/crud.py @@ -2,7 +2,7 @@ import datetime import secrets import threading from uuid import UUID -from typing import List +from typing import List, Optional from sqlmodel import Session, select from langflow.services.database.models.api_key import ( ApiKey, @@ -46,21 +46,21 @@ def delete_api_key(session: Session, api_key_id: UUID) -> None: session.commit() -def check_key(session: Session, api_key: str) -> bool: +def check_key(session: Session, api_key: str) -> Optional[ApiKey]: """Check if the API key is valid.""" query = select(ApiKey).where(ApiKey.api_key == api_key) - api_key = session.exec(query).first() - if api_key is None: - return False + api_key_object: Optional[ApiKey] = session.exec(query).first() + if api_key_object is None: + return api_key_object threading.Thread( target=update_total_uses, args=( session, - api_key, + api_key_object, ), ).start() - return True + return api_key_object.user def update_total_uses(session, api_key: ApiKey):