aboutsummaryrefslogtreecommitdiff
path: root/jb/managers/bonus.py
diff options
context:
space:
mode:
authorMax Nanis2026-09-09 20:32:17 -0700
committerMax Nanis2026-09-09 20:32:17 -0700
commit832aecaddce80e312095ecdb572d7756eb9df5e9 (patch)
tree3f9b27bf285ef0648e22a1480ab213255d2cfee6 /jb/managers/bonus.py
parente7e5b5f8bf81e78e417ecdb9dd6ab1616e722fa0 (diff)
downloadamt-jb-832aecaddce80e312095ecdb572d7756eb9df5e9.tar.gz
amt-jb-832aecaddce80e312095ecdb572d7756eb9df5e9.zip
Ruff auto fix, and requirements bump
Diffstat (limited to 'jb/managers/bonus.py')
-rw-r--r--jb/managers/bonus.py22
1 files changed, 9 insertions, 13 deletions
diff --git a/jb/managers/bonus.py b/jb/managers/bonus.py
index 89b81f0..15d0e5b 100644
--- a/jb/managers/bonus.py
+++ b/jb/managers/bonus.py
@@ -1,4 +1,4 @@
-from typing import List, Any
+from typing import Any
from psycopg import sql
@@ -11,8 +11,7 @@ class BonusManager(PostgresManager):
def create(self, bonus: Bonus) -> None:
assert bonus.id is None
data = bonus.to_postgres()
- query = sql.SQL(
- """
+ query = sql.SQL("""
INSERT INTO mtwerk_bonus
(payout_event_id, amt_worker_id, amount, grant_time, assignment_id, reason)
VALUES (
@@ -29,20 +28,17 @@ class BonusManager(PostgresManager):
%(reason)s
)
RETURNING id, assignment_id;
- """
- )
+ """)
- with self.pg_config.make_connection() as conn:
- with conn.cursor() as c:
- c.execute(query, data)
- res = c.fetchone()
- conn.commit()
+ with self.pg_config.make_connection() as conn, conn.cursor() as c:
+ c.execute(query, data)
+ res = c.fetchone()
+ conn.commit()
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: List[Any] = self.pg_config.execute_sql_query(
+ def filter(self, amt_assignment_id: str) -> list[Bonus]:
+ res: list[Any] = self.pg_config.execute_sql_query(
"""
SELECT mb.*, ma.amt_assignment_id
FROM mtwerk_bonus mb