Size now is a BaseAPI subclass

koalalorenzo · Aug 11, 2014 · e83807f · e83807f
1 parent 4f27422
commit e83807f
Showing 1 changed file with 15 additions and 10 deletions.
diff --git a/digitalocean/Size.py b/digitalocean/Size.py
@@ -1,11 +1,16 @@
-class Size(object):
+from .baseapi import BaseAPI
+
+class Size(BaseAPI):
+    slug = None
+    memory = None
+    vcpus = None
+    disk = None
+    transfer = None
+    price_monthly = None
+    price_hourly = None
+    regions = []
+
     def __init__(self, token=""):
-        self.token = token
-        self.slug = None
-        self.memory = None
-        self.vcpus = None
-        self.disk = None
-        self.transfer = None
-        self.price_monthly = None
-        self.price_hourly = None
-        self.regions = []
+        super(Size, self).__init__()
+        if token:
+            self.token = token