From 981e08a9ae2f0e3d861a4170e5fa147289c04ab0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Sun, 26 Aug 2018 23:41:51 -0400
Subject: [PATCH] new: Add helpers for new server related APIs

Fix #266
---
 pymisp/api.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/pymisp/api.py b/pymisp/api.py
index e189c15..0e747bf 100644
--- a/pymisp/api.py
+++ b/pymisp/api.py
@@ -1622,6 +1622,25 @@ class PyMISP(object):
         response = self._prepare_request('POST', url, json.dumps(jdata))
         return self._check_response(response)
 
+    def server_pull(self, server_id, event_id=None):
+        url = urljoin(self.root_url, 'servers/pull/{}'.format(server_id))
+        if event_id is not None:
+            url += '/{}'.format(event_id)
+        response = self._prepare_request('GET', url)
+        return self._check_response(response)
+
+    def server_push(self, server_id, event_id=None):
+        url = urljoin(self.root_url, 'servers/push/{}'.format(server_id))
+        if event_id is not None:
+            url += '/{}'.format(event_id)
+        response = self._prepare_request('GET', url)
+        return self._check_response(response)
+
+    def servers_index(self):
+        url = urljoin(self.root_url, 'servers/index')
+        response = self._prepare_request('GET', url)
+        return self._check_response(response)
+
     # ############## Roles ##################
 
     def get_roles_list(self):