get_kernel_available() gets kernels from all pages

koalalorenzo · Aug 15, 2014 · 602b4f4 · 602b4f4
1 parent 66d5657
commit 602b4f4
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/digitalocean/Droplet.py b/digitalocean/Droplet.py
@@ -317,12 +317,17 @@ def get_kernel_available(self):
         """
             Get a list of kernels available
         """
-        data = self.get_data("droplets/%s/kernels/" % self.id)
 
         kernels = list()
-        for jsond in data[u'kernels']:
-            kernel = Kernel(**jsond)
-            kernel.token = self.token
-            kernels.append(kernel)
+        data = self.get_data("droplets/%s/kernels/" % self.id)
+        while True:
+                for jsond in data[u'kernels']:
+                    kernel = Kernel(**jsond)
+                    kernel.token = self.token
+                    kernels.append(kernel)
+                url = data[u'links'][u'pages'].get(u'next')
+                if not url:
+                        break
+                data = self.get_data(data[u'links'][u'pages'].get(u'next'))
 
         return kernels