mirror of
https://github.com/suaveolent/hoymiles-wifi.git
synced 2024-09-20 03:10:11 +00:00
Merge e211658185
into 255a9109d8
This commit is contained in:
commit
6444872feb
@ -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":
|
||||
@ -351,6 +359,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, args.local_addr)
|
||||
@ -378,6 +395,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:
|
||||
|
Loading…
Reference in New Issue
Block a user