diff --git a/src/gallia/cli.py b/src/gallia/cli.py index 1c2f436c9..13a0bcb76 100644 --- a/src/gallia/cli.py +++ b/src/gallia/cli.py @@ -290,6 +290,7 @@ def cmd_template(args: argparse.Namespace) -> None: # power_supply = # power_cycle = # power_cycle_sleep = +# power_cycle_sleep_after = # dumpcap = # artifacts_dir = # artifacts_base = diff --git a/src/gallia/command/base.py b/src/gallia/command/base.py index a0e4def2b..9a3052ae4 100644 --- a/src/gallia/command/base.py +++ b/src/gallia/command/base.py @@ -532,7 +532,8 @@ async def setup(self, args: Namespace) -> None: self.power_supply = await PowerSupply.connect(args.power_supply) if args.power_cycle is True: await self.power_supply.power_cycle( - args.power_cycle_sleep, lambda: asyncio.sleep(2) + args.power_cycle_sleep, + args.power_cycle_sleep_after, ) elif args.power_cycle is True: self.parser.error("--power-cycle needs --power-supply") @@ -598,6 +599,15 @@ def configure_class_parser(self) -> None: metavar="SECs", type=float, default=self.config.get_value("gallia.scanner.power_cycle_sleep", 5.0), + help="time to sleep after the power-off", + ) + group.add_argument( + "--power-cycle-sleep-after", + metavar="SECs", + type=float, + default=self.config.get_value( + "gallia.scanner.power_cycle_sleep_after", 2.0 + ), help="time to sleep after the power-cycle", )