correcting sql_data

kotorkovsciy · Aug 10, 2022 · 36c0af1 · 36c0af1
1 parent 8007591
commit 36c0af1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/sql_data.py b/scripts/sql_data.py
@@ -1,4 +1,5 @@
-from os import getenv
+from os import getenv, mkdir
+from os.path import exists
 from dotenv import load_dotenv
 from psycopg2 import connect
 from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT
@@ -230,6 +231,8 @@ async def deleteJokes(self):
 
     async def dump(self, user_id):
         """Дамп бд"""
+        if not exists("sql/"):
+            mkdir("sql/")
         self.__open__()
         self.cursor.execute('SELECT * FROM users')
         with open(f"sql\dump_users_{user_id}.sql", "w", encoding='utf 8') as file: