feat(edge_services): add search waf stage method

scaleway-async/scaleway_async/edge_services/v1beta1/__init__.py

@@ -23,6 +23,7 @@
 from .types import RuleHttpMatchMethodFilter
 from .types import RuleHttpMatchPathFilterPathFilterType
 from .types import SearchBackendStagesRequestOrderBy
+from .types import SearchWafStagesRequestOrderBy
 from .types import WafStageMode
 from .types import ScalewayLb
 from .types import RuleHttpMatchPathFilter
@@ -112,6 +113,7 @@
 from .types import ListWafStagesResponse
 from .types import Plan
 from .types import SearchBackendStagesRequest
+from .types import SearchWafStagesRequest
 from .types import SelectPlanRequest
 from .types import SetHeadStageRequest
 from .types import SetRouteRulesRequest
@@ -149,6 +151,7 @@
     "RuleHttpMatchMethodFilter",
     "RuleHttpMatchPathFilterPathFilterType",
     "SearchBackendStagesRequestOrderBy",
+    "SearchWafStagesRequestOrderBy",
     "WafStageMode",
     "ScalewayLb",
     "RuleHttpMatchPathFilter",
@@ -238,6 +241,7 @@
     "ListWafStagesResponse",
     "Plan",
     "SearchBackendStagesRequest",
+    "SearchWafStagesRequest",
     "SelectPlanRequest",
     "SetHeadStageRequest",
     "SetRouteRulesRequest",

scaleway-async/scaleway_async/edge_services/v1beta1/api.py

@@ -22,6 +22,7 @@
     ListWafStagesRequestOrderBy,
     PlanName,
     SearchBackendStagesRequestOrderBy,
+    SearchWafStagesRequestOrderBy,
     WafStageMode,
     AddRouteRulesRequest,
     AddRouteRulesResponse,
@@ -1584,6 +1585,50 @@ async def delete_backend_stage(
 
         self._throw_on_error(res)
 
+    async def search_backend_stages(
+        self,
+        *,
+        order_by: Optional[SearchBackendStagesRequestOrderBy] = None,
+        page: Optional[int] = None,
+        page_size: Optional[int] = None,
+        project_id: Optional[str] = None,
+        bucket_name: Optional[str] = None,
+        bucket_region: Optional[str] = None,
+        lb_id: Optional[str] = None,
+    ) -> ListBackendStagesResponse:
+        """
+        :param order_by:
+        :param page:
+        :param page_size:
+        :param project_id:
+        :param bucket_name:
+        :param bucket_region:
+        :param lb_id:
+        :return: :class:`ListBackendStagesResponse <ListBackendStagesResponse>`
+
+        Usage:
+        ::
+
+            result = await api.search_backend_stages()
+        """
+
+        res = self._request(
+            "GET",
+            "/edge-services/v1beta1/search-backend-stages",
+            params={
+                "bucket_name": bucket_name,
+                "bucket_region": bucket_region,
+                "lb_id": lb_id,
+                "order_by": order_by,
+                "page": page,
+                "page_size": page_size or self.client.default_page_size,
+                "project_id": project_id or self.client.default_project_id,
+            },
+        )
+
+        self._throw_on_error(res)
+        return unmarshal_ListBackendStagesResponse(res.json())
+
     async def list_waf_stages(
         self,
         *,
@@ -1808,6 +1853,41 @@ async def delete_waf_stage(
 
         self._throw_on_error(res)
 
+    async def search_waf_stages(
+        self,
+        *,
+        order_by: Optional[SearchWafStagesRequestOrderBy] = None,
+        page: Optional[int] = None,
+        page_size: Optional[int] = None,
+        project_id: Optional[str] = None,
+    ) -> ListWafStagesResponse:
+        """
+        :param order_by:
+        :param page:
+        :param page_size:
+        :param project_id:
+        :return: :class:`ListWafStagesResponse <ListWafStagesResponse>`
+
+        Usage:
+        ::
+
+            result = await api.search_waf_stages()
+        """
+
+        res = self._request(
+            "GET",
+            "/edge-services/v1beta1/search-waf-stages",
+            params={
+                "order_by": order_by,
+                "page": page,
+                "page_size": page_size or self.client.default_page_size,
+                "project_id": project_id or self.client.default_project_id,
+            },
+        )
+
+        self._throw_on_error(res)
+        return unmarshal_ListWafStagesResponse(res.json())
+
     async def list_route_stages(
         self,
         *,
@@ -2212,50 +2292,6 @@ async def check_pem_chain(
         self._throw_on_error(res)
         return unmarshal_CheckPEMChainResponse(res.json())
 
-    async def search_backend_stages(
-        self,
-        *,
-        order_by: Optional[SearchBackendStagesRequestOrderBy] = None,
-        page: Optional[int] = None,
-        page_size: Optional[int] = None,
-        project_id: Optional[str] = None,
-        bucket_name: Optional[str] = None,
-        bucket_region: Optional[str] = None,
-        lb_id: Optional[str] = None,
-    ) -> ListBackendStagesResponse:
-        """
-        :param order_by:
-        :param page:
-        :param page_size:
-        :param project_id:
-        :param bucket_name:
-        :param bucket_region:
-        :param lb_id:
-        :return: :class:`ListBackendStagesResponse <ListBackendStagesResponse>`
-
-        Usage:
-        ::
-
-            result = await api.search_backend_stages()
-        """
-
-        res = self._request(
-            "GET",
-            "/edge-services/v1beta1/search-backend-stages",
-            params={
-                "bucket_name": bucket_name,
-                "bucket_region": bucket_region,
-                "lb_id": lb_id,
-                "order_by": order_by,
-                "page": page,
-                "page_size": page_size or self.client.default_page_size,
-                "project_id": project_id or self.client.default_project_id,
-            },
-        )
-
-        self._throw_on_error(res)
-        return unmarshal_ListBackendStagesResponse(res.json())
-
     async def list_purge_requests(
         self,
         *,

scaleway-async/scaleway_async/edge_services/v1beta1/types.py

@@ -239,6 +239,14 @@ def __str__(self) -> str:
         return str(self.value)
 
 
+class SearchWafStagesRequestOrderBy(str, Enum, metaclass=StrEnumMeta):
+    CREATED_AT_ASC = "created_at_asc"
+    CREATED_AT_DESC = "created_at_desc"
+
+    def __str__(self) -> str:
+        return str(self.value)
+
+
 class WafStageMode(str, Enum, metaclass=StrEnumMeta):
     UNKNOWN_MODE = "unknown_mode"
     DISABLE = "disable"
@@ -1610,6 +1618,17 @@ class SearchBackendStagesRequest:
     lb_id: Optional[str]
 
 
+@dataclass
+class SearchWafStagesRequest:
+    order_by: Optional[SearchWafStagesRequestOrderBy]
+
+    page: Optional[int]
+
+    page_size: Optional[int]
+
+    project_id: Optional[str]
+
+
 @dataclass
 class SelectPlanRequest:
     project_id: Optional[str]

scaleway/scaleway/edge_services/v1beta1/__init__.py

@@ -23,6 +23,7 @@
 from .types import RuleHttpMatchMethodFilter
 from .types import RuleHttpMatchPathFilterPathFilterType
 from .types import SearchBackendStagesRequestOrderBy
+from .types import SearchWafStagesRequestOrderBy
 from .types import WafStageMode
 from .types import ScalewayLb
 from .types import RuleHttpMatchPathFilter
@@ -112,6 +113,7 @@
 from .types import ListWafStagesResponse
 from .types import Plan
 from .types import SearchBackendStagesRequest
+from .types import SearchWafStagesRequest
 from .types import SelectPlanRequest
 from .types import SetHeadStageRequest
 from .types import SetRouteRulesRequest
@@ -149,6 +151,7 @@
     "RuleHttpMatchMethodFilter",
     "RuleHttpMatchPathFilterPathFilterType",
     "SearchBackendStagesRequestOrderBy",
+    "SearchWafStagesRequestOrderBy",
     "WafStageMode",
     "ScalewayLb",
     "RuleHttpMatchPathFilter",
@@ -238,6 +241,7 @@
     "ListWafStagesResponse",
     "Plan",
     "SearchBackendStagesRequest",
+    "SearchWafStagesRequest",
     "SelectPlanRequest",
     "SetHeadStageRequest",
     "SetRouteRulesRequest",