Merge pull request #46 from packethost/misc-tweaks

Misc tweaks

Author: mmlb

.drone.yml

@@ -10,8 +10,8 @@ pipeline:
   fmt_and_lint:
     image: python:3.6-alpine
     commands:
-      - pip install pylama yapf
-      - yapf -dr packet | (! grep '.')
+      - pip install black pylama
+      - black --check --diff .
       - pylama packet test setup.py
 
   test:

.envrc

@@ -0,0 +1 @@
+which nix &>/dev/null && use nix

CHANGELOG.md

@@ -7,6 +7,7 @@ This project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.html
 ## Unreleased
 ### Added
 - Support for python3.7
+- `legacy` param to `get_capacity` function
 ### Removed
 - Support for python3.3
 ### Changed

packet/Device.py

@@ -3,56 +3,59 @@
 from .OperatingSystem import OperatingSystem
 
 
-class Device():
-
+class Device:
     def __init__(self, data, manager):
         self.manager = manager
 
-        self.billing_cycle = data['billing_cycle']
-        self.created_at = data['created_at']
-        self.facility = data['facility']
-        self.hostname = data['hostname']
-        self.href = data['href']
-        self.id = data['id']
-        self.ip_addresses = data['ip_addresses']
-        self.locked = data['locked']
-        self.operating_system = OperatingSystem(data['operating_system'])
-        self.plan = data['plan']
-        self.spot_instance = data.get('spot_instance')
-        self.spot_price_max = data.get('spot_price_max')
-        self.ssh_keys = data.get('ssh_keys', [])
-        self.state = data['state']
-        self.tags = data['tags']
-        self.termination_time = data.get('termination_time')
-        self.updated_at = data['updated_at']
-        self.user = data['user']
+        self.billing_cycle = data["billing_cycle"]
+        self.created_at = data["created_at"]
+        self.facility = data["facility"]
+        self.hostname = data["hostname"]
+        self.href = data["href"]
+        self.id = data["id"]
+        self.ip_addresses = data["ip_addresses"]
+        self.locked = data["locked"]
+        self.operating_system = OperatingSystem(data["operating_system"])
+        self.plan = data["plan"]
+        self.spot_instance = data.get("spot_instance")
+        self.spot_price_max = data.get("spot_price_max")
+        self.ssh_keys = data.get("ssh_keys", [])
+        self.state = data["state"]
+        self.tags = data["tags"]
+        self.termination_time = data.get("termination_time")
+        self.updated_at = data["updated_at"]
+        self.user = data["user"]
 
     def update(self):
-        params = {
-            "hostname": self.hostname,
-            "locked": self.locked,
-            "tags": self.tags,
-        }
+        params = {"hostname": self.hostname, "locked": self.locked, "tags": self.tags}
 
-        return self.manager.call_api("devices/%s" % self.id, type='PATCH', params=params)
+        return self.manager.call_api(
+            "devices/%s" % self.id, type="PATCH", params=params
+        )
 
     def delete(self):
-        return self.manager.call_api("devices/%s" % self.id, type='DELETE')
+        return self.manager.call_api("devices/%s" % self.id, type="DELETE")
 
     def power_off(self):
-        params = {'type': 'power_off'}
-        return self.manager.call_api("devices/%s/actions" % self.id, type='POST', params=params)
+        params = {"type": "power_off"}
+        return self.manager.call_api(
+            "devices/%s/actions" % self.id, type="POST", params=params
+        )
 
     def power_on(self):
-        params = {'type': 'power_on'}
-        return self.manager.call_api("devices/%s/actions" % self.id, type='POST', params=params)
+        params = {"type": "power_on"}
+        return self.manager.call_api(
+            "devices/%s/actions" % self.id, type="POST", params=params
+        )
 
     def reboot(self):
-        params = {'type': 'reboot'}
-        return self.manager.call_api("devices/%s/actions" % self.id, type='POST', params=params)
+        params = {"type": "reboot"}
+        return self.manager.call_api(
+            "devices/%s/actions" % self.id, type="POST", params=params
+        )
 
     def __str__(self):
         return "%s" % self.hostname
 
     def __repr__(self):
-        return '{}: {}'.format(self.__class__.__name__, self.id)
+        return "{}: {}".format(self.__class__.__name__, self.id)

packet/Facility.py

@@ -1,17 +1,16 @@
 # -*- coding: utf-8 -*-
 
 
-class Facility():
-
+class Facility:
     def __init__(self, data):
-        self.id = data['id']
-        self.code = data['code']
-        self.name = data['name']
-        self.features = data['features']
-        self.address = data['address']
+        self.id = data["id"]
+        self.code = data["code"]
+        self.name = data["name"]
+        self.features = data["features"]
+        self.address = data["address"]
 
     def __str__(self):
         return "%s" % self.code
 
     def __repr__(self):
-        return '{}: {}'.format(self.__class__.__name__, self.id)
+        return "{}: {}".format(self.__class__.__name__, self.id)