Merge pull request #63 from obynio/add-hardware-reservations

Add hardware-reservations support

Author: jasmingacic

packet/HardwareReservation.py

@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+# SPDX-License-Identifier: LGPL-3.0-only
+
+from .Plan import Plan
+from .Project import Project
+from .Device import Device
+
+
+class HardwareReservation:
+    def __init__(self, data, manager):
+        self.manager = manager
+
+        self.id = data.get("id")
+        self.created_at = data.get("created_at")
+        self.billing_cycle = data.get("billing_cycle")
+        self.created_at = data.get("created_at")
+        self.short_id = data.get("short_id")
+        self.intervals = data.get("intervals")
+        self.current_period = data.get("current_period")
+        self.started_at = data.get("started_at")
+        self.custom_rate = data.get("custom_rate")
+        self.remove_at = data.get("remove_at")
+        self.project = data.get("project")
+        # self.facility = data.get("facility")
+        self.device = data.get("device")
+        self.provisionable = data.get("provisionable")
+        self.spare = data.get("spare")
+        self.need_of_service = data.get("need_of_service")
+        self.plan = Plan(data.get("plan"))
+        self.switch_uuid = data.get("switch_uuid")
+
+        try:
+            project_data = self.manager.call_api(data["project"]["href"], type="GET")
+            self.project = Project(project_data, self.manager)
+        except (KeyError, IndexError):
+            self.attached_to = None
+
+        # endpoint is not working yet
+        # try:
+        #     facility_data = self.manager.call_api(
+        #         data["facility"]["href"], type="GET"
+        #     )
+        #     self.project = Facility(facility_data, self.manager)
+        # except (KeyError, IndexError):
+        #     self.attached_to = None
+
+        try:
+            device_data = self.manager.call_api(data["device"]["href"], type="GET")
+            self.device = Device(device_data, self.manager)
+        except (KeyError, IndexError):
+            self.attached_to = None
+
+    def __str__(self):
+        return "%s" % self.id
+
+    def __repr__(self):
+        return "{}: {}".format(self.__class__.__name__, self.id)
+
+    def __getitem__(self, item):
+        return getattr(self, item)

packet/Manager.py

@@ -14,6 +14,7 @@
 from .BGPConfig import BGPConfig
 from .BGPSession import BGPSession
 from .IPAddress import IPAddress
+from .HardwareReservation import HardwareReservation
 from .Snapshot import Snapshot
 from .Organization import Organization
 from .Email import Email
@@ -73,6 +74,20 @@ def create_project(self, name):
         data = self.call_api("projects", type="POST", params=params)
         return Project(data, self)
 
+    def list_hardware_reservations(self, project_id, params={}):
+        data = self.call_api(
+            "projects/%s/hardware-reservations" % project_id, params=params
+        )
+        hardware_reservations = list()
+        for jsoned in data["hardware_reservations"]:
+            hardware_reservation = HardwareReservation(jsoned, self)
+            hardware_reservations.append(hardware_reservation)
+        return hardware_reservations
+
+    def get_hardware_reservation(self, hardware_reservation_id):
+        data = self.call_api("hardware-reservations/%s" % hardware_reservation_id)
+        return HardwareReservation(data, self)
+
     def list_devices(self, project_id, params={}):
         data = self.call_api("projects/%s/devices" % project_id, params=params)
         devices = list()