mealie/mealie/db/data_access_layer/recipe_access_model.py

from random import randint

from mealie.db.models.recipe.recipe import RecipeModel
from mealie.db.models.recipe.settings import RecipeSettings
from mealie.schema.recipe import Recipe
from sqlalchemy.orm.session import Session

from ._base_access_model import BaseAccessModel


class RecipeDataAccessModel(BaseAccessModel[Recipe, RecipeModel]):
    def get_all_public(self, session: Session, limit: int = None, order_by: str = None, start=0, override_schema=None):
        eff_schema = override_schema or self.schema

        if order_by:
            order_attr = getattr(self.sql_model, str(order_by))

            return [
                eff_schema.from_orm(x)
                for x in session.query(self.sql_model)
                .join(RecipeSettings)
                .filter(RecipeSettings.public == True)  # noqa: 711
                .order_by(order_attr.desc())
                .offset(start)
                .limit(limit)
                .all()
            ]

        return [
            eff_schema.from_orm(x)
            for x in session.query(self.sql_model)
            .join(RecipeSettings)
            .filter(RecipeSettings.public == True)  # noqa: 711
            .offset(start)
            .limit(limit)
            .all()
        ]

    def update_image(self, session: Session, slug: str, _: str = None) -> str:
        entry: RecipeModel = self._query_one(session, match_value=slug)
        entry.image = randint(0, 255)
        session.commit()

        return entry.image

    def count_uncategorized(self, session: Session, count=True, override_schema=None) -> int:
        return self._count_attribute(
            session,
            attribute_name=RecipeModel.recipe_category,
            attr_match=None,
            count=count,
            override_schema=override_schema,
        )

    def count_untagged(self, session: Session, count=True, override_schema=None) -> int:
        return self._count_attribute(
            session, attribute_name=RecipeModel.tags, attr_match=None, count=count, override_schema=override_schema
        )