[Python-modules-commits] [python-digitalocean] 07/19: Implementing Tags (#137)

[Andrew Starr-Bochicchio]

This is an automated email from the git hooks/post-receive script.

asb pushed a commit to annotated tag 1.10.1
in repository python-digitalocean.

commit e1e2626068e6085b110478c873f4b3f3d1687f02
Author: Lorenzo Setale <koalalorenzo at users.noreply.github.com>
Date:   Mon Oct 10 21:38:28 2016 +0200

    Implementing Tags (#137)
    
    * Create Tag
    
    * add to get_all_droplets filtering by tag_name
    
    * Update and rename Tag to Tag.py
    
    * Update Tag.py
    
    * Small changes applied to refactor the code
---
 digitalocean/Manager.py |   8 +++-
 digitalocean/Tag.py     | 121 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 127 insertions(+), 2 deletions(-)

diff --git a/digitalocean/Manager.py b/digitalocean/Manager.py
index 728958c..c40dffd 100644
--- a/digitalocean/Manager.py
+++ b/digitalocean/Manager.py
@@ -84,11 +84,15 @@ class Manager(BaseAPI):
             regions.append(region)
         return regions
 
-    def get_all_droplets(self):
+    def get_all_droplets(self, tag_name=None):
         """
             This function returns a list of Droplet object.
         """
-        data = self.get_data("droplets/")
+        url = "droplets/"
+        if tag_name:
+            url += '?tag_name=' + tag_name
+
+        data = self.get_data(url)
         droplets = list()
         for jsoned in data['droplets']:
             droplet = Droplet(**jsoned)
diff --git a/digitalocean/Tag.py b/digitalocean/Tag.py
new file mode 100644
index 0000000..88eddc5
--- /dev/null
+++ b/digitalocean/Tag.py
@@ -0,0 +1,121 @@
+from .baseapi import BaseAPI
+from .Droplet import Droplet
+
+
+class Tag(BaseAPI):
+    def __init__(self, *args, **kwargs):
+        self.name = ""
+        self.resources = {}
+        super(Tag, self).__init__(*args, **kwargs)
+
+    @classmethod
+    def get_object(cls, api_token, tag_name):
+        tag = cls(token=api_token, name=tag_name)
+        tag.load()
+        return tag
+
+    def load(self):
+        """
+           Fetch data about tag
+        """
+        tags = self.get_data("tags/%s" % self.name)
+        tag = tags['tag']
+
+        for attr in tag.keys():
+            setattr(self, attr, tag[attr])
+
+        return self
+
+    def create(self, **kwargs):
+        """
+            Create the tag.
+        """
+        for attr in kwargs.keys():
+            setattr(self, attr, kwargs[attr])
+
+        query = {"name": self.name}
+
+        output = self.get_data("tags/", type="POST", params=query)
+        if output:
+            self.name = output['tag']['name']
+            self.resources = output['tag']['resources']
+
+    def get_resources(self, resources, method):
+        """ Method used to talk directly to the API (TAGs' Resources) """
+        tagged = self.get_data(
+            'tags/%s/resources' % self.name, params={
+                "resources": resources
+            },
+            type=method,
+        )
+        return tagged
+
+    def add_resources(self, resources):
+        """
+            Add to the resources to this tag.
+
+            Attributes accepted at creation time:
+                resources: array - See API.
+        """
+        return self.get_resources(resources, method='POST')
+
+    def remove_resources(self, resources):
+        """
+            Remove resources from this tag.
+
+            Attributes accepted at creation time:
+                resources: array - See API.
+        """
+        return self.get_resources(resources, method='DELETE')
+
+    def __extract_resources_from_droplets(self, data):
+        """
+            Private method to extract from a value, the resources.
+            It will check the type of object in the array provided and build
+            the right structure for the API.
+        """
+        resources = []
+        if not isinstance(data, list): return data
+        for a_droplet in data:
+            res = {}
+            if isinstance(a_droplet, str) or isinstance(a_droplet, int):
+                res = {"resource_id": a_droplet, "resource_type": "droplet"}
+            elif isinstance(a_droplet, Droplet):
+                res = {"resource_id": a_droplet.id, "resource_type": "droplet"}
+            else:
+                continue
+            resources.append(res)
+
+    def add_droplets(self, droplet):
+        """
+            Add the Tag to a Droplet.
+
+            Attributes accepted at creation time:
+                droplet: array of string or array of int, or array of Droplets.
+        """
+        if isinstance(droplet, list):
+            # Extracting data from the Droplet object
+            resources = self.__extract_resources_from_droplets(droplet)
+            return self.add_resources(resources)
+        else:
+            return self.add_resources([{
+                "resource_id": droplet.id,
+                "resource_type": "droplet"
+            }])
+
+    def remove_droplets(self, droplet):
+        """
+            Remove the Tag from the Droplet.
+
+            Attributes accepted at creation time:
+                droplet: array of string or array of int, or array of Droplets.
+        """
+        if isinstance(droplet, list):
+            # Extracting data from the Droplet object
+            resources = self.__extract_resources_from_droplets(droplet)
+            return self.remove_resources(resources)
+        else:
+            return self.remove_resources([{
+                "resource_id": droplet.id,
+                "resource_type": "droplet"
+            }])

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/python-modules/packages/python-digitalocean.git