added multi parameter update (#47)

Author: ZegerUser

pyvts/tests/test_utils.py

@@ -72,21 +72,22 @@ async def handler(self, websocket):
                     del self.custom_param[pname]
                     send_msg["data"] = {"parameter": pname, "message": "Deleted"}
             elif dict_msg["messageType"] == "InjectParameterDataRequest":
-                pname = dict_msg["data"]["parameterValues"][0]["id"]
-                if pname in self.custom_param:
-                    self.custom_param[pname]["value"] = float(
-                        dict_msg["data"]["parameterValues"][0]["value"]
-                    ) * float(dict_msg["data"]["parameterValues"][0]["weight"]) + float(
-                        self.custom_param[pname]["value"]
-                    ) * float(
-                        (1 - dict_msg["data"]["parameterValues"][0]["weight"])
-                    )
-                    send_msg["data"] = {"parameter": pname, "message": "set"}
-                else:
-                    send_msg["data"] = {
-                        "errorID": 500,
-                        "message": "Given parameter start parameter does not exist.",
-                    }
+                for pdata in dict_msg["data"]["parameterValues"]:
+                    pname = pdata["id"]
+                    if pname in self.custom_param:
+                        self.custom_param[pname]["value"] = float(
+                            pdata["value"]
+                        ) * float(pdata["weight"]) + float(
+                            self.custom_param[pname]["value"]
+                        ) * float(
+                            (1 - pdata["weight"])
+                        )
+                        send_msg["data"] = {"parameter": pname, "message": "set"}
+                    else:
+                        send_msg["data"] = {
+                            "errorID": 500,
+                            "message": "Given parameter start parameter does not exist.",
+                        }
 
             # About event subscribe
             elif dict_msg["messageType"] == "EventSubscriptionRequest":

pyvts/tests/test_vts.py

@@ -108,6 +108,43 @@ async def test_vts_custon_parameter(myvts: pyvts.vts):
     await fake_server.stop()
 
 
+@pytest.mark.asyncio
+async def test_vts_custon_parameter_multi(myvts: pyvts.vts):
+    """test vts functions for multi parameter update"""
+    fake_server = FakeVtubeStudioAPIServer()
+    await fake_server.start(port=PORT)
+    await myvts.connect()
+    param_names = ["test-1", "test-2"]
+    vals = [0.5, 0.5]
+    # create custom parameters
+    for name in param_names:
+        await myvts.request(
+            myvts.vts_request.requestCustomParameter(name, default_value=0)
+        )
+        custom_param_value = await myvts.request(
+            myvts.vts_request.requestParameterValue(name)
+        )
+        assert custom_param_value["data"]["value"] == 0
+    # set value for these parameters
+    await myvts.request(
+        myvts.vts_request.requestSetMultiParameterValue(param_names, vals)
+    )
+
+    for name in param_names:
+        custom_param_value = await myvts.request(
+            myvts.vts_request.requestParameterValue(name)
+        )
+        assert custom_param_value["data"]["value"] == 0.5
+        # delete parameter
+        await myvts.request(myvts.vts_request.requestDeleteCustomParameter(name))
+        custom_param_value = await myvts.request(
+            myvts.vts_request.requestParameterValue(name)
+        )
+        assert custom_param_value["data"]["errorID"] == 500
+    await myvts.close()
+    await fake_server.stop()
+
+
 @pytest.mark.asyncio
 async def test_vts_event_subscription(myvts: pyvts.vts):
     """test vts functions about event subscribe"""

pyvts/vts_request.py

@@ -243,6 +243,43 @@ def requestSetParameterValue(
         }
         return self.BaseRequest("InjectParameterDataRequest", data=data)
 
+    def requestSetMultiParameterValue(
+        self,
+        parameters: list[str],
+        values: list[float],
+        weight: float = 1,
+        face_found=False,
+        mode="set",
+    ) -> dict:
+        """
+        Set values for any default or custom parameters.
+
+        Args
+        ----------
+        parameter : list[str]
+            Names of the parameters
+        value : list[float]
+            Values of the data, [-1000000, 1000000]
+        weight : float, optional
+            You can mix the your value with vts face tracking parameter, from 0 to 1,
+        face_found : bool, optional
+            if true, you will tell VTubeStudio to consider the user face as found,
+            s.t. you can control when the "tracking lost"
+
+        Returns
+        -------
+            organized message sending to ``Vtubestudio API``
+        """
+        data = {
+            "faceFound": face_found,
+            "mode": mode,
+            "parameterValues": [
+                {"id": parameters[i], "weight": weight, "value": values[i]}
+                for i, _ in enumerate(parameters)
+            ],
+        }
+        return self.BaseRequest("InjectParameterDataRequest", data=data)
+
     def requestDeleteCustomParameter(self, parameter) -> dict:
         """
         Delete custom parameter