diff --git a/zyrarbot/storage/adapters/postgres/database.py b/zyrarbot/storage/adapters/postgres/database.py index 390d4eb..0014453 100644 --- a/zyrarbot/storage/adapters/postgres/database.py +++ b/zyrarbot/storage/adapters/postgres/database.py @@ -2,9 +2,10 @@ from __future__ import annotations import json import logging +from collections.abc import Iterator from contextlib import contextmanager from pathlib import Path -from typing import Any, Iterator +from typing import Any import psycopg @@ -71,7 +72,11 @@ class PostgresDatabase: for key, entries in history.items(): if key.isdigit() and isinstance(entries, list): _import_history(connection, int(key), entries) - elif key == "history" and isinstance(entries, list) and first_guild_id is not None: + elif ( + key == "history" + and isinstance(entries, list) + and first_guild_id is not None + ): _import_history(connection, first_guild_id, entries) for path in (base / "queues").glob("*.json"): @@ -109,7 +114,12 @@ def _upsert_guild( try: value = int(values[field]) except (TypeError, ValueError): - logger.error("Nieprawidłowa wartość %s serwera %s: %r", field, guild_id, values[field]) + logger.error( + "Nieprawidłowa wartość %s serwera %s: %r", + field, + guild_id, + values[field], + ) continue connection.execute(f"UPDATE guilds SET {field}=%s WHERE id=%s", (value, guild_id)) diff --git a/zyrarbot/storage/adapters/postgres/guild_config.py b/zyrarbot/storage/adapters/postgres/guild_config.py index dd82b50..e7acc15 100644 --- a/zyrarbot/storage/adapters/postgres/guild_config.py +++ b/zyrarbot/storage/adapters/postgres/guild_config.py @@ -12,7 +12,9 @@ class PostgresGuildConfig: def _ensure(self, guild_id: int) -> None: with self._database.connect() as connection: - connection.execute("INSERT INTO guilds(id) VALUES (%s) ON CONFLICT(id) DO NOTHING", (guild_id,)) + connection.execute( + "INSERT INTO guilds(id) VALUES (%s) ON CONFLICT(id) DO NOTHING", (guild_id,) + ) def get_notification_channel_id(self, guild_id: int) -> int | None: with self._database.connect() as connection: @@ -32,10 +34,14 @@ class PostgresGuildConfig: if guild_id is None: return self._fallback_plays with self._database.connect() as connection: - row = connection.execute("SELECT default_plays FROM guilds WHERE id=%s", (guild_id,)).fetchone() + row = connection.execute( + "SELECT default_plays FROM guilds WHERE id=%s", (guild_id,) + ).fetchone() return int(row[0]) if row and row[0] is not None else self._fallback_plays def set_default_plays(self, guild_id: int, value: int) -> None: self._ensure(guild_id) with self._database.connect() as connection: - connection.execute("UPDATE guilds SET default_plays=%s WHERE id=%s", (int(value), guild_id)) + connection.execute( + "UPDATE guilds SET default_plays=%s WHERE id=%s", (int(value), guild_id) + )