Login table check (#4277)

check to see if login table exists before attempting record insert
PokemonGoF · Aug 19, 2016 · f002251 · f002251
1 parent 7293896
commit f002251
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/pokemongo_bot/__init__.py b/pokemongo_bot/__init__.py
@@ -715,7 +715,23 @@ def login(self):
             time.sleep(10)
 
         with self.database as conn:
-            conn.execute('''INSERT INTO login (timestamp, message) VALUES (?, ?)''', (time.time(), 'LOGIN_SUCCESS'))
+            c = conn.cursor()
+            c.execute("SELECT COUNT(name) FROM sqlite_master WHERE type='table' AND name='login'")
+
+        result = c.fetchone()        
+
+        while True:
+            if result[0] == 1:
+                conn.execute('''INSERT INTO login (timestamp, message) VALUES (?, ?)''', (time.time(), 'LOGIN_SUCCESS'))
+                break
+            else:
+                self.event_manager.emit(
+                    'login_failed',
+                    sender=self,
+                    level='info',
+                    formatted="Login table not founded, skipping log"
+                )
+                break
 
         self.event_manager.emit(
             'login_successful',