Softban Tracker (#4299)

* Log softbans + source / date to database * Log softbans + source / date to database * Update __init__.py * Update colored_logging_handler.py * Create softban_log.py * re-add new berry stuff
PokemonGoF · Aug 20, 2016 · a5cab06 · a5cab06
1 parent a5f8491
commit a5cab06
Show file tree

Hide file tree

Showing 5 changed files with 44 additions and 0 deletions.
diff --git a/pokemongo_bot/__init__.py b/pokemongo_bot/__init__.py
@@ -517,6 +517,7 @@ def _register_events(self):
         self.event_manager.register_event('login_log')
         self.event_manager.register_event('transfer_log')
         self.event_manager.register_event('pokestop_log')
+        self.event_manager.register_event('softban_log')
 
     def tick(self):
         self.health_record.heartbeat()

diff --git a/pokemongo_bot/cell_workers/migrations/softban_log.py b/pokemongo_bot/cell_workers/migrations/softban_log.py
@@ -0,0 +1,5 @@
+from yoyo import step
+
+step(
+    "CREATE TABLE IF NOT EXISTS softban_log (status text, source text, dated datetime DEFAULT CURRENT_TIMESTAMP)"
+)
diff --git a/pokemongo_bot/cell_workers/pokemon_catch_worker.py b/pokemongo_bot/cell_workers/pokemon_catch_worker.py
@@ -277,6 +277,24 @@ def _use_berry(self, berry_id, berry_count, encounter_id, catch_rate_by_ball, cu
                     level='warning',
                     formatted='Failed to use berry. You may be softbanned.'
                 )
+                with self.bot.database as conn:
+                    c = conn.cursor()
+                    c.execute("SELECT COUNT(name) FROM sqlite_master WHERE type='table' AND name='softban_log'")
+                result = c.fetchone()        
+
+                while True:
+                    if result[0] == 1:
+                        source = str("PokemonCatchWorker")
+                        status = str("Possible Softban")
+                        conn.execute('''INSERT INTO softban_log (status, source) VALUES (?, ?)''', (status, source))
+                    break
+                else:
+                    self.emit_event(
+                        'softban_log',
+                        sender=self,
+                        level='info',
+                        formatted="softban_log table not found, skipping log"
+                    )
 
         # unknown status code
         else:

diff --git a/pokemongo_bot/cell_workers/spin_fort.py b/pokemongo_bot/cell_workers/spin_fort.py
@@ -150,6 +150,25 @@ def work(self):
                         'softban',
                         formatted='Probably got softban.'
                     )
+                with self.bot.database as conn:
+                    c = conn.cursor()
+                    c.execute("SELECT COUNT(name) FROM sqlite_master WHERE type='table' AND name='softban_log'")
+                result = c.fetchone()        
+
+                while True:
+                    if result[0] == 1:
+                        source = str("PokemonCatchWorker")
+                        status = str("Possible Softban")
+                        conn.execute('''INSERT INTO softban_log (status, source) VALUES (?, ?)''', (status, source))
+                        break
+                    else:
+                        self.emit_event(
+                        'softban_log',
+                        sender=self,
+                        level='info',
+                        formatted="softban_log table not found, skipping log"
+                        )
+                        break
                 else:
                     self.bot.fort_timeouts[fort["id"]] = (time.time() + 300) * 1000  # Don't spin for 5m
                 return WorkerResult.ERROR

diff --git a/pokemongo_bot/event_handlers/colored_logging_handler.py b/pokemongo_bot/event_handlers/colored_logging_handler.py
@@ -88,6 +88,7 @@ class ColoredLoggingHandler(EventHandler):
         'position_update':                   'white',
         'set_start_location':                'white',
         'softban_fix':                       'white',
+        'softban_log':                       'magenta',
         'softban_fix_done':                  'white',
         'spun_fort':                         'white',
         'threw_berry':                       'white',