blob: e7a0e9a505a9da33ebf78ce7d3e8b0f4f3aeae71 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
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
|
