Implementing the method to get attributes and properties from the arg…

…s & kwargs in all the __init__ methods

digitalocean/Action.py

@@ -11,11 +11,15 @@ class Action(BaseAPI):
     resource_type = None
     region = None
 
-    def __init__(self, action_id=""):
+    def __init__(self, action_id="", *args, **kwargs):
         super(Action, self).__init__()
         if action_id:
             self.id = action_id
 
+        #Setting the attribute values
+        for attr in kwargs.keys():
+            setattr(self,attr,kwargs[attr])
+
     def load(self):
         action = self.get_data("actions/%s" % self.id)
         if action:

digitalocean/Domain.py

@@ -10,6 +10,7 @@ class Domain(BaseAPI):
 
     def __init__(self, *args, **kwargs):
         super(Domain, self).__init__()
+
         #Setting the attribute values
         for attr in kwargs.keys():
             setattr(self,attr,kwargs[attr])

digitalocean/Image.py

@@ -10,10 +10,12 @@ class Image(BaseAPI):
     regions = []
     created_at = None
 
-    def __init__(self, token=""):
+    def __init__(self, *args, **kwargs):
         super(Image, self).__init__()
-        if token:
-            self.token = token
+
+        #Setting the attribute values
+        for attr in kwargs.keys():
+            setattr(self,attr,kwargs[attr])
 
     def destroy(self):
         """

digitalocean/Manager.py

@@ -8,11 +8,14 @@
 
 
 class Manager(BaseAPI):
-    def __init__(self, token=""):
+    def __init__(self, token="", *args, **kwargs):
         super(Manager, self).__init__()
         if token:
             self.token = token
 
+        for attr in kwargs.keys():
+            setattr(self,attr,kwargs[attr])
+
     def get_data(self, *args, **kwargs):
         """
             Customized version of get_data to perform __check_actions_in_data

digitalocean/Record.py

@@ -11,13 +11,13 @@ class Record(BaseAPI):
     port = None
     weight = None
 
-    def __init__(self, domain_name, id="", token=""):
+    def __init__(self, domain_name, *args, **kwargs):
         super(Record, self).__init__()
         self.domain = domain_name
-        if id:
-            self.id = id
-        if token:
-            self.token = token
+
+        #Setting the attribute values
+        for attr in kwargs.keys():
+            setattr(self,attr,kwargs[attr])
 
     def create(self):
         """

digitalocean/Region.py

@@ -7,7 +7,9 @@ class Region(BaseAPI):
     available = None
     features = []
 
-    def __init__(self, token=""):
+    def __init__(self, *args, **kwargs):
         super(Region, self).__init__()
-        if token:
-            self.token = token
+
+        #Setting the attribute values
+        for attr in kwargs.keys():
+            setattr(self,attr,kwargs[attr])

digitalocean/SSHKey.py

@@ -7,10 +7,8 @@ class SSHKey(BaseAPI):
     public_key = None
     fingerprint = None
 
-    def __init__(self, token="", *args, **kwargs):
+    def __init__(self, *args, **kwargs):
         super(SSHKey, self).__init__()
-        if token:
-            self.token = token
 
         #Setting the attribute values
         for attr in kwargs.keys():

digitalocean/Size.py

@@ -10,7 +10,9 @@ class Size(BaseAPI):
     price_hourly = None
     regions = []
 
-    def __init__(self, token=""):
+    def __init__(self, *args, **kwargs):
         super(Size, self).__init__()
-        if token:
-            self.token = token
+
+        #Setting the attribute values
+        for attr in kwargs.keys():
+            setattr(self,attr,kwargs[attr])

digitalocean/baseapi.py

@@ -9,12 +9,15 @@ class BaseAPI(object):
     call_response = None
     end_point = "https://api.digitalocean.com/v2/"
 
-    def __init__(self):
+    def __init__(self, *args, **kwargs):
         super(BaseAPI, self).__init__()
         self.token = ""
         self.call_response = None
         self.end_point = "https://api.digitalocean.com/v2/"
 
+        for attr in kwargs.keys():
+            setattr(self,attr,kwargs[attr])
+
     def __perform_get(self, url, headers=dict(), params=dict()):
         return requests.get(url, headers=headers, params=params)