github/mealie
1
0
Fork 0
mirror of https://github.com/hay-kot/mealie.git synced 2025-08-22 14:33:33 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 20

abstract count function

Browse source
This commit is contained in:
hay-kot 2021-05-05 13:57:14 -08:00
commit a6a24a5aee
1 changed files with 6 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

22
mealie/db/database.py
View file

@ -37,24 +37,14 @@ class _Recipes(BaseDocument):
return f"{slug}.{extension}"
def count_uncategorized(self, session: Session, count=True, override_schema=None) -> int:
eff_schema = override_schema or self.schema
if count:
return session.query(self.sql_model).filter(RecipeModel.recipe_category == None).count() # noqa: 711
else:
return [
eff_schema.from_orm(x)
for x in session.query(self.sql_model).filter(RecipeModel.tags == None).all() # noqa: 711
]
return self._countr_attribute(
session, attribute_name=RecipeModel.recipe_category, attr_match=None, count=True, override_schema=None
)
def count_untagged(self, session: Session, count=True, override_schema=None) -> int:
eff_schema = override_schema or self.schema
if count:
return session.query(self.sql_model).filter(RecipeModel.tags == None).count() # noqa: 711
else:
return [
eff_schema.from_orm(x)
for x in session.query(self.sql_model).filter(RecipeModel.tags == None).all() # noqa: 711
]
return self._countr_attribute(
session, attribute_name=RecipeModel.tags, attr_match=None, count=True, override_schema=None
)
class _Categories(BaseDocument):