Reservationer/assets/reservation_loader.py

import sqlmodel
import sqlmodel.sql._expression_select_cls

from .product import product
from .reservation import reservation
from .reservation import reservation_factory
from .reservations_collection import reservations_collection

class reservation_loader(sqlmodel.Session): 
    def get_by_email(self, target: str) -> reservations_collection:
        pass

    def get_by_phone_number(self, target: str) -> reservations_collection:
        pass

    def get_by_target(self, target: product) -> reservations_collection:
        pass

    def store(self, target: reservation) -> bool:
        if not target.ready:
            raise RuntimeError("Reservation is not setup.")

        if target.in_database:
            error = "Target reservation is already in database. Reservation "
            error = error + "object is not editable."

            raise reservation_loader_exception(error)

        try:
            self.add(target)
            self.commit()
            self.refresh(target)

            return True
        
        except:
            return False

    def drop(self, target: reservation) -> bool:
        if not target.in_database:
            raise reservation_loader_exception("Reservation does not exists.")

        try:
            self.delete(target)
            self.commit()

            return True

        except:
            return False

    @property
    def __select(self) -> sqlmodel.sql._expression_select_cls.SelectOfScalar:
        return sqlmodel.select(reservation)