1 files changed, 16 insertions, 1 deletions
diff --git a/util.py b/util.py
index 88bff8e..0590c49 100644
--- a/util.py
+++ b/util.py
@@ -1,13 +1,15 @@
 import logging
 import os
+import sqlite3
 
 from dotenv import load_dotenv
 
 load_dotenv()
 
+dbFile = 'db.sqlite3'
+
 
 def setupLogger():
-    # kayanya python logger itu singleton, bisa dipanggil berkali2
     logger = logging.getLogger(__name__)
     logging.basicConfig(
         filename=os.getenv('LOGFILE'),
@@ -16,3 +18,16 @@ def setupLogger():
         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