add admin mode

This commit is contained in:
kandrusyak
2023-08-02 22:06:07 +03:00
parent e83616e4c0
commit f88c5a27ea
2 changed files with 8 additions and 6 deletions

View File

@@ -11,9 +11,11 @@ class PersonService(BaseService[Person, PersonCreate, Any]):
def __init__(self, db_session: Session):
super(PersonService, self).__init__(Person, db_session)
def list(self, full_name: str) -> List[Person]:
#objs: List[Person] = self.db_session.query(Person).all()
objs: List[Person] = self.db_session.query(Person).filter(Person.last_name.ilike(full_name)).all()
def list(self, full_name: str, role: str) -> List[Person]:
if role == "admin":
objs: List[Person] = self.db_session.query(Person).all()
else:
objs: List[Person] = self.db_session.query(Person).filter(Person.last_name.ilike(full_name)).all()
return objs
def create(self, obj: PersonCreate) -> Person: