diff --git a/modules/upl.py b/modules/upl.py index fb6f9cd..8afba56 100644 --- a/modules/upl.py +++ b/modules/upl.py @@ -34,6 +34,12 @@ async def upl(client: Client, message: Message): await message.edit(format_exc(e)) +@Client.on_message(filters.command("dlf", prefix) & filters.me) +async def dlf(client: Client, message: Message): + if message.reply_to_message.document: + await client.download_media(message.reply_to_message) + + @Client.on_message(filters.command("moonlogs", prefix) & filters.me) async def mupl(client: Client, message: Message): link = "moonlogs.txt" diff --git a/utils/db.py b/utils/db.py index 0a7f986..2cac220 100644 --- a/utils/db.py +++ b/utils/db.py @@ -129,7 +129,8 @@ class SqliteDatabase(Database): self._lock.acquire() try: - return self._cursor.execute(*args, **kwargs) + cursor = self._conn.cursor() + return cursor.execute(*args, **kwargs) except sqlite3.OperationalError as e: if str(e).startswith("no such table"): sql = f""" @@ -139,9 +140,10 @@ class SqliteDatabase(Database): type TEXT NOT NULL ) """ - self._cursor.execute(sql) + cursor = self._conn.cursor() + cursor.execute(sql) self._conn.commit() - return self._cursor.execute(*args, **kwargs) + return cursor.execute(*args, **kwargs) raise e from None finally: self._lock.release()