mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-20 04:12:25 -06:00
73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
from dcim.models import Interface, Device, DeviceRole
|
|
|
|
from ruamel.yaml import YAML
|
|
from pathlib import Path
|
|
|
|
import sys
|
|
|
|
|
|
def template_customer_locker(devices, port_count):
|
|
from dcim.models import Interface
|
|
|
|
for locker in devices:
|
|
i = 1
|
|
while i < 24:
|
|
strand0 = i
|
|
strand1 = i + 1
|
|
name = '{}-{}'.format(strand0, strand1)
|
|
|
|
i += 2
|
|
interface_data = {
|
|
'name': name,
|
|
'device': locker,
|
|
'type': 'keystone',
|
|
}
|
|
|
|
interface, created = Interface.objects.update_or_create(name=name, device=locker, defaults=interface_data)
|
|
if created:
|
|
print("🔗 Created interface {} for {}".format(interface.name, locker.name))
|
|
|
|
|
|
def template_efr(devices, port_count):
|
|
from dcim.models import Interface
|
|
|
|
for switch in devices:
|
|
i = 1
|
|
while i <= port_count:
|
|
interface, created = Interface.objects.update_or_create(name=f'xe-0/0/{i}', device=switch, defaults={'device': switch, 'type': '10gbase-x-sfpp'})
|
|
i += 1
|
|
if created:
|
|
print("🔗 Created interface {} for {}".format(interface.name, switch.name))
|
|
|
|
|
|
templates = {
|
|
'efr': template_efr,
|
|
'customer-locker': template_customer_locker,
|
|
}
|
|
|
|
file = Path('/opt/netbox/initializers/interfaces.yml')
|
|
if not file.is_file():
|
|
sys.exit()
|
|
|
|
with file.open('r') as stream:
|
|
yaml = YAML(typ='safe')
|
|
config = yaml.load(stream)
|
|
|
|
|
|
for c in config:
|
|
device_role = DeviceRole.objects.get(slug=c.get('device_role'))
|
|
|
|
if not device_role:
|
|
continue
|
|
|
|
interface_template = c.get('template')
|
|
|
|
if not interface_template or interface_template not in templates:
|
|
continue
|
|
|
|
port_count = c.get('ports', 0)
|
|
|
|
devices = Device.objects.all().filter(device_role=device_role.id)
|
|
|
|
templates[interface_template](devices, port_count)
|