summaryrefslogtreecommitdiff
path: root/util.py
blob: 0590c49cf827d6186ad4ef51c4267e4a58b7ee57 (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
import logging
import os
import sqlite3

from dotenv import load_dotenv

load_dotenv()

dbFile = 'db.sqlite3'


def setupLogger():
    logger = logging.getLogger(__name__)
    logging.basicConfig(
        filename=os.getenv('LOGFILE'),
        encoding='utf-8',
        level=logging.DEBUG,
        format='%(asctime)s - %(levelname)s - %(message)s'
    )
    return logger


def initDb():
    conn = sqlite3.connect(dbFile)
    cur = conn.cursor()
    cur.execute("CREATE TABLE incoming (id, msg_date, msg_from, text, is_read)")
    conn.commit()
    conn.close()


def getDbConn():
    conn = sqlite3.connect(dbFile)
    return conn