ZEN-35546: Add API that allows to get productionState from deviceRouter in batch

Author: Deer-WarLord

.gitignore

@@ -10,3 +10,4 @@ dist
 *.pyc
 *~
 *.sw?
+.vscode/

src/Products/Zuul/facades/devicefacade.py

@@ -18,7 +18,7 @@
 
 from AccessControl import getSecurityManager
 from Acquisition import aq_base
-from Products.AdvancedQuery import Eq, Or, Generic, And, MatchGlob
+from Products.AdvancedQuery import Eq, Or, Generic, And, MatchGlob, In
 from ZODB.transact import transact
 from zope.component import getMultiAdapter, getUtility
 from zope.event import notify
@@ -1486,3 +1486,26 @@ def maskPropertyPassword(self, inst, propname):
         if inst.zenPropIsPassword(propname):
             prop = "*" * len(prop)
         return prop
+
+    def getDevicesProductionState(self, uids):
+        """
+        Retrieves the production state for a list of devices.
+        """
+        model_catalog = IModelCatalogTool(self._dmd)
+        uids_set = set(uids)
+        query = In("uid", list(uids_set))
+        results = list(
+            model_catalog.search(
+                types="Products.ZenModel.Device.Device",
+                query=query,
+                fields=["productionState", "uid"],
+                limit=None,
+            ).results
+        )
+
+        if len(results) != len(uids_set):
+            found_uids = set(b.uid for b in results)
+            invalid_uids = uids_set - found_uids
+            raise Exception("Invalid device UIDs: %s" % ", ".join(invalid_uids))
+
+        return [{"uid": b.uid, "productionState": b.productionState} for b in results]

src/Products/Zuul/interfaces/device.py

@@ -221,3 +221,12 @@ def resetBoundTemplates(self, uid):
         @type uid: str
         @rtype: void
         """
+    def getDevicesProductionState(uids):
+        """
+        Retrieves the production state for a list of devices.
+
+        @type  uids: list
+        @param uids: List of device uids to get production state for
+        @rtype:   list
+        @return:  List of dictionaries containing device uid and productionState
+        """

src/Products/Zuul/routers/device.py

@@ -509,6 +509,25 @@ def getDevices(self, uid=None, start=0, params=None, limit=50, sort='name',
         return DirectResponse(devices=data, totalCount=devices.total,
                               hash=devices.hash_)
 
+    @serviceConnectionError
+    def getDevicesProductionState(self, uids):
+        """
+        Retrieves the production state for a list of devices.
+
+        @type  uids: list
+        @param uids: List of device uids to get production state for
+        @rtype:   DirectResponse
+        @return:  B{Properties}:
+           - data: (list) List of dictionaries containing device uid and productionState
+        """
+        facade = self._getFacade()
+        try:
+            data = facade.getDevicesProductionState(uids)
+            return DirectResponse.succeed(data=data)
+        except Exception as e:
+            log.exception(e)
+            return DirectResponse.exception(e, "Failed to get production states.")
+
     def renameDevice(self, uid, newId, retainGraphData=False):
         """
         Set the device specified by the uid,"uid" to have the