Merge pull request #10 from dan-smalley/device_router_add_comment_fun…

…ction

Add add_comment to device router

src/zenossapi/routers/device.py

@@ -864,6 +864,35 @@ def get_device_class(self, device_class):
             dc_data['data']
         )
 
+    def add_comment_by_uid(self, device_uid, comment):
+        """
+        Append a new comment to a device 
+
+        Arguments:
+            device_uid (str): The uid of the device to comment on 
+            comment (str): Text of the desired comment
+
+        Returns:
+            str: Response message
+        """
+
+        # Get existing comments 
+        device = self.get_device_by_uid(device_uid)
+        if len(device.comments) > 0:
+            comment = f'{device.comments}\n{comment}'
+
+        response_data = self._router_request(
+            self._make_request_data(
+                'setInfo',
+                dict(
+                    uid=device_uid,
+                    comments=comment,
+                )
+            )
+        )
+
+        return response_data['msg']
+
 
 class ZenossDeviceClass(DeviceRouter):
     """
@@ -1934,6 +1963,24 @@ def set_priority(self, priority):
         self.priority = priority
         return message
 
+    def add_comment(self, comment):
+        """
+        Append a comment to a device.
+
+        Arguments:
+            comment (str): Desired comment text
+
+        Returns:
+            str: Reponse message
+        """
+
+        message = self.add_comment_by_uid(self.uid, comment)
+        if len(self.comments) > 0:
+            self.comments = f'{self.comments}\n{comment}'
+        else:
+            self.comments = comment
+
+
     def set_collector(self, collector):
         """
         Set the collector for the device.