Allow users to set / force the power limit as an argument

This allows us to use the tool in scripts without having to deal with interactive input. In this first change the --force parameter is supported by set-power-limit.

Found no good place for adding it into help text, maybe a "help" command would come in handy ;)
This commit is contained in:
/)/) -- Daniel Haslinger 2024-08-15 19:32:37 +02:00 committed by GitHub
parent f595e64303
commit e211658185
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -108,10 +108,11 @@ async def async_app_get_hist_power(
async def async_set_power_limit(
dtu: DTU,
dtu: DTU, power_limit=-1
) -> CommandPB_pb2.CommandResDTO | None:
"""Set the power limit of the inverter asynchronously."""
if power_limit == -1:
print( # noqa: T201
RED
+ "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n"
@ -127,12 +128,19 @@ async def async_set_power_limit(
return None
power_limit = int(input("Enter the new power limit (0-100): "))
forced_argument = False # we've retrieved the power limit in interactive mode
else:
forced_argument = True # we've retrieved the power limit via argument
if power_limit < 0 or power_limit > MAX_POWER_LIMIT:
print("Error. Invalid power limit!") # noqa: T201
return None
print(f"Setting power limit to {power_limit}%") # noqa: T201
if not forced_argument:
# the limit has been set in interactive mode, so we ask for confirmation
cont = input("Are you sure? (y/n): ")
if cont != "y":
@ -343,6 +351,15 @@ async def main() -> None:
],
help="Command to execute",
)
parser.add_argument(
"--force",
type=int,
nargs='?',
const=-1,
default=-1,
help="Do not confirm certain operations and take arguments directly from commandline",
)
args = parser.parse_args()
dtu = DTU(args.host)
@ -370,6 +387,9 @@ async def main() -> None:
}
command_func = switch.get(args.command, print_invalid_command)
if args.command == 'set-power-limit':
response = await command_func(dtu, args.force)
else:
response = await command_func(dtu)
if response: