mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
Remove extraneous demo scripts
This commit is contained in:
@ -1,66 +0,0 @@
|
||||
from django.utils.text import slugify
|
||||
|
||||
from dcim.constants import *
|
||||
from dcim.models import Device, DeviceRole, DeviceType, Site
|
||||
from extras.scripts import *
|
||||
|
||||
|
||||
class NewBranchScript(Script):
|
||||
script_name = "New Branch"
|
||||
script_description = "Provision a new branch site"
|
||||
script_fields = ['site_name', 'switch_count', 'switch_model']
|
||||
|
||||
site_name = StringVar(
|
||||
description="Name of the new site"
|
||||
)
|
||||
switch_count = IntegerVar(
|
||||
description="Number of access switches to create"
|
||||
)
|
||||
switch_model = ObjectVar(
|
||||
description="Access switch model",
|
||||
queryset=DeviceType.objects.filter(
|
||||
manufacturer__name='Cisco',
|
||||
model__in=['Catalyst 3560X-48T', 'Catalyst 3750X-48T']
|
||||
)
|
||||
)
|
||||
x = BooleanVar(
|
||||
description="Check me out"
|
||||
)
|
||||
|
||||
def run(self, data):
|
||||
|
||||
# Create the new site
|
||||
site = Site(
|
||||
name=data['site_name'],
|
||||
slug=slugify(data['site_name']),
|
||||
status=SITE_STATUS_PLANNED
|
||||
)
|
||||
site.save()
|
||||
self.log_success("Created new site: {}".format(site))
|
||||
|
||||
# Create access switches
|
||||
switch_role = DeviceRole.objects.get(name='Access Switch')
|
||||
for i in range(1, data['switch_count'] + 1):
|
||||
switch = Device(
|
||||
device_type=data['switch_model'],
|
||||
name='{}-switch{}'.format(site.slug, i),
|
||||
site=site,
|
||||
status=DEVICE_STATUS_PLANNED,
|
||||
device_role=switch_role
|
||||
)
|
||||
switch.save()
|
||||
self.log_success("Created new switch: {}".format(switch))
|
||||
|
||||
# Generate a CSV table of new devices
|
||||
output = [
|
||||
'name,make,model'
|
||||
]
|
||||
for switch in Device.objects.filter(site=site):
|
||||
attrs = [
|
||||
switch.name,
|
||||
switch.device_type.manufacturer.name,
|
||||
switch.device_type.model
|
||||
]
|
||||
output.append(','.join(attrs))
|
||||
|
||||
return '\n'.join(output)
|
@ -1,54 +0,0 @@
|
||||
from dcim.models import Site
|
||||
from extras.scripts import Script, BooleanVar, IntegerVar, ObjectVar, StringVar
|
||||
|
||||
|
||||
class NoInputScript(Script):
|
||||
description = "This script does not require any input"
|
||||
|
||||
def run(self, data):
|
||||
|
||||
self.log_debug("This a debug message.")
|
||||
self.log_info("This an info message.")
|
||||
self.log_success("This a success message.")
|
||||
self.log_warning("This a warning message.")
|
||||
self.log_failure("This a failure message.")
|
||||
|
||||
|
||||
class DemoScript(Script):
|
||||
name = "Script Demo"
|
||||
description = "A quick demonstration of the available field types"
|
||||
|
||||
my_string1 = StringVar(
|
||||
description="Input a string between 3 and 10 characters",
|
||||
min_length=3,
|
||||
max_length=10
|
||||
)
|
||||
my_string2 = StringVar(
|
||||
description="This field enforces a regex: three letters followed by three numbers",
|
||||
regex=r'[a-z]{3}\d{3}'
|
||||
)
|
||||
my_number = IntegerVar(
|
||||
description="Pick a number between 1 and 255 (inclusive)",
|
||||
min_value=1,
|
||||
max_value=255
|
||||
)
|
||||
my_boolean = BooleanVar(
|
||||
description="Use the checkbox to toggle true/false"
|
||||
)
|
||||
my_object = ObjectVar(
|
||||
description="Select a NetBox site",
|
||||
queryset=Site.objects.all()
|
||||
)
|
||||
|
||||
def run(self, data):
|
||||
|
||||
self.log_info("Your string was {}".format(data['my_string1']))
|
||||
self.log_info("Your second string was {}".format(data['my_string2']))
|
||||
self.log_info("Your number was {}".format(data['my_number']))
|
||||
if data['my_boolean']:
|
||||
self.log_info("You ticked the checkbox")
|
||||
else:
|
||||
self.log_info("You did not tick the checkbox")
|
||||
self.log_info("You chose the sites {}".format(data['my_object']))
|
||||
|
||||
return "Here's some output"
|
Reference in New Issue
Block a user