fixes #7 - add addpiece commandline argument

Author: haklein

certabo-lichess.py

@@ -24,6 +24,7 @@
 parser = argparse.ArgumentParser()
 parser.add_argument("--port")
 parser.add_argument("--calibrate", action="store_true")
+parser.add_argument("--addpiece", action="store_true")
 parser.add_argument("--devmode", action="store_true")
 parser.add_argument("--quiet", action="store_true")
 parser.add_argument("--debug", action="store_true")
@@ -33,9 +34,12 @@
 if args.port is not None:
     portname = args.port
 
-calibrate = False
+calibrate = 0 # don't do calibration by default
 if args.calibrate:
-    calibrate = True
+    calibrate = 2 # do fresh calibration
+
+if args.addpiece:
+    calibrate = 1 # add further pieces to existing calibration
 
 DEBUG=False
 if args.debug:

certabo/certabo.py

@@ -29,11 +29,17 @@
 os.makedirs(CERTABO_DATA_PATH, exist_ok=True)
 
 class Certabo():
-    def __init__(self, port='auto', calibrate=False, **kwargs):
+    def __init__(self, port='auto', calibrate=0, **kwargs):
         super().__init__(**kwargs)
         self.portname = port
-        self.calibration = calibrate
-        self.new_setup = False
+        if calibrate:
+            self.calibration = True
+        else:
+            self.calibration = False
+        if calibrate > 1:
+            self.new_setup = True
+        else:
+            self.new_setup = False
         self.rotate180 = False
         self.color = chess.WHITE
         self.starting_position = chess.STARTING_FEN