src/blunderboard/__init__.py

@@ -1,20 +1,24 @@
 from blunderboard.blunderevaluator import BlunderEvaluator
 from blunderboard.boardreader import BoardReader
 from blunderboard.movegenerator import MoveGenerator
-from time import sleep
 import cProfile
+from pstats import SortKey
+from time import sleep
 
 
 def main_content():
-    blunder_evaluator = BlunderEvaluator()
-    move_generator = MoveGenerator(blunder_evaluator)
-    reader = BoardReader(move_generator)
-    reader.scan()
-    reader.print()
-    while True:
+    try:
+        blunder_evaluator = BlunderEvaluator()
+        move_generator = MoveGenerator(blunder_evaluator)
+        reader = BoardReader(move_generator)
         reader.scan()
-        sleep(0.1)
+        reader.print()
+        while True:
+            reader.scan()
+            sleep(0.1)
+    except KeyboardInterrupt:
+        pass
 
 
 def main():
-    cProfile.runctx("main_content()", globals(), locals(), filename=__file__)
+    cProfile.runctx("main_content()", globals(), locals(), sort=SortKey.CUMULATIVE)