diff options
Diffstat (limited to 'jb/managers/bonus.py')
| -rw-r--r-- | jb/managers/bonus.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/jb/managers/bonus.py b/jb/managers/bonus.py index 0cb8b02..89b81f0 100644 --- a/jb/managers/bonus.py +++ b/jb/managers/bonus.py @@ -1,4 +1,4 @@ -from typing import List +from typing import List, Any from psycopg import sql @@ -37,12 +37,12 @@ class BonusManager(PostgresManager): c.execute(query, data) res = c.fetchone() conn.commit() - bonus.id = res["id"] - bonus.assignment_id = res["assignment_id"] + bonus.id = res["id"] # type: ignore + bonus.assignment_id = res["assignment_id"] # type: ignore return None def filter(self, amt_assignment_id: str) -> List[Bonus]: - res = self.pg_config.execute_sql_query( + res: List[Any] = self.pg_config.execute_sql_query( """ SELECT mb.*, ma.amt_assignment_id FROM mtwerk_bonus mb @@ -51,4 +51,5 @@ class BonusManager(PostgresManager): """, params={"amt_assignment_id": amt_assignment_id}, ) - return [Bonus.from_postgres(x) for x in res] + + return [Bonus.from_postgres(data=x) for x in res] |