Fix otaceni fotek, drobny refaktoring

This commit is contained in:
Michal Kváček
2023-10-15 14:24:57 +02:00
parent bc258efa3a
commit 322e61bce3
8 changed files with 25 additions and 30 deletions
+7 -2
View File
@@ -1,5 +1,5 @@
from typing import Optional, Type
from sqlalchemy import select
from sqlalchemy import select, or_
from database import models
@@ -23,7 +23,12 @@ class QueryBuilder:
query = query.filter(self.model.deleted.is_(False))
if hasattr(self.model, "created_by_id") and created_by_id:
query = query.filter(self.model.created_by_id == created_by_id)
ownership = [self.model.created_by_id == created_by_id]
if hasattr(self.model, "is_public"):
ownership.append(self.model.is_public.is_(True))
query = query.filter(or_(*ownership))
if order_by:
query = query.order_by(*order_by)