42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from sqlmodel import Session, select
|
|
from src.models import Productor
|
|
|
|
def get_all(session: Session) -> list[Productor]:
|
|
statement = select(Productor)
|
|
return session.exec(statement).all()
|
|
|
|
def get_one(session: Session, productor_id: int) -> Productor:
|
|
return session.get(Productor, productor_id)
|
|
|
|
def create_one(session: Session, productor: Productor) -> Productor:
|
|
productor_create = productor.model_dump(exclude_unset=True)
|
|
new_productor = Productor(**productor_create)
|
|
session.add(new_productor)
|
|
session.commit()
|
|
session.refresh(new_productor)
|
|
return new_productor
|
|
|
|
def update_one(session: Session, id: int, productor: Productor) -> Productor:
|
|
statement = select(Productor).where(Productor.id == id)
|
|
result = session.exec(statement)
|
|
new_productor = result.first()
|
|
if not new_productor:
|
|
return None
|
|
productor_updates = productor.model_dump(exclude_unset=True)
|
|
for key, value in productor_updates.items():
|
|
setattr(new_productor, key, value)
|
|
session.add(new_productor)
|
|
session.commit()
|
|
session.refresh(new_productor)
|
|
return new_productor
|
|
|
|
def delete_one(session: Session, id: int) -> Productor:
|
|
statement = select(Productor).where(Productor.id == id)
|
|
result = session.exec(statement)
|
|
productor = result.first()
|
|
if not productor:
|
|
return None
|
|
session.delete(productor)
|
|
session.commit()
|
|
return productor
|