This commit is contained in:
hay-kot 2021-05-08 20:43:02 -08:00
commit 519f59d341
2 changed files with 3 additions and 3 deletions

View file

@ -38,7 +38,7 @@ class _Recipes(BaseDocument):
return f"{slug}.{extension}" return f"{slug}.{extension}"
def count_uncategorized(self, session: Session, count=True, override_schema=None) -> int: def count_uncategorized(self, session: Session, count=True, override_schema=None) -> int:
return self._countr_attribute( return self._count_attribute(
session, session,
attribute_name=RecipeModel.recipe_category, attribute_name=RecipeModel.recipe_category,
attr_match=None, attr_match=None,
@ -47,7 +47,7 @@ class _Recipes(BaseDocument):
) )
def count_untagged(self, session: Session, count=True, override_schema=None) -> int: def count_untagged(self, session: Session, count=True, override_schema=None) -> int:
return self._countr_attribute( return self._count_attribute(
session, attribute_name=RecipeModel.tags, attr_match=None, count=count, override_schema=override_schema session, attribute_name=RecipeModel.tags, attr_match=None, count=count, override_schema=override_schema
) )

View file

@ -178,7 +178,7 @@ class BaseDocument:
else: else:
return session.query(self.sql_model).filter_by(**{match_key: match_value}).count() return session.query(self.sql_model).filter_by(**{match_key: match_value}).count()
def _countr_attribute( def _count_attribute(
self, session: Session, attribute_name: str, attr_match: str = None, count=True, override_schema=None self, session: Session, attribute_name: str, attr_match: str = None, count=True, override_schema=None
) -> Union[int, BaseModel]: ) -> Union[int, BaseModel]:
eff_schema = override_schema or self.schema eff_schema = override_schema or self.schema