def show_ipaddr(self, address=None, hostname=None, subnet_id=None, subnet_cidr=None):
if subnet_cidr:
subnet_id = self.find_subnet(cidr=subnet_cidr)
if address:
url = f"{self.base_url}/addresses/search/{address}/"
elif hostname:
url = f"{self.base_url}/addresses/search_hostname/{hostname}/"
else:
raise ValueError("Address or Hostname are needed!")
response = utils.make_request(
"GET",
url=url,
headers=dict(token=self.token)
)
if not response.get("data"):
if address: msg = f"ipv4 ({address})"
elif hostname: msg = f"hostname ({hostname})"
msg = f"Address with {msg} not found"
raise errors.NotFound(msg)
response = utils.JSONParser("IPInfoResponse", response)
if subnet_id:
subnet = self.show_subnet(subnet_id=subnet_id)
data = list(filter(lambda x: x.subnetId == subnet_id, response.data))
if not data:
if address: msg = f"ipv4 ({address})"
elif hostname: msg = f"hostname ({hostname})"
msg = f"Address with {msg} in the subnet ({subnet.subnet}/{subnet.mask}) not found!"
raise errors.NotFound(msg)
return data[0].export("IPInfoData")
return response.data
def check_free_ipaddr(self, subnet_id=None, subnet_cidr=None):
subnet_id = self.check_subnet(cidr=subnet_cidr, subnet_id=subnet_id)
url = f"{self.base_url}/addresses/first_free/{subnet_id}/"
response = utils.make_request(
"GET",
url=url,
headers=dict(token=self.token)
)
return utils.JSONParser("IPFreeResponse", response)
def show_subnet(self, subnet_id=None, cidr=None):
subnet_id = self.check_subnet(cidr=cidr, subnet_id=subnet_id)
url = f"{self.base_url}/subnets/{subnet_id}/"
response = utils.make_request(
"GET",
url=url,
headers=dict(token=self.token)
)
response = utils.JSONParser("SubnetInfoResponse", response)
return response.data.export("SubnetInfoData")
def release_ipaddr(self, address, subnet_id=None, subnet_cidr=None):
subnet_id = self.check_subnet(cidr=subnet_cidr, subnet_id=subnet_id)
url = f"{self.base_url}/addresses/{address}/{subnet_id}/"
response = utils.make_request(
"DELETE",
url=url,
headers=dict(token=self.token)
)
return utils.JSONParser("IPReleaseResponse", response)
def update_ipaddr(self, address, payload={}, subnet_id=None, subnet_cidr=None, show_result=False):
subnet_id = self.check_subnet(cidr=subnet_cidr, subnet_id=subnet_id)
address = self.show_ipaddr(address=address, subnet_id=subnet_id)
url = f"{self.base_url}/addresses/{address.id}/"
response = utils.make_request(
"PATCH",
url=url,
headers=dict(token=self.token),
payload=payload
)
if show_result:
response = utils.make_request(
"GET",
url=url,
headers=dict(token=self.token)
)
response = utils.JSONParser("IPUpdateResponse", response)
return response.data
return utils.JSONParser("IPUpdateResponse", response)
def add_ipaddr(self, payload={}, subnet_id=None, subnet_cidr=None, show_result=False):
subnet_id = self.check_subnet(cidr=subnet_cidr, subnet_id=subnet_id)
payload["subnetId"] = subnet_id
url = f"{self.base_url}/addresses/"
response = utils.make_request(
"POST",
url=url,
headers=dict(token=self.token),
payload=payload
)
response = utils.JSONParser("IPNewResponse", response)
if show_result:
response = utils.make_request(
"GET",
url=f"{url}{response.id}/",
headers=dict(token=self.token)
)
response = utils.JSONParser("IPNewResponse", response)
return response.data
return response.id
def reserve_ipaddr(self, subnet_id=None, subnet_cidr=None):
if subnet_cidr:
subnet_id = self.find_subnet(subnet_cidr)
elif not subnet_id:
raise ValueError("Subnet ID or CIDR are needed!")
url = f"{self.base_url}/addresses/first_free/{subnet_id}/"
response = utils.make_request(
"POST",
url=url,
headers=dict(token=self.token)
)
return utils.JSONParser("IPReserveResponse", response)
def find_subnet(self, cidr):
url = f"{self.base_url}/subnets/cidr/{cidr}/"
response = utils.make_request(
"GET",
url=url,
headers=dict(token=self.token)
)
if not response.get("data"):
raise errors.NotFound(f"Subnet CIDR ({cidr}) not found")
response = utils.JSONParser("SubnetInfoResponse", response)
subnet_id = response.data[0].id
return subnet_id