Skip to content

Commit

Permalink
Sucessfully creating devices with interfacesd and virtual machines (w…
Browse files Browse the repository at this point in the history 
…ithout its interfaces yet).
  • Loading branch information
@emersonfelipesp
emersonfelipesp committed Aug 22, 2024
1 parent 792f8c7 commit 6697a2f
Show file tree
Hide file tree
Showing 2 changed files with 132 additions and 77 deletions.
4 changes: 3 additions & 1 deletion netbox_proxbox/backend/routes/netbox/generic.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -503,11 +503,13 @@ async def _check_duplicate(self, search_params: dict = None, object: dict = None

result_by_tag = None
try:
logger.info("[CHECK DUPLICATE] (2.1) Searching object using 'get' method")
result_by_tag = await asyncio.to_thread(self.pynetbox_path.get,
name=object.get("name"),
slug=object.get("slug"),
tag=[self.nb.tag.slug]
)
print(result_by_tag)

except Exception as error:

Expand Down Expand Up @@ -539,7 +541,7 @@ async def _check_duplicate(self, search_params: dict = None, object: dict = None
return result_by_tag



logger.info(f"[CHECK DUPLICATE] (3) Checking duplicate object using only NAME and SLUG")
result_by_name_and_slug = await asyncio.to_thread(self.pynetbox_path.get,
name=object.get("name"),
slug=object.get("slug"),
Expand Down
205 changes: 129 additions & 76 deletions netbox_proxbox/backend/routes/proxbox/clusters/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@

from netbox_proxbox.backend.logging import logger

from netbox_proxbox.backend.exception import ProxboxException

from netbox_proxbox.backend import (
ClusterType,
Cluster,
Expand Down Expand Up @@ -61,24 +63,31 @@ async def proxbox_get_clusters(
description = cluster_description

# Create Cluster Type object before the Cluster itself
cluster_type_obj = await ClusterType(nb = nb).post(
data = {
"name": cluster_type_name,
"slug": cluster_type_slug,
"description": description
}
)
try:
logger.info("Creating the Cluster Type before Cluster...")
cluster_type_obj = await ClusterType(nb = nb).post(
data = {
"name": cluster_type_name,
"slug": cluster_type_slug,
"description": description
}
)
except Exception as error: raise ProxboxException(message="Error trying to create the cluster type.", python_exception=error)

# Create the Cluster
cluster_obj = await Cluster(nb = nb).post(
data = {
"name": px.name,
"slug": px.name,
"type": cluster_type_obj["id"],
"status": "active",
}
)

try:
logger.info("Creating the Cluster...")
cluster_obj = await Cluster(nb = nb).post(
data = {
"name": px.name,
"slug": px.name,
"type": cluster_type_obj["id"],
"status": "active",
}
)
except Exception as error: raise ProxboxException(message="Error trying to create the cluster.", python_exception=error)


result.append(
{
"name": px.name,
Expand Down Expand Up @@ -133,13 +142,20 @@ async def get_nodes(

for node in proxmox_nodes:

current_node = await Device(nb=nb).post(
data = {
"name": node.get("node"),
"cluster": get_cluster_from_netbox.id,
"status": "active",
}
)
try:
logger.info("Creating Device related with the Virtual Machine(s)")
current_node = await Device(nb=nb).post(
data = {
"name": node.get("node"),
"cluster": get_cluster_from_netbox.id,
"status": "active",
}
)
except Exception as error:
raise ProxboxException(
message="Error trying to create Netbox Device object.",
python_exception=error
)

nodes.append(current_node)

Expand All @@ -165,15 +181,21 @@ async def get_nodes(
enabled = True
else: enabled = False


create_interface = await Interface(nb=nb).post(data={
"device": current_node.id,
"name": interface_name,
"enabled": enabled,
"type": interface_type,
"mtu": interface.get("mtu", None),
"description": interface.get("comments", "")
})
try:
logger.info("Creating Netbox interface...")
create_interface = await Interface(nb=nb).post(data={
"device": current_node.id,
"name": interface_name,
"enabled": enabled,
"type": interface_type,
"mtu": interface.get("mtu", None),
"description": interface.get("comments", "")
})
except Exception as error:
raise ProxboxException(
message="Error trying to create Netbox interface.",
python_exception=error
)

print(f'create_interface: {create_interface}')

Expand All @@ -183,14 +205,20 @@ async def get_nodes(
print(f"cidr: {cidr}")

if cidr:
logger.info("Interface with CIDR/Network. Creating the IP Address object on Netbox...")
# If interface with network configured, create IP Address and attach interface to it.
create_ipaddress = await IPAddress(nb=nb, primary_field_value=cidr).post(data={
"address": cidr,
"assigned_object_id": create_interface.id,
"assigned_object_type": "dcim.interface"
})
print(f'create_ipaddress: {create_ipaddress}')
try:
logger.info("Interface with CIDR/Network. Creating the IP Address object on Netbox...")
# If interface with network configured, create IP Address and attach interface to it.
create_ipaddress = await IPAddress(nb=nb, primary_field_value=cidr).post(data={
"address": cidr,
"assigned_object_id": create_interface.id,
"assigned_object_type": "dcim.interface"
})
print(f'create_ipaddress: {create_ipaddress}')
except Exception as error:
raise ProxboxException(
message="Error trying to create IP Address of Interface on Netbox.",
python_exception=error
)



Expand All @@ -205,10 +233,14 @@ async def get_nodes(
for port in bridge_ports:
print(f'current_node: {current_node}')

netbox_port = await Interface(nb=nb).get(
device=current_node.name,
name=port
)
try:
logger.info("Searching children interface of a bridge.")
netbox_port = await Interface(nb=nb).get(
device=current_node.name,
name=port
)
except Exception as error: raise ProxboxException(message="Error trying to search bridge child interface.", python_exception=error)

print(f"port: {port}")
print(f"netbox_port: {netbox_port}")
if not netbox_port:
Expand All @@ -221,9 +253,13 @@ async def get_nodes(
enabled = True
else: enabled = False

# Interface and Bridge Interface must belong to the same Device
if create_interface.device == current_node.id:
print("Creating interface...")


# Interface and Bridge Interface must belong to the same Device
logger.info("Creating child interface of a bridge. Bridge interface and child must belong to the same device.")
if create_interface.device == current_node.id:
print("Creating interface...")
try:
new_netbox_port = await Interface(nb=nb).post(data={
"device": current_node.id,
"name": port,
Expand All @@ -233,22 +269,29 @@ async def get_nodes(
"description": proxmox_port.get("comments", ""),
"bridge": create_interface.id
})

cidr = proxmox_port.get("cidr")
print(f"[2] cidr: {cidr}")

if cidr:
# If interface with network configured, create IP Address and attach interface to it.
except Exception as error:
raise ProxboxException(
message="Error trying to create child interface of bridge interface.",
python_exception=error
)

cidr = proxmox_port.get("cidr")
print(f"[2] cidr: {cidr}")

if cidr:
logger.info("If interface with network configured, create IP Address and attach interface to it.")
try:
create_ipaddress = await IPAddress(nb=nb, primary_field_value=cidr).post(data={
"address": cidr,
"assigned_object_id": new_netbox_port.id,
"assigned_object_type": "dcim.interface"
})


except Exception as error: raise ProxboxException(message="Error trying to create IP Address on Netbox", python_exception=error)



else:
print("Interface already exists. Attaching Bridge to Interface")
logger.info("Interface already exists. Attaching Bridge to Interface")
print(f'create_interface: {create_interface}')
# Interface and Bridge Interface must belong to the same Device
if create_interface.device == current_node.id:
Expand Down Expand Up @@ -544,27 +587,37 @@ class VirtualMachineStatus(Enum):

print(f"\nplatform: {platform}\n")

# Create Custom Field and add Virtual Machine Proxmox ID
new_virtual_machine = await VirtualMachine(nb = nb).post(data = {
"name": vm.get("name"),
"cluster": cluster.id,
"device": device,
"status": VirtualMachineStatus(vm.get("status")).name,
"vcpus": int(vm.get("maxcpu", 0)),
"memory": int(vm_config.get("memory")),
"disk": int(int(vm.get("maxdisk", 0)) / 1000000000),
"role": role.id,
"custom_fields": {
"proxmox_vm_id": vm.get("vmid"),
"proxmox_start_at_boot": start_at_boot,
"proxmox_unprivileged_container": unprivileged_container,
"proxmox_qemu_agent": qemu_agent,
"proxmox_search_domain": search_domain,
},
"platform": platform
}
)
print(f"device: {device} / type: {type(device)}")

# Create Custom Field and add Virtual Machine Proxmox ID
try:
logger.info("Creating Virtual Machine on Netbox...")
new_virtual_machine = await VirtualMachine(nb = nb).post(data = {
"name": vm.get("name"),
"cluster": cluster.id,
"device": int(device.id),
"status": VirtualMachineStatus(vm.get("status")).name,
"vcpus": int(vm.get("maxcpu", 0)),
"memory": int(vm_config.get("memory")),
"disk": int(int(vm.get("maxdisk", 0)) / 1000000000),
"role": role.id,
"custom_fields": {
"proxmox_vm_id": vm.get("vmid"),
"proxmox_start_at_boot": start_at_boot,
"proxmox_unprivileged_container": unprivileged_container,
"proxmox_qemu_agent": qemu_agent,
"proxmox_search_domain": search_domain,
},
"platform": platform
}
)
except Exception as error:
raise ProxboxException(
message=f"[CHECK DUPLICATE] Error trying to create Virtual Machine {vm.get("name")} on Netbox.",
python_exception=f"{error}"
)





Expand Down

0 comments on commit 6697a2f

Please sign in to comment.