From 5fe5c0a7554b706106b65764afd88176762f08a8 Mon Sep 17 00:00:00 2001 From: Jeff Fredrickson Date: Mon, 10 Apr 2023 12:11:53 -0600 Subject: [PATCH] refactor to newer syntax --- training/repositories/base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/training/repositories/base.py b/training/repositories/base.py index 7e8c49c5..33753f10 100644 --- a/training/repositories/base.py +++ b/training/repositories/base.py @@ -1,4 +1,4 @@ -from typing import Any, Generic, Optional, List, Type, TypeVar +from typing import Any, Generic, Type, TypeVar from sqlalchemy.orm import Session from training import models @@ -18,10 +18,10 @@ def save(self, item: T) -> T: self._session.refresh(item) return item - def find_by_id(self, id: int) -> Optional[T]: + def find_by_id(self, id: int) -> T | None: return self._session.query(self._model).filter_by(id=id).first() - def find_all(self, filters: dict[str, Any] = {}) -> List[T]: + def find_all(self, filters: dict[str, Any] = {}) -> list[T]: query = self._session.query(self._model) for key, value in filters.items(): query = query.filter(getattr(self._model, key) == value)