import logging import os import sqlite3 from dotenv import load_dotenv load_dotenv() dbFile = os.getenv('DB_FILENAME', 'db.sqlite3') def setupLogger(): logger = logging.getLogger(__name__) logging.basicConfig( filename=os.getenv('LOGFILE', "watcher.log"), encoding='utf-8', level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s' ) logging.getLogger("requests").setLevel(logging.WARNING) logging.getLogger("urllib3").setLevel(logging.WARNING) return logger def initDb(): try: conn = sqlite3.connect(dbFile) cur = conn.cursor() cur.execute("CREATE TABLE incoming (id, msg_date, msg_from, text, is_read, update_id)") conn.commit() conn.close() except sqlite3.Error as e: logging.error("Failed to initialize database") logging.error(e) def getDbConn(): try: conn = sqlite3.connect(dbFile) return conn except sqlite3.Error as e: logging.error("Failed making connection to database") logging.error(e) return None