Merge pull request PokemonGoF#296 from Klohto/dev

Interface: Exit message on Keyboard Interrupt
lhung · Jul 22, 2016 · 69f1356 · 69f1356
2 parents d933c30 + 44024ff
commit 69f1356
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/pokecli.py b/pokecli.py
@@ -102,13 +102,18 @@ def main():
     print_green('[x] PokemonGO Bot v1.0')
     print_yellow('[x] Configuration initialized')
 
-    bot = PokemonGoBot(config)
-    bot.start()
+    try:
+        bot = PokemonGoBot(config)
+        bot.start()
 
-    print_green('[x] Starting PokemonGo Bot....')
+        print_green('[x] Starting PokemonGo Bot....')
 
-    while(True):
-        bot.take_step()
+        while(True):
+            bot.take_step()
+
+    except KeyboardInterrupt:
+        print_red("\n"'[x]Exiting PokemonGo Bot')
+        #TODO Add number of pokemon catched, pokestops visited, highest CP pokemon catched, etc.
 
 if __name__ == '__main__':
     main()