diff --git a/aleksis/apps/untis/util/mysql/util.py b/aleksis/apps/untis/util/mysql/util.py index e98ab3157ac456d5cf1093d16d83b575b0065b38..128efa8fc393370b5f48b66b7b9e5c03e221ea94 100644 --- a/aleksis/apps/untis/util/mysql/util.py +++ b/aleksis/apps/untis/util/mysql/util.py @@ -32,7 +32,7 @@ def get_term(date: Optional[date] = None) -> mysql_models.Terms: def run_default_filter( - qs: QuerySet, date: Optional[date] = None, filter_term: bool = True + qs: QuerySet, date: Optional[date] = None, filter_term: bool = True, filter_deleted: bool = True ) -> QuerySet: """ Add a default filter in order to select the correct term """ @@ -44,18 +44,19 @@ def run_default_filter( term.version_id, ) - if filter_term: - return run_using(qs).filter( + qs = run_using(qs).filter( school_id=school_id, schoolyear_id=schoolyear_id, version_id=version_id, - term_id=term_id, - deleted=0 - ) - else: - return run_using(qs).filter( - school_id=school_id, schoolyear_id=schoolyear_id, version_id=version_id, deleted=0 - ) + ) + + if filter_term: + qs = qs.filter(term_id=term_id) + + if filter_deleted: + qs = qs.filter(deleted=0) + + return qs def clean_array(a: list, conv=None) -> list: