Complete conversion from text-file database to sqlite database

Author: sdliriano

bot.py

@@ -3,6 +3,7 @@
 from exchange_data import *
 from telegram.ext import CommandHandler
 from telegram.ext import Updater
+from dbhelper import DBHelper
 
 updater = Updater(token="INSERT TOKEN HERE", use_context=True)
 
@@ -13,6 +14,7 @@
 logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                      level=logging.INFO)
 
+
 instructions = ("This is the Cryptocurrency Price Alert Bot\n" +
 "To set a price alert,use the setalert command and enter an exchange, trading pair and price as follows:\n" + 
 "/setalert binance,btcusdc,20000\n" + "To get the current price of a coin on a supported exchange use the getprice command:\n"+
@@ -22,7 +24,7 @@
 def help(update, context):
     context.bot.send_message(chat_id=update.message.chat_id, text=instructions)
 
-# /getprice command bot can fetch the price on any supported exchange
+# /getprice command, bot can fetch the price on any supported exchange
 def getprice(update, context):
     message_text = update.message.text[10::]
     msg = message_text.replace(" ","").split(",")
@@ -32,7 +34,6 @@ def getprice(update, context):
     else:
         context.bot.send_message(chat_id=update.message.chat_id, text="The current price of " + msg[1].upper() + " on " + msg[0] + " is " + "{:.8f}".format(float(ans)))
 
-
 # /setalert command
 def setalert(update, context):
     try:
@@ -41,44 +42,49 @@ def setalert(update, context):
             context.bot.send_message(chat_id=update.message.chat_id, text="Your Price Alert Was Not Set Successfully. Choose a valid price then try again.")
         else:
             context.bot.send_message(chat_id=update.message.chat_id, text="Your Price Alert Was Set Successfully.")
-    except:
+    except Exception as e:
         context.bot.send_message(chat_id=update.message.chat_id, text="Your Price Alert Was NOT Set Successfully. Try Again.")
+        print(e)
 
 # checks if the alert request is valid
 def is_valid_request(msg,chat_id):
+    db = DBHelper()
+    db.setup()
     # remove whitespace and split the chosen exchange,trading pair and price into a list
     msg = msg.replace(" ","").split(",")
-    cur_price = get_price(msg[0].lower(),msg[1].lower())
-    # if request is valid, add to 'database'
+    cur_price = get_price(msg[0],msg[1])
+    # if request is valid, add to database
     if cur_price != False:
         if cur_price < float(msg[2]):
             trigger = '>'
         elif cur_price > float(msg[2]):
             trigger = '<'
         else:
             return False
-        f = open("alert_requests.txt", "a")
-        f.write(','.join(msg)+","+str(chat_id)+","+trigger+'\n')
-        f.close()
+        row_id = len(db.get_table()) + 1
+        db.add_alert((row_id, msg[0], msg[1], msg[2], chat_id, trigger))
     else: 
         return False
 
 # bot continuously checks if an alert has been reached
 def check_prices(context):
-    f = open("alert_requests.txt", "r")
-    lines = f.readlines()
-    f.close()
-    f = open("alert_requests.txt", "w")
-    for line in lines:
-        line = line.split(",")
-        cur_price = get_price(line[0].lower(),line[1].lower())
-        if line[4][0] == '>'  and cur_price >= float(line[2]) or line[4][0] == '<' and cur_price <= float(line[2]):
-            response = line[1].upper() + " has reached " + line[2] + " on " + line[0]
-            context.bot.send_message(chat_id = int(line[3]), text = response )
-        else:
-            f.write(",".join(line))
+    db = DBHelper()
+    db.setup()
 
+    # Sample Row in Database: (ID, Exchange, Pair, Price, chat_id, trigger)
+    table = db.get_table()
+    for row in table:
+        cur_price = get_price(row[1],row[2])
+        if row[5] == '>' and cur_price >= float(row[3]) or row[5] == '<' and cur_price <= float(row[3]):
+            response = row[2].upper() + " has reached " + row[3] + " on " + row[1]
+            db.remove_alert(row[0])
+            context.bot.send_message(chat_id = row[4], text = response )
+        
 if __name__ == "__main__":
+    #initialize database
+    db = DBHelper()
+    db.setup()
+
     # bot checks prices with an interval of every 5 seconds    
     job_check_prices = j.run_repeating(check_prices, interval= 5, first =0)
     # create and add handlers to dispatcher

dbhelper.py

@@ -2,7 +2,7 @@
 
 class DBHelper:
 
-    def __init(self, dbname="alerts.db"):
+    def __init__(self, dbname="alerts.db"):
         # sets database name and establishes a connection to it
         self.dbname = dbname
         self.conn = sqlite3.connect(dbname)
@@ -21,4 +21,12 @@ def add_alert(self, request_tuple):
 
     def remove_alert(self, id):
         remove = "DELETE FROM requests WHERE id = (?)"
-        self.conn.execute()
+        id_tuple = (id,)
+        self.conn.execute(remove,id_tuple)
+        self.conn.commit()
+
+    def get_table(self):
+        table = []
+        for row in self.conn.execute('SELECT * FROM requests'):
+            table.append(row)
+        return table