vendimia-compose/backend/cruds/finca_company.py

from typing import List

from sqlalchemy.orm import Session
from sqlalchemy.sql.sqltypes import Boolean

from cruds.utils import order_fincas
from models.finca_company import FincaCompany as FincaCompanyModel
from schemas.finca_company import FincaCompany as FincaCompanySchema


def get_finca_company_by_company_id(db: Session, company_id: int) -> FincaCompanyModel:
    return (
        db.query(FincaCompanyModel)
        .filter(FincaCompanyModel.company_id == company_id)
        .all()
    )


def create_finca_company(
    db: Session, finca_company: FincaCompanySchema
) -> FincaCompanyModel:
    db_finca_company = FincaCompanyModel(
        company_id=finca_company.company_id,
        finca_station_code=finca_company.finca_station_code,
    )
    db.add(db_finca_company)
    db.commit()
    db.refresh(db_finca_company)
    return db_finca_company


def delete_finca_company(db: Session, company_id: int, station_code: str) -> Boolean:
    finca_company = (
        db.query(FincaCompanyModel)
        .filter(
            FincaCompanyModel.company_id == company_id,
            FincaCompanyModel.finca_station_code == station_code,
        )
        .first()
    )
    db.delete(finca_company)
    db.commit()
    return True