Mods

cromulant/ants.py

@@ -6,6 +6,7 @@ import itertools
 from typing import ClassVar, Any
 
 from .config import Config
+from .args import Args
 from .utils import Utils
 from .storage import Storage
 
@@ -180,7 +181,10 @@ class Ants:
 
     @staticmethod
     def get() -> None:
-        objs = Storage.get_ants()
+        if Args.clean:
+            objs = []
+        else:
+            objs = Storage.get_ants()
 
         for obj in objs:
             ant = Ant()

cromulant/args.py

@@ -20,6 +20,7 @@ class Args:
     height: int = 0
     program: str = ""
     speed: str = ""
+    clean: bool = False
 
     @staticmethod
     def prepare() -> None:
@@ -45,6 +46,7 @@ class Args:
             "height",
             "program",
             "speed",
+            "clean",
         ]
 
         for n_item in normals:

cromulant/argspec.py

@@ -136,3 +136,9 @@ class ArgSpec:
             choices=["fast", "normal", "slow"],
             info="Use this update speed",
         )
+
+        ArgSpec.add_argument(
+            "clean",
+            action="store_true",
+            info="Start with clean ants data",
+        )