PENG-3689 - Introducing support for datasets functionality

Author: cpontes-ns1

ns1/rest/datasets.py

@@ -13,7 +13,9 @@ class Datasets(resource.BaseResource):
         "recipient_emails",
     ]
 
-    def _buildBody(self, name, datatype, repeat, timeframe, export_type, recipient_emails, **kwargs):
+    def _buildBody(
+        self, name, datatype, repeat, timeframe, export_type, recipient_emails, **kwargs
+    ):
         body = {
             "name": name,
             "datatype": datatype,
@@ -26,10 +28,20 @@ def _buildBody(self, name, datatype, repeat, timeframe, export_type, recipient_e
         return body
 
     def create(
-            self, name, datatype, repeat, timeframe, export_type, recipient_emails, callback=None, errback=None,
-            **kwargs
+        self,
+        name,
+        datatype,
+        repeat,
+        timeframe,
+        export_type,
+        recipient_emails,
+        callback=None,
+        errback=None,
+        **kwargs
     ):
-        body = self._buildBody(name, datatype, repeat, timeframe, export_type, recipient_emails, **kwargs)
+        body = self._buildBody(
+            name, datatype, repeat, timeframe, export_type, recipient_emails, **kwargs
+        )
         return self._make_request(
             "PUT",
             "%s" % self.ROOT,

ns1/rest/transport/requests.py

@@ -45,7 +45,9 @@ def _rateLimitHeaders(self, headers):
             "remaining": int(headers.get("X-RateLimit-Remaining", 100)),
         }
 
-    def _send(self, method, url, headers, data, files, params, errback, skip_json_parsing):
+    def _send(
+        self, method, url, headers, data, files, params, errback, skip_json_parsing
+    ):
         resp = self.REQ_MAP[method](
             url,
             headers=headers,
@@ -92,9 +94,7 @@ def _send(self, method, url, headers, data, files, params, errback, skip_json_pa
                     errback(resp)
                     return
                 else:
-                    raise ResourceException(
-                        "invalid json in response", resp, resp.text
-                    )
+                    raise ResourceException("invalid json in response", resp, resp.text)
         else:
             return response_headers, None
 
@@ -121,7 +121,14 @@ def send(
             while next_page is not None:
                 self._log.debug("following pagination to: %s" % next_page)
                 next_headers, next_json = self._send(
-                    method, next_page, headers, data, files, params, errback, skip_json_parsing
+                    method,
+                    next_page,
+                    headers,
+                    data,
+                    files,
+                    params,
+                    errback,
+                    skip_json_parsing,
                 )
                 jsonOut = pagination_handler(jsonOut, next_json)
                 next_page = get_next_page(next_headers)