From 6142b2d4ab1ad428cd4ef18fb07eff7a8aea06a5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Giovanni=20Lagan=C3=A0?= Date: Sat, 27 May 2023 15:50:34 +0200 Subject: [PATCH] fix connection issue --- db/connection.py | 1 - db/cursor.py | 1 + db/queries.py | 4 +--- main.py | 4 ++++ 4 files changed, 6 insertions(+), 4 deletions(-) create mode 100644 db/cursor.py diff --git a/db/connection.py b/db/connection.py index b13b683..954d8bb 100644 --- a/db/connection.py +++ b/db/connection.py @@ -1,7 +1,6 @@ import psycopg2 from db.config import config -""" Connect to the PostgreSQL database server """ def connect(): conn = None try: diff --git a/db/cursor.py b/db/cursor.py new file mode 100644 index 0000000..6b79b85 --- /dev/null +++ b/db/cursor.py @@ -0,0 +1 @@ +connection = None \ No newline at end of file diff --git a/db/queries.py b/db/queries.py index 45ba3bc..908bac1 100644 --- a/db/queries.py +++ b/db/queries.py @@ -1,9 +1,7 @@ from utils.constants import default_target, default_time, table_name, default_message from psycopg2 import sql from utils.utils import compute_next_wednesday -from db.connection import connect - -connection = connect() +from db.cursor import connection def create_bamboo_chat_id_row(chat_id): pitch = "Crespi Sport Village \n" \ diff --git a/main.py b/main.py index 9a02892..2f7ddbb 100644 --- a/main.py +++ b/main.py @@ -12,12 +12,16 @@ from callbacks.help_func import help_func from callbacks.echo import echo from callbacks.bamboo import bamboo +from db.connection import connect +import db.cursor as db_cursor import logging import os logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO) if __name__ == '__main__': + db_cursor.connection = connect() + updater = Updater(token=token, use_context=True) dispatcher = updater.dispatcher