Merge pull request #123 from andrewsomething/droplet_features

Set the bool values for Droplet features on load.

Author: koalalorenzo

digitalocean/Droplet.py

@@ -138,6 +138,20 @@ def load(self):
                 self.ip_address = net['ip_address']
         if self.networks['v6']:
             self.ip_v6_address = self.networks['v6'][0]['ip_address']
+
+            if "backups" in self.features:
+                self.backups = True
+            else:
+                self.backups = False
+            if "ipv6" in self.features:
+                self.ipv6 = True
+            else:
+                self.ipv6 = False
+            if "private_networking" in self.features:
+                self.private_networking = True
+            else:
+                self.private_networking = False
+
         return self
 
     def _perform_action(self, params, return_dict=True):

digitalocean/Manager.py

@@ -99,7 +99,22 @@ def get_all_droplets(self):
                     droplet.ip_address = net['ip_address']
             if droplet.networks['v6']:
                 droplet.ip_v6_address = droplet.networks['v6'][0]['ip_address']
+
+            if "backups" in droplet.features:
+                droplet.backups = True
+            else:
+                droplet.backups = False
+            if "ipv6" in droplet.features:
+                droplet.ipv6 = True
+            else:
+                droplet.ipv6 = False
+            if "private_networking" in droplet.features:
+                droplet.private_networking = True
+            else:
+                droplet.private_networking = False
+
             droplets.append(droplet)
+
         return droplets
 
     def get_droplet(self, droplet_id):

digitalocean/tests/test_droplet.py

@@ -40,6 +40,9 @@ def test_load(self):
         self.assertEqual(d.memory, 512)
         self.assertEqual(d.vcpus, 1)
         self.assertEqual(d.disk, 20)
+        self.assertEqual(d.backups, False)
+        self.assertEqual(d.ipv6, True)
+        self.assertEqual(d.private_networking, False)
         self.assertEqual(d.region['slug'], "nyc3")
         self.assertEqual(d.status, "active")
         self.assertEqual(d.image['slug'], "ubuntu-14-04-x64")

digitalocean/tests/test_manager.py

@@ -68,6 +68,9 @@ def test_droplets(self):
         self.assertEqual(droplet.memory, 512)
         self.assertEqual(droplet.vcpus, 1)
         self.assertEqual(droplet.disk, 20)
+        self.assertEqual(droplet.backups, True)
+        self.assertEqual(droplet.ipv6, True)
+        self.assertEqual(droplet.private_networking, False)
         self.assertEqual(droplet.region['slug'], "nyc3")
         self.assertEqual(droplet.status, "active")
         self.assertEqual(droplet.image['slug'], "ubuntu-14-04-x64")