jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔒 chore(api_key.py): refactor mask_api_key validator to improve readability and maintainability

Browse source 
🔒 chore(api_key.py): update api_key field in ApiKeyRead model to remove index=True and unique=True constraints
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-28 06:48:38 -03:00
commit 4bc4a39e5e
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/services/database/models/api_key/api_key.py
View file

@ -39,10 +39,10 @@ class UnmaskedApiKeyRead(ApiKeyBase):
class ApiKeyRead(ApiKeyBase): class ApiKeyRead(ApiKeyBase):
id: UUID id: UUID
api_key: str = Field(index=True, unique=True) api_key: str = Field()
user_id: UUID = Field() user_id: UUID = Field()
@validator("api_key", always=True) @validator("api_key", always=True)
def mask_api_key(cls, v): def mask_api_key(cls, v):
# This validator will always run, and will mask the API key # This validator will always run, and will mask the API key
return f"{v[:2]}{'*' * (len(v) - 4)}{v[-2:]}" return f"{v[:8]}{'*' * (len(v) - 8)}"