Refactor credential.py to use SQLModel's select
method
This commit is contained in:
parent
580f3a1e0d
commit
62c053030b
1 changed files with 7 additions and 5 deletions
|
|
@ -4,10 +4,13 @@ from uuid import UUID
|
||||||
from fastapi import APIRouter, Depends, HTTPException
|
from fastapi import APIRouter, Depends, HTTPException
|
||||||
from langflow.services.auth import utils as auth_utils
|
from langflow.services.auth import utils as auth_utils
|
||||||
from langflow.services.auth.utils import get_current_active_user
|
from langflow.services.auth.utils import get_current_active_user
|
||||||
from langflow.services.database.models.credential import Credential, CredentialCreate, CredentialRead, CredentialUpdate
|
from langflow.services.database.models.credential import (Credential,
|
||||||
|
CredentialCreate,
|
||||||
|
CredentialRead,
|
||||||
|
CredentialUpdate)
|
||||||
from langflow.services.database.models.user.model import User
|
from langflow.services.database.models.user.model import User
|
||||||
from langflow.services.deps import get_session, get_settings_service
|
from langflow.services.deps import get_session, get_settings_service
|
||||||
from sqlmodel import Session
|
from sqlmodel import Session, select
|
||||||
|
|
||||||
router = APIRouter(prefix="/credentials", tags=["Credentials"])
|
router = APIRouter(prefix="/credentials", tags=["Credentials"])
|
||||||
|
|
||||||
|
|
@ -55,7 +58,7 @@ def read_credentials(
|
||||||
):
|
):
|
||||||
"""Read all credentials."""
|
"""Read all credentials."""
|
||||||
try:
|
try:
|
||||||
credentials = session.query(Credential).filter(Credential.user_id == current_user.id).all()
|
credentials = session.exec(select(Credential).where(Credential.user_id == current_user.id)).all()
|
||||||
return credentials
|
return credentials
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise HTTPException(status_code=500, detail=str(e)) from e
|
raise HTTPException(status_code=500, detail=str(e)) from e
|
||||||
|
|
@ -72,8 +75,7 @@ def update_credential(
|
||||||
"""Update a credential."""
|
"""Update a credential."""
|
||||||
try:
|
try:
|
||||||
db_credential = (
|
db_credential = (
|
||||||
session.query(Credential)
|
session.exec(select(Credential).where(Credential.id == credential_id and Credential.user_id == current_user.id))
|
||||||
.filter(Credential.id == credential_id, Credential.user_id == current_user.id)
|
|
||||||
.first()
|
.first()
|
||||||
)
|
)
|
||||||
if not db_credential:
|
if not db_credential:
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue