NAS-133688 / 25.04 / convert snapshot_count.py to new API

src/middlewared/middlewared/api/v25_04_0/__init__.py

@@ -47,6 +47,7 @@
 from .pool import *  # noqa
 from .pool_resilver import *  # noqa
 from .pool_scrub import *  # noqa
+from .pool_snapshot_count import *  # noqa
 from .pool_snapshottask import *  # noqa
 from .privilege import *  # noqa
 from .rdma import *  # noqa

src/middlewared/middlewared/api/v25_04_0/pool_snapshot_count.py

@@ -0,0 +1,9 @@
+from middlewared.api.base import BaseModel
+
+
+class PoolDatasetSnapshotCountArgs(BaseModel):
+    dataset: str
+
+
+class PoolDatasetSnapshotCountResults(BaseModel):
+    result: int

src/middlewared/middlewared/plugins/pool_/snapshot_count.py

@@ -1,16 +1,20 @@
-from middlewared.schema import accepts, returns, Int, Str
+from middlewared.api import api_method
+from middlewared.api.current import (
+    PoolDatasetSnapshotCountArgs,
+    PoolDatasetSnapshotCountResults,
+)
 from middlewared.service import Service
 
 
 class PoolDatasetService(Service):
-
     class Config:
         namespace = "pool.dataset"
 
-    @accepts(Str("dataset"), roles=['DATASET_READ'])
-    @returns(Int())
+    @api_method(
+        PoolDatasetSnapshotCountArgs,
+        PoolDatasetSnapshotCountResults,
+        roles=["DATASET_READ"],
+    )
     def snapshot_count(self, dataset):
-        """
-        Returns snapshot count for specified `dataset`.
-        """
+        """Returns snapshot count for specified `dataset`."""
         return self.middleware.call_sync("zfs.snapshot.count", [dataset])[dataset]