diff --git a/database/filters_db.py b/database/filters_db.py index f7275a0f..6688012b 100644 --- a/database/filters_db.py +++ b/database/filters_db.py @@ -1,4 +1,5 @@ from database import dbname +from typing import Dict, Union filtersdb = dbname.filters diff --git a/database/warn_db.py b/database/warn_db.py index 953f7779..06ce9086 100644 --- a/database/warn_db.py +++ b/database/warn_db.py @@ -31,10 +31,9 @@ async def add_warn(chat_id: int, name: str, warn: dict): warns = await get_warns(chat_id) warns[name] = warn - await warnsdb.update_one({"chat_id": chat_id}, {"$set": { - "warns": warns - }}, - upsert=True) + await warnsdb.update_one( + {"chat_id": chat_id}, {"$set": {"warns": warns}}, upsert=True + ) async def remove_warns(chat_id: int, name: str) -> bool: @@ -44,9 +43,7 @@ async def remove_warns(chat_id: int, name: str) -> bool: del warnsd[name] await warnsdb.update_one( {"chat_id": chat_id}, - {"$set": { - "warns": warnsd - }}, + {"$set": {"warns": warnsd}}, upsert=True, ) return True