ED/SquadsActivityMonitor
1
0
Fork 0
mirror of https://github.com/norohind/SquadsActivityMonitor.git synced 2025-04-13 13:27:13 +03:00
Code Issues Projects Releases Wiki Activity

remove context manager self.db for read request

Browse Source
This commit is contained in:
norohind 2021-12-09 01:00:07 +03:00
parent f58df5e45f
commit 0c64bde728
Signed by: norohind
GPG Key ID: 01C3BECC26FB59E1
1 changed files with 12 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
model/postgres_model.py
View File

@ -63,17 +63,16 @@ class PostgresModel(AbstractModel):
logger.debug(f'Not cached result for {cache_key}')
with self.db:
with self.db.cursor(cursor_factory=psycopg2.extras.RealDictCursor) as cursor:
cursor.execute(postgres_sql_requests.select_activity_pretty_names, {
'LB_type': utils.LeaderboardTypes(leaderboard_type.lower()).value,
'platform': utils.Platform(platform.upper()).value,
'limit': limit,
'high_timestamp': high_timestamp,
'low_timestamp': low_timestamp
})
with self.db.cursor(cursor_factory=psycopg2.extras.RealDictCursor) as cursor:
cursor.execute(postgres_sql_requests.select_activity_pretty_names, {
'LB_type': utils.LeaderboardTypes(leaderboard_type.lower()).value,
'platform': utils.Platform(platform.upper()).value,
'limit': limit,
'high_timestamp': high_timestamp,
'low_timestamp': low_timestamp
})
result: list = cursor.fetchall()
result: list = cursor.fetchall()
if not cache.disabled:
cache.set(cache_key, json.dumps(result))
@ -136,10 +135,9 @@ class PostgresModel(AbstractModel):
logger.debug(f'Not cached result for {cache_key}')
with self.db:
with self.db.cursor(cursor_factory=psycopg2.extras.RealDictCursor) as cursor:
cursor.execute(postgres_sql_requests.select_diff_by_action_id, {'action_id': action_id})
result: list = cursor.fetchall()
with self.db.cursor(cursor_factory=psycopg2.extras.RealDictCursor) as cursor:
cursor.execute(postgres_sql_requests.select_diff_by_action_id, {'action_id': action_id})
result: list = cursor.fetchall()
if not cache.disabled:
cache.set(cache_key, json.dumps(result))