telegram troubleshooting and fine-tuning

active_bots/telegrambot.py

@@ -13,24 +13,24 @@ class TelegramBot(Bot):
         updates = tb.get_updates().wait()
         reports = []
         for update in updates:
-            if update.message.text.lower() == "/start":
+            try:
-                user.add_telegram_subscribers(update.message.sender.id)
+                if update.message.text.lower() == "/start":
-                tb.send_message(update.message.sender.id, "You are now \
+                    user.add_telegram_subscribers(update.message.sender.id)
-                        subscribed to report notifications.")
+                    tb.send_message(update.message.sender.id, "You are now subscribed to report notifications.")
-                #TODO: /start message should be set in frontend
+                    # TODO: /start message should be set in frontend
-            elif update.message.text.lower() == "/stop":
+                elif update.message.text.lower() == "/stop":
-                user.remove_telegram_subscribers(update.message.sender.id)
+                    user.remove_telegram_subscribers(update.message.sender.id)
-                tb.send_message(update.message.sender.id, "You are now \
+                    tb.send_message(update.message.sender.id, "You are now unsubscribed from report notifications.")
-                        unsubscribed from report notifications.")
+                    # TODO: /stop message should be set in frontend
-                #TODO: /stop message should be set in frontend
+                elif update.message.text.lower() == "/help":
-            elif update.message.text.lower() == "/help":
+                    tb.send_message(update.message.sender.id, "Send reports here to share them with other users. Use /start and /stop to be included/excluded.")
-                tb.send_message(update.message.sender.id, "Send reports here to \
+                    # TODO: /help message should be set in frontend
-                        share them with other users. Use /start and /stop to \
+                else:
-                        be included/excluded.")
+                    reports.append(Report(update.message.sender.username, self,
-                #TODO: /help message should be set in frontend
+                                   update.message.text, None, update.message.date))
-            else:
+            except AttributeError:
-                reports.append(Report(update.message.sender.username, self,
+                print(updates[0], updates[1])  # Telegram API returns an Error
-                               update.message.text, None, update.message.date))
+                return reports
         return reports
 
     def post(self, user, report):

db.py

@@ -126,7 +126,6 @@ class DB(object):
                 id          INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT UNIQUE,
                 user_id     INTEGER,
                 subscriber_id   INTEGER,
-                active      INTEGER,
                 FOREIGN KEY(user_id) REFERENCES user(id),
                 UNIQUE(user_id, subscriber_id) ON CONFLICT IGNORE
             );

frontend.py

@@ -160,11 +160,10 @@ def update_badwords(user):
 
 
 @post('/settings/telegram')
-@view('template/settings.tpl')
 def register_telegram(user):
     apikey = request.forms['apikey']
-    user.set_telegram_key(apikey)
+    user.update_telegram_key(apikey)
-    return user.state()
+    return city_page(user.get_city(), info="Thanks for registering Telegram!")
 
 
 @get('/api/state')

user.py

@@ -94,20 +94,13 @@ schlitz
         return True
 
     def get_telegram_credentials(self):
-        db.execute("""SELECT api_token 
+        db.execute("""SELECT apikey 
                           FROM telegram_accounts 
                           WHERE user_id = ? AND active = 1;""",
                    (self.uid,))
         row = db.cur.fetchone()
         return row[0]
 
-    def save_telegram_credentials(self, api_token):
-        db.execute("""INSERT INTO telegram_accounts (
-                          user_id, api_token, active) VALUES(?, ?, 1);""",
-                   (self.uid, api_token))
-        db.commit()
-
-
     def get_telegram_subscribers(self):
         db.execute("""SELECT subscriber_id 
                           FROM telegram_subscribers 
@@ -272,7 +265,7 @@ schlitz
                    (self.uid, ))
         return db.cur.fetchall()
 
-    def set_telegram_key(self, apikey):
+    def update_telegram_key(self, apikey):
         db.execute("UPDATE telegram_accounts SET apikey = ? WHERE user_id = ?;", (apikey, self.uid))
         db.commit()