Merge e211658185 into 18f88d810c

hoymiles_wifi/__main__.py

@@ -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":
@@ -345,6 +353,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)
@@ -372,6 +389,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: