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}"
def count_uncategorized(self, session: Session, count=True, override_schema=None) -> int:
return self._countr_attribute(
return self._count_attribute(
session,
attribute_name=RecipeModel.recipe_category,
attr_match=None,
@ -47,7 +47,7 @@ class _Recipes(BaseDocument):
)
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
)

View file

@ -178,7 +178,7 @@ class BaseDocument:
else:
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
) -> Union[int, BaseModel]:
eff_schema = override_schema or self.schema