This commit is contained in:
Auric Vente 2024-07-25 21:29:18 -06:00
parent c79c61f79b
commit 6c09901496
3 changed files with 13 additions and 1 deletions

View File

@ -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()

View File

@ -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:

View File

@ -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",
)