Update ASM groups to retrieve particular records.

Author: thinkingserious

example_v3_test.py

@@ -10,7 +10,7 @@
 
 client = sendgrid.SendGridAPIClient(os.environ.get('SENDGRID_API_KEY'))
 
-status, msg = client.asm_groups.get()
+status, msg = client.asm_groups.get([66,67,50])
 print status
 print msg
 

sendgrid/client.py

@@ -72,7 +72,7 @@ def _build_request(self, url, json_header=False, method='GET', data=None):
         return response.getcode(), body
 
     def get(self, api):
-        url = self.host + api.base_endpoint
+        url = self.host + api.endpoint
         response, body = self._build_request(url, False, 'GET')
         return response, body
 

sendgrid/resources/asm_groups.py

@@ -36,5 +36,21 @@ def client(self):
         return self._client
 
     # Retrieve all suppression groups associated with the user.
-    def get(self):
+    def get(self, id=None):
+        if id == None:
+            return self.client.get(self)
+        
+        if isinstance(id, int):
+            self._endpoint = self._base_endpoint + "/" + str(id)
+            return self.client.get(self)
+        
+        if len(id) > 1:
+            count = 0
+            for i in id:
+                if count == 0:
+                    self._endpoint = self._endpoint + "?id=" + str(i)
+                else:
+                    self._endpoint = self._endpoint + "&id=" + str(i)
+                count = count + 1
+            
         return self.client.get(self)