| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771 |
- import re
- from django import forms
- from django.contrib.auth.models import User
- from django.contrib.contenttypes.models import ContentType
- from django.contrib.postgres.forms.array import SimpleArrayField
- from django.core.exceptions import ObjectDoesNotExist
- from django.db.models import Q
- from mptt.forms import TreeNodeChoiceField
- from netaddr import EUI
- from netaddr.core import AddrFormatError
- from taggit.forms import TagField
- from timezone_field import TimeZoneFormField
- from circuits.models import Circuit, Provider
- from extras.forms import AddRemoveTagsForm, CustomFieldForm, CustomFieldBulkEditForm, CustomFieldFilterForm
- from ipam.models import IPAddress, VLAN, VLANGroup
- from tenancy.forms import TenancyForm
- from tenancy.forms import TenancyFilterForm
- from tenancy.models import Tenant, TenantGroup
- from utilities.forms import (
- APISelect, APISelectMultiple, add_blank_choice, ArrayFieldSelectMultiple, BootstrapMixin, BulkEditForm,
- BulkEditNullBooleanSelect, ChainedFieldsMixin, ChainedModelChoiceField, ColorSelect, CommentField, ComponentForm,
- ConfirmationForm, CSVChoiceField, ExpandableNameField, FilterChoiceField, FlexibleModelChoiceField, JSONField,
- SelectWithPK, SmallTextarea, SlugField, StaticSelect2, StaticSelect2Multiple, BOOLEAN_WITH_BLANK_CHOICES
- )
- from virtualization.models import Cluster, ClusterGroup
- from .constants import *
- from .models import (
- Cable, DeviceBay, DeviceBayTemplate, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate,
- Device, DeviceRole, DeviceType, FrontPort, FrontPortTemplate, Interface, InterfaceTemplate, Manufacturer,
- InventoryItem, Platform, PowerFeed, PowerOutlet, PowerOutletTemplate, PowerPanel, PowerPort, PowerPortTemplate,
- Rack, RackGroup, RackReservation, RackRole, RearPort, RearPortTemplate, Region, Site, VirtualChassis,
- )
- DEVICE_BY_PK_RE = r'{\d+\}'
- INTERFACE_MODE_HELP_TEXT = """
- Access: One untagged VLAN<br />
- Tagged: One untagged VLAN and/or one or more tagged VLANs<br />
- Tagged All: Implies all VLANs are available (w/optional untagged VLAN)
- """
- def get_device_by_name_or_pk(name):
- """
- Attempt to retrieve a device by either its name or primary key ('{pk}').
- """
- if re.match(DEVICE_BY_PK_RE, name):
- pk = name.strip('{}')
- device = Device.objects.get(pk=pk)
- else:
- device = Device.objects.get(name=name)
- return device
- class BulkRenameForm(forms.Form):
- """
- An extendable form to be used for renaming device components in bulk.
- """
- find = forms.CharField()
- replace = forms.CharField()
- use_regex = forms.BooleanField(
- required=False,
- initial=True,
- label='Use regular expressions'
- )
- def clean(self):
- # Validate regular expression in "find" field
- if self.cleaned_data['use_regex']:
- try:
- re.compile(self.cleaned_data['find'])
- except re.error:
- raise forms.ValidationError({
- 'find': "Invalid regular expression"
- })
- #
- # Fields
- #
- class MACAddressField(forms.Field):
- widget = forms.CharField
- default_error_messages = {
- 'invalid': 'MAC address must be in EUI-48 format',
- }
- def to_python(self, value):
- value = super().to_python(value)
- # Validate MAC address format
- try:
- value = EUI(value.strip())
- except AddrFormatError:
- raise forms.ValidationError(self.error_messages['invalid'], code='invalid')
- return value
- #
- # Regions
- #
- class RegionForm(BootstrapMixin, forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = Region
- fields = [
- 'parent', 'name', 'slug',
- ]
- widgets = {
- 'parent': APISelect(
- api_url="/api/dcim/regions/"
- )
- }
- class RegionCSVForm(forms.ModelForm):
- parent = forms.ModelChoiceField(
- queryset=Region.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Name of parent region',
- error_messages={
- 'invalid_choice': 'Region not found.',
- }
- )
- class Meta:
- model = Region
- fields = Region.csv_headers
- help_texts = {
- 'name': 'Region name',
- 'slug': 'URL-friendly slug',
- }
- class RegionFilterForm(BootstrapMixin, forms.Form):
- model = Site
- q = forms.CharField(
- required=False,
- label='Search'
- )
- #
- # Sites
- #
- class SiteForm(BootstrapMixin, TenancyForm, CustomFieldForm):
- region = TreeNodeChoiceField(
- queryset=Region.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/regions/"
- )
- )
- slug = SlugField()
- comments = CommentField()
- tags = TagField(
- required=False
- )
- class Meta:
- model = Site
- fields = [
- 'name', 'slug', 'status', 'region', 'tenant_group', 'tenant', 'facility', 'asn', 'time_zone', 'description',
- 'physical_address', 'shipping_address', 'latitude', 'longitude', 'contact_name', 'contact_phone',
- 'contact_email', 'comments', 'tags',
- ]
- widgets = {
- 'physical_address': SmallTextarea(
- attrs={
- 'rows': 3,
- }
- ),
- 'shipping_address': SmallTextarea(
- attrs={
- 'rows': 3,
- }
- ),
- 'status': StaticSelect2(),
- 'time_zone': StaticSelect2(),
- }
- help_texts = {
- 'name': "Full name of the site",
- 'facility': "Data center provider and facility (e.g. Equinix NY7)",
- 'asn': "BGP autonomous system number",
- 'time_zone': "Local time zone",
- 'description': "Short description (will appear in sites list)",
- 'physical_address': "Physical location of the building (e.g. for GPS)",
- 'shipping_address': "If different from the physical address",
- 'latitude': "Latitude in decimal format (xx.yyyyyy)",
- 'longitude': "Longitude in decimal format (xx.yyyyyy)"
- }
- class SiteCSVForm(forms.ModelForm):
- status = CSVChoiceField(
- choices=SITE_STATUS_CHOICES,
- required=False,
- help_text='Operational status'
- )
- region = forms.ModelChoiceField(
- queryset=Region.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Name of assigned region',
- error_messages={
- 'invalid_choice': 'Region not found.',
- }
- )
- tenant = forms.ModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Name of assigned tenant',
- error_messages={
- 'invalid_choice': 'Tenant not found.',
- }
- )
- class Meta:
- model = Site
- fields = Site.csv_headers
- help_texts = {
- 'name': 'Site name',
- 'slug': 'URL-friendly slug',
- 'asn': '32-bit autonomous system number',
- }
- class SiteBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Site.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- status = forms.ChoiceField(
- choices=add_blank_choice(SITE_STATUS_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- region = TreeNodeChoiceField(
- queryset=Region.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/regions/"
- )
- )
- tenant = forms.ModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/tenancy/tenants",
- )
- )
- asn = forms.IntegerField(
- min_value=1,
- max_value=4294967295,
- required=False,
- label='ASN'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- time_zone = TimeZoneFormField(
- choices=add_blank_choice(TimeZoneFormField().choices),
- required=False,
- widget=StaticSelect2()
- )
- class Meta:
- nullable_fields = [
- 'region', 'tenant', 'asn', 'description', 'time_zone',
- ]
- class SiteFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
- model = Site
- field_order = ['q', 'status', 'region', 'tenant_group', 'tenant']
- q = forms.CharField(
- required=False,
- label='Search'
- )
- status = forms.MultipleChoiceField(
- choices=SITE_STATUS_CHOICES,
- required=False,
- widget=StaticSelect2Multiple()
- )
- region = forms.ModelMultipleChoiceField(
- queryset=Region.objects.all(),
- to_field_name='slug',
- required=False,
- widget=APISelectMultiple(
- api_url="/api/dcim/regions/",
- value_field="slug",
- )
- )
- #
- # Rack groups
- #
- class RackGroupForm(BootstrapMixin, forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = RackGroup
- fields = [
- 'site', 'name', 'slug',
- ]
- widgets = {
- 'site': APISelect(
- api_url="/api/dcim/sites/"
- )
- }
- class RackGroupCSVForm(forms.ModelForm):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- to_field_name='name',
- help_text='Name of parent site',
- error_messages={
- 'invalid_choice': 'Site not found.',
- }
- )
- class Meta:
- model = RackGroup
- fields = RackGroup.csv_headers
- help_texts = {
- 'name': 'Name of rack group',
- 'slug': 'URL-friendly slug',
- }
- class RackGroupFilterForm(BootstrapMixin, forms.Form):
- site = FilterChoiceField(
- queryset=Site.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/sites/",
- value_field="slug",
- )
- )
- #
- # Rack roles
- #
- class RackRoleForm(BootstrapMixin, forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = RackRole
- fields = [
- 'name', 'slug', 'color',
- ]
- class RackRoleCSVForm(forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = RackRole
- fields = RackRole.csv_headers
- help_texts = {
- 'name': 'Name of rack role',
- 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
- }
- #
- # Racks
- #
- class RackForm(BootstrapMixin, TenancyForm, CustomFieldForm):
- group = ChainedModelChoiceField(
- queryset=RackGroup.objects.all(),
- chains=(
- ('site', 'site'),
- ),
- required=False,
- widget=APISelect(
- api_url='/api/dcim/rack-groups/',
- )
- )
- comments = CommentField()
- tags = TagField(
- required=False
- )
- class Meta:
- model = Rack
- fields = [
- 'site', 'group', 'name', 'facility_id', 'tenant_group', 'tenant', 'status', 'role', 'serial', 'asset_tag',
- 'type', 'width', 'u_height', 'desc_units', 'outer_width', 'outer_depth', 'outer_unit', 'comments', 'tags',
- ]
- help_texts = {
- 'site': "The site at which the rack exists",
- 'name': "Organizational rack name",
- 'facility_id': "The unique rack ID assigned by the facility",
- 'u_height': "Height in rack units",
- }
- widgets = {
- 'site': APISelect(
- api_url="/api/dcim/sites/",
- filter_for={
- 'group': 'site_id',
- }
- ),
- 'status': StaticSelect2(),
- 'role': APISelect(
- api_url="/api/dcim/rack-roles/"
- ),
- 'type': StaticSelect2(),
- 'width': StaticSelect2(),
- 'outer_unit': StaticSelect2(),
- }
- class RackCSVForm(forms.ModelForm):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- to_field_name='name',
- help_text='Name of parent site',
- error_messages={
- 'invalid_choice': 'Site not found.',
- }
- )
- group_name = forms.CharField(
- help_text='Name of rack group',
- required=False
- )
- tenant = forms.ModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Name of assigned tenant',
- error_messages={
- 'invalid_choice': 'Tenant not found.',
- }
- )
- status = CSVChoiceField(
- choices=RACK_STATUS_CHOICES,
- required=False,
- help_text='Operational status'
- )
- role = forms.ModelChoiceField(
- queryset=RackRole.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Name of assigned role',
- error_messages={
- 'invalid_choice': 'Role not found.',
- }
- )
- type = CSVChoiceField(
- choices=RACK_TYPE_CHOICES,
- required=False,
- help_text='Rack type'
- )
- width = forms.ChoiceField(
- choices=(
- (RACK_WIDTH_19IN, '19'),
- (RACK_WIDTH_23IN, '23'),
- ),
- help_text='Rail-to-rail width (in inches)'
- )
- outer_unit = CSVChoiceField(
- choices=RACK_DIMENSION_UNIT_CHOICES,
- required=False,
- help_text='Unit for outer dimensions'
- )
- class Meta:
- model = Rack
- fields = Rack.csv_headers
- help_texts = {
- 'name': 'Rack name',
- 'u_height': 'Height in rack units',
- }
- def clean(self):
- super().clean()
- site = self.cleaned_data.get('site')
- group_name = self.cleaned_data.get('group_name')
- name = self.cleaned_data.get('name')
- facility_id = self.cleaned_data.get('facility_id')
- # Validate rack group
- if group_name:
- try:
- self.instance.group = RackGroup.objects.get(site=site, name=group_name)
- except RackGroup.DoesNotExist:
- raise forms.ValidationError("Rack group {} not found for site {}".format(group_name, site))
- # Validate uniqueness of rack name within group
- if Rack.objects.filter(group=self.instance.group, name=name).exists():
- raise forms.ValidationError(
- "A rack named {} already exists within group {}".format(name, group_name)
- )
- # Validate uniqueness of facility ID within group
- if facility_id and Rack.objects.filter(group=self.instance.group, facility_id=facility_id).exists():
- raise forms.ValidationError(
- "A rack with the facility ID {} already exists within group {}".format(facility_id, group_name)
- )
- class RackBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Rack.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/sites",
- filter_for={
- 'group': 'site_id',
- }
- )
- )
- group = forms.ModelChoiceField(
- queryset=RackGroup.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/rack-groups",
- )
- )
- tenant = forms.ModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/tenancy/tenants",
- )
- )
- status = forms.ChoiceField(
- choices=add_blank_choice(RACK_STATUS_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- role = forms.ModelChoiceField(
- queryset=RackRole.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/rack-roles",
- )
- )
- serial = forms.CharField(
- max_length=50,
- required=False,
- label='Serial Number'
- )
- asset_tag = forms.CharField(
- max_length=50,
- required=False
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(RACK_TYPE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- width = forms.ChoiceField(
- choices=add_blank_choice(RACK_WIDTH_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- u_height = forms.IntegerField(
- required=False,
- label='Height (U)'
- )
- desc_units = forms.NullBooleanField(
- required=False,
- widget=BulkEditNullBooleanSelect,
- label='Descending units'
- )
- outer_width = forms.IntegerField(
- required=False,
- min_value=1
- )
- outer_depth = forms.IntegerField(
- required=False,
- min_value=1
- )
- outer_unit = forms.ChoiceField(
- choices=add_blank_choice(RACK_DIMENSION_UNIT_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- comments = CommentField(
- widget=SmallTextarea
- )
- class Meta:
- nullable_fields = [
- 'group', 'tenant', 'role', 'serial', 'asset_tag', 'outer_width', 'outer_depth', 'outer_unit', 'comments',
- ]
- class RackFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
- model = Rack
- field_order = ['q', 'site', 'group_id', 'status', 'role', 'tenant_group', 'tenant']
- q = forms.CharField(
- required=False,
- label='Search'
- )
- site = FilterChoiceField(
- queryset=Site.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/sites/",
- value_field="slug",
- filter_for={
- 'group_id': 'site'
- }
- )
- )
- group_id = ChainedModelChoiceField(
- label='Rack group',
- queryset=RackGroup.objects.select_related('site'),
- chains=(
- ('site', 'site'),
- ),
- required=False,
- widget=APISelectMultiple(
- api_url="/api/dcim/rack-groups/",
- null_option=True,
- )
- )
- status = forms.MultipleChoiceField(
- choices=RACK_STATUS_CHOICES,
- required=False,
- widget=StaticSelect2Multiple()
- )
- role = FilterChoiceField(
- queryset=RackRole.objects.all(),
- to_field_name='slug',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/dcim/rack-roles/",
- value_field="slug",
- null_option=True,
- )
- )
- #
- # Rack reservations
- #
- class RackReservationForm(BootstrapMixin, TenancyForm, forms.ModelForm):
- units = SimpleArrayField(
- base_field=forms.IntegerField(),
- widget=ArrayFieldSelectMultiple(
- attrs={
- 'size': 10,
- }
- )
- )
- user = forms.ModelChoiceField(
- queryset=User.objects.order_by(
- 'username'
- ),
- widget=StaticSelect2()
- )
- class Meta:
- model = RackReservation
- fields = [
- 'units', 'user', 'tenant_group', 'tenant', 'description',
- ]
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Populate rack unit choices
- self.fields['units'].widget.choices = self._get_unit_choices()
- def _get_unit_choices(self):
- rack = self.instance.rack
- reserved_units = []
- for resv in rack.reservations.exclude(pk=self.instance.pk):
- for u in resv.units:
- reserved_units.append(u)
- unit_choices = [(u, {'label': str(u), 'disabled': u in reserved_units}) for u in rack.units]
- return unit_choices
- class RackReservationBulkEditForm(BootstrapMixin, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=RackReservation.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- user = forms.ModelChoiceField(
- queryset=User.objects.order_by(
- 'username'
- ),
- required=False,
- widget=StaticSelect2()
- )
- tenant = forms.ModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/tenancy/tenant",
- )
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- class Meta:
- nullable_fields = []
- class RackReservationFilterForm(BootstrapMixin, TenancyFilterForm):
- field_order = ['q', 'site', 'group_id', 'tenant_group', 'tenant']
- q = forms.CharField(
- required=False,
- label='Search'
- )
- site = FilterChoiceField(
- queryset=Site.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/sites/",
- value_field="slug",
- )
- )
- group_id = FilterChoiceField(
- queryset=RackGroup.objects.select_related('site'),
- label='Rack group',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/dcim/rack-groups/",
- null_option=True,
- )
- )
- #
- # Manufacturers
- #
- class ManufacturerForm(BootstrapMixin, forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = Manufacturer
- fields = [
- 'name', 'slug',
- ]
- class ManufacturerCSVForm(forms.ModelForm):
- class Meta:
- model = Manufacturer
- fields = Manufacturer.csv_headers
- help_texts = {
- 'name': 'Manufacturer name',
- 'slug': 'URL-friendly slug',
- }
- #
- # Device types
- #
- class DeviceTypeForm(BootstrapMixin, CustomFieldForm):
- slug = SlugField(
- slug_source='model'
- )
- tags = TagField(
- required=False
- )
- class Meta:
- model = DeviceType
- fields = [
- 'manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'subdevice_role', 'comments',
- 'tags',
- ]
- widgets = {
- 'manufacturer': APISelect(
- api_url="/api/dcim/manufacturers/"
- ),
- 'subdevice_role': StaticSelect2()
- }
- class DeviceTypeCSVForm(forms.ModelForm):
- manufacturer = forms.ModelChoiceField(
- queryset=Manufacturer.objects.all(),
- required=True,
- to_field_name='name',
- help_text='Manufacturer name',
- error_messages={
- 'invalid_choice': 'Manufacturer not found.',
- }
- )
- subdevice_role = CSVChoiceField(
- choices=SUBDEVICE_ROLE_CHOICES,
- required=False,
- help_text='Parent/child status'
- )
- class Meta:
- model = DeviceType
- fields = DeviceType.csv_headers
- help_texts = {
- 'model': 'Model name',
- 'slug': 'URL-friendly slug',
- }
- class DeviceTypeBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=DeviceType.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- manufacturer = forms.ModelChoiceField(
- queryset=Manufacturer.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/manufactureres"
- )
- )
- u_height = forms.IntegerField(
- min_value=1,
- required=False
- )
- is_full_depth = forms.NullBooleanField(
- required=False,
- widget=BulkEditNullBooleanSelect(),
- label='Is full depth'
- )
- class Meta:
- nullable_fields = []
- class DeviceTypeFilterForm(BootstrapMixin, CustomFieldFilterForm):
- model = DeviceType
- q = forms.CharField(
- required=False,
- label='Search'
- )
- manufacturer = FilterChoiceField(
- queryset=Manufacturer.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/manufacturers/",
- value_field="slug",
- )
- )
- subdevice_role = forms.NullBooleanField(
- required=False,
- label='Subdevice role',
- widget=StaticSelect2(
- choices=add_blank_choice(SUBDEVICE_ROLE_CHOICES)
- )
- )
- console_ports = forms.NullBooleanField(
- required=False,
- label='Has console ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- console_server_ports = forms.NullBooleanField(
- required=False,
- label='Has console server ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- power_ports = forms.NullBooleanField(
- required=False,
- label='Has power ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- power_outlets = forms.NullBooleanField(
- required=False,
- label='Has power outlets',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- interfaces = forms.NullBooleanField(
- required=False,
- label='Has interfaces',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- pass_through_ports = forms.NullBooleanField(
- required=False,
- label='Has pass-through ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- #
- # Device component templates
- #
- class ConsolePortTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = ConsolePortTemplate
- fields = [
- 'device_type', 'name',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- }
- class ConsolePortTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- class ConsoleServerPortTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = ConsoleServerPortTemplate
- fields = [
- 'device_type', 'name',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- }
- class ConsoleServerPortTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- class PowerPortTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = PowerPortTemplate
- fields = [
- 'device_type', 'name', 'maximum_draw', 'allocated_draw',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- }
- class PowerPortTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- maximum_draw = forms.IntegerField(
- min_value=1,
- required=False,
- help_text="Maximum current draw (watts)"
- )
- allocated_draw = forms.IntegerField(
- min_value=1,
- required=False,
- help_text="Allocated current draw (watts)"
- )
- class PowerOutletTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = PowerOutletTemplate
- fields = [
- 'device_type', 'name', 'power_port', 'feed_leg',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- }
- class PowerOutletTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- power_port = forms.ModelChoiceField(
- queryset=PowerPortTemplate.objects.all(),
- required=False
- )
- feed_leg = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_LEG_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Limit power_port choices to current DeviceType
- self.fields['power_port'].queryset = PowerPortTemplate.objects.filter(
- device_type=self.parent
- )
- class InterfaceTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = InterfaceTemplate
- fields = [
- 'device_type', 'name', 'type', 'mgmt_only',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- 'type': StaticSelect2(),
- }
- class InterfaceTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- type = forms.ChoiceField(
- choices=IFACE_TYPE_CHOICES,
- widget=StaticSelect2()
- )
- mgmt_only = forms.BooleanField(
- required=False,
- label='Management only'
- )
- class InterfaceTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=InterfaceTemplate.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(IFACE_TYPE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- mgmt_only = forms.NullBooleanField(
- required=False,
- widget=BulkEditNullBooleanSelect,
- label='Management only'
- )
- class Meta:
- nullable_fields = []
- class FrontPortTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = FrontPortTemplate
- fields = [
- 'device_type', 'name', 'type', 'rear_port', 'rear_port_position',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- 'rear_port': StaticSelect2(),
- }
- class FrontPortTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- type = forms.ChoiceField(
- choices=PORT_TYPE_CHOICES,
- widget=StaticSelect2()
- )
- rear_port_set = forms.MultipleChoiceField(
- choices=[],
- label='Rear ports',
- help_text='Select one rear port assignment for each front port being created.',
- )
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Determine which rear port positions are occupied. These will be excluded from the list of available mappings.
- occupied_port_positions = [
- (front_port.rear_port_id, front_port.rear_port_position)
- for front_port in self.parent.frontport_templates.all()
- ]
- # Populate rear port choices
- choices = []
- rear_ports = RearPortTemplate.objects.filter(device_type=self.parent)
- for rear_port in rear_ports:
- for i in range(1, rear_port.positions + 1):
- if (rear_port.pk, i) not in occupied_port_positions:
- choices.append(
- ('{}:{}'.format(rear_port.pk, i), '{}:{}'.format(rear_port.name, i))
- )
- self.fields['rear_port_set'].choices = choices
- def clean(self):
- # Validate that the number of ports being created equals the number of selected (rear port, position) tuples
- front_port_count = len(self.cleaned_data['name_pattern'])
- rear_port_count = len(self.cleaned_data['rear_port_set'])
- if front_port_count != rear_port_count:
- raise forms.ValidationError({
- 'rear_port_set': 'The provided name pattern will create {} ports, however {} rear port assignments '
- 'were selected. These counts must match.'.format(front_port_count, rear_port_count)
- })
- def get_iterative_data(self, iteration):
- # Assign rear port and position from selected set
- rear_port, position = self.cleaned_data['rear_port_set'][iteration].split(':')
- return {
- 'rear_port': int(rear_port),
- 'rear_port_position': int(position),
- }
- class RearPortTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = RearPortTemplate
- fields = [
- 'device_type', 'name', 'type', 'positions',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- 'type': StaticSelect2(),
- }
- class RearPortTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- type = forms.ChoiceField(
- choices=PORT_TYPE_CHOICES,
- widget=StaticSelect2(),
- )
- positions = forms.IntegerField(
- min_value=1,
- max_value=64,
- initial=1,
- help_text='The number of front ports which may be mapped to each rear port'
- )
- class DeviceBayTemplateForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = DeviceBayTemplate
- fields = [
- 'device_type', 'name',
- ]
- widgets = {
- 'device_type': forms.HiddenInput(),
- }
- class DeviceBayTemplateCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- #
- # Device roles
- #
- class DeviceRoleForm(BootstrapMixin, forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = DeviceRole
- fields = [
- 'name', 'slug', 'color', 'vm_role',
- ]
- class DeviceRoleCSVForm(forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = DeviceRole
- fields = DeviceRole.csv_headers
- help_texts = {
- 'name': 'Name of device role',
- 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
- }
- #
- # Platforms
- #
- class PlatformForm(BootstrapMixin, forms.ModelForm):
- slug = SlugField()
- class Meta:
- model = Platform
- fields = [
- 'name', 'slug', 'manufacturer', 'napalm_driver', 'napalm_args',
- ]
- widgets = {
- 'manufacturer': APISelect(
- api_url="/api/dcim/manufacturers/"
- ),
- 'napalm_args': SmallTextarea(),
- }
- class PlatformCSVForm(forms.ModelForm):
- slug = SlugField()
- manufacturer = forms.ModelChoiceField(
- queryset=Manufacturer.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Manufacturer name',
- error_messages={
- 'invalid_choice': 'Manufacturer not found.',
- }
- )
- class Meta:
- model = Platform
- fields = Platform.csv_headers
- help_texts = {
- 'name': 'Platform name',
- }
- #
- # Devices
- #
- class DeviceForm(BootstrapMixin, TenancyForm, CustomFieldForm):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- widget=APISelect(
- api_url="/api/dcim/sites/",
- filter_for={
- 'rack': 'site_id'
- }
- )
- )
- rack = ChainedModelChoiceField(
- queryset=Rack.objects.all(),
- chains=(
- ('site', 'site'),
- ),
- required=False,
- widget=APISelect(
- api_url='/api/dcim/racks/',
- display_field='display_name'
- )
- )
- position = forms.TypedChoiceField(
- required=False,
- empty_value=None,
- help_text="The lowest-numbered unit occupied by the device",
- widget=APISelect(
- api_url='/api/dcim/racks/{{rack}}/units/',
- disabled_indicator='device'
- )
- )
- manufacturer = forms.ModelChoiceField(
- queryset=Manufacturer.objects.all(),
- widget=APISelect(
- api_url="/api/dcim/manufacturers/",
- filter_for={
- 'device_type': 'manufacturer_id'
- }
- )
- )
- device_type = ChainedModelChoiceField(
- queryset=DeviceType.objects.all(),
- chains=(
- ('manufacturer', 'manufacturer'),
- ),
- label='Device type',
- widget=APISelect(
- api_url='/api/dcim/device-types/',
- display_field='model'
- )
- )
- cluster_group = forms.ModelChoiceField(
- queryset=ClusterGroup.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/virtualization/cluster-groups/",
- filter_for={
- 'cluster': 'group_id'
- },
- attrs={
- 'nullable': 'true'
- }
- )
- )
- cluster = ChainedModelChoiceField(
- queryset=Cluster.objects.all(),
- chains=(
- ('group', 'cluster_group'),
- ),
- required=False,
- widget=APISelect(
- api_url='/api/virtualization/clusters/',
- )
- )
- comments = CommentField()
- tags = TagField(required=False)
- local_context_data = JSONField(required=False)
- class Meta:
- model = Device
- fields = [
- 'name', 'device_role', 'device_type', 'serial', 'asset_tag', 'site', 'rack', 'position', 'face',
- 'status', 'platform', 'primary_ip4', 'primary_ip6', 'cluster_group', 'cluster', 'tenant_group', 'tenant',
- 'comments', 'tags', 'local_context_data'
- ]
- help_texts = {
- 'device_role': "The function this device serves",
- 'serial': "Chassis serial number",
- 'local_context_data': "Local config context data overwrites all source contexts in the final rendered "
- "config context",
- }
- widgets = {
- 'face': StaticSelect2(
- filter_for={
- 'position': 'face'
- }
- ),
- 'device_role': APISelect(
- api_url='/api/dcim/device-roles/'
- ),
- 'status': StaticSelect2(),
- 'platform': APISelect(
- api_url="/api/dcim/platforms/"
- ),
- 'primary_ip4': StaticSelect2(),
- 'primary_ip6': StaticSelect2(),
- }
- def __init__(self, *args, **kwargs):
- # Initialize helper selectors
- instance = kwargs.get('instance')
- if 'initial' not in kwargs:
- kwargs['initial'] = {}
- # Using hasattr() instead of "is not None" to avoid RelatedObjectDoesNotExist on required field
- if instance and hasattr(instance, 'device_type'):
- kwargs['initial']['manufacturer'] = instance.device_type.manufacturer
- if instance and instance.cluster is not None:
- kwargs['initial']['cluster_group'] = instance.cluster.group
- super().__init__(*args, **kwargs)
- if self.instance.pk:
- # Compile list of choices for primary IPv4 and IPv6 addresses
- for family in [4, 6]:
- ip_choices = [(None, '---------')]
- # Gather PKs of all interfaces belonging to this Device or a peer VirtualChassis member
- interface_ids = self.instance.vc_interfaces.values('pk')
- # Collect interface IPs
- interface_ips = IPAddress.objects.select_related('interface').filter(
- family=family, interface_id__in=interface_ids
- )
- if interface_ips:
- ip_list = [(ip.id, '{} ({})'.format(ip.address, ip.interface)) for ip in interface_ips]
- ip_choices.append(('Interface IPs', ip_list))
- # Collect NAT IPs
- nat_ips = IPAddress.objects.select_related('nat_inside').filter(
- family=family, nat_inside__interface__in=interface_ids
- )
- if nat_ips:
- ip_list = [(ip.id, '{} ({})'.format(ip.address, ip.nat_inside.address)) for ip in nat_ips]
- ip_choices.append(('NAT IPs', ip_list))
- self.fields['primary_ip{}'.format(family)].choices = ip_choices
- # If editing an existing device, exclude it from the list of occupied rack units. This ensures that a device
- # can be flipped from one face to another.
- self.fields['position'].widget.add_additional_query_param('exclude', self.instance.pk)
- # Limit platform by manufacturer
- self.fields['platform'].queryset = Platform.objects.filter(
- Q(manufacturer__isnull=True) | Q(manufacturer=self.instance.device_type.manufacturer)
- )
- else:
- # An object that doesn't exist yet can't have any IPs assigned to it
- self.fields['primary_ip4'].choices = []
- self.fields['primary_ip4'].widget.attrs['readonly'] = True
- self.fields['primary_ip6'].choices = []
- self.fields['primary_ip6'].widget.attrs['readonly'] = True
- # Rack position
- pk = self.instance.pk if self.instance.pk else None
- try:
- if self.is_bound and self.data.get('rack') and str(self.data.get('face')):
- position_choices = Rack.objects.get(pk=self.data['rack']) \
- .get_rack_units(face=self.data.get('face'), exclude=pk)
- elif self.initial.get('rack') and str(self.initial.get('face')):
- position_choices = Rack.objects.get(pk=self.initial['rack']) \
- .get_rack_units(face=self.initial.get('face'), exclude=pk)
- else:
- position_choices = []
- except Rack.DoesNotExist:
- position_choices = []
- self.fields['position'].choices = [('', '---------')] + [
- (p['id'], {
- 'label': p['name'],
- 'disabled': bool(p['device'] and p['id'] != self.initial.get('position')),
- }) for p in position_choices
- ]
- # Disable rack assignment if this is a child device installed in a parent device
- if pk and self.instance.device_type.is_child_device and hasattr(self.instance, 'parent_bay'):
- self.fields['site'].disabled = True
- self.fields['rack'].disabled = True
- self.initial['site'] = self.instance.parent_bay.device.site_id
- self.initial['rack'] = self.instance.parent_bay.device.rack_id
- class BaseDeviceCSVForm(forms.ModelForm):
- device_role = forms.ModelChoiceField(
- queryset=DeviceRole.objects.all(),
- to_field_name='name',
- help_text='Name of assigned role',
- error_messages={
- 'invalid_choice': 'Invalid device role.',
- }
- )
- tenant = forms.ModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Name of assigned tenant',
- error_messages={
- 'invalid_choice': 'Tenant not found.',
- }
- )
- manufacturer = forms.ModelChoiceField(
- queryset=Manufacturer.objects.all(),
- to_field_name='name',
- help_text='Device type manufacturer',
- error_messages={
- 'invalid_choice': 'Invalid manufacturer.',
- }
- )
- model_name = forms.CharField(
- help_text='Device type model name'
- )
- platform = forms.ModelChoiceField(
- queryset=Platform.objects.all(),
- required=False,
- to_field_name='name',
- help_text='Name of assigned platform',
- error_messages={
- 'invalid_choice': 'Invalid platform.',
- }
- )
- status = CSVChoiceField(
- choices=DEVICE_STATUS_CHOICES,
- help_text='Operational status'
- )
- class Meta:
- fields = []
- model = Device
- help_texts = {
- 'name': 'Device name',
- }
- def clean(self):
- super().clean()
- manufacturer = self.cleaned_data.get('manufacturer')
- model_name = self.cleaned_data.get('model_name')
- # Validate device type
- if manufacturer and model_name:
- try:
- self.instance.device_type = DeviceType.objects.get(manufacturer=manufacturer, model=model_name)
- except DeviceType.DoesNotExist:
- raise forms.ValidationError("Device type {} {} not found".format(manufacturer, model_name))
- class DeviceCSVForm(BaseDeviceCSVForm):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- to_field_name='name',
- help_text='Name of parent site',
- error_messages={
- 'invalid_choice': 'Invalid site name.',
- }
- )
- rack_group = forms.CharField(
- required=False,
- help_text='Parent rack\'s group (if any)'
- )
- rack_name = forms.CharField(
- required=False,
- help_text='Name of parent rack'
- )
- face = CSVChoiceField(
- choices=RACK_FACE_CHOICES,
- required=False,
- help_text='Mounted rack face'
- )
- cluster = forms.ModelChoiceField(
- queryset=Cluster.objects.all(),
- to_field_name='name',
- required=False,
- help_text='Virtualization cluster',
- error_messages={
- 'invalid_choice': 'Invalid cluster name.',
- }
- )
- class Meta(BaseDeviceCSVForm.Meta):
- fields = [
- 'name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag', 'status',
- 'site', 'rack_group', 'rack_name', 'position', 'face', 'cluster', 'comments',
- ]
- def clean(self):
- super().clean()
- site = self.cleaned_data.get('site')
- rack_group = self.cleaned_data.get('rack_group')
- rack_name = self.cleaned_data.get('rack_name')
- # Validate rack
- if site and rack_group and rack_name:
- try:
- self.instance.rack = Rack.objects.get(site=site, group__name=rack_group, name=rack_name)
- except Rack.DoesNotExist:
- raise forms.ValidationError("Rack {} not found in site {} group {}".format(rack_name, site, rack_group))
- elif site and rack_name:
- try:
- self.instance.rack = Rack.objects.get(site=site, group__isnull=True, name=rack_name)
- except Rack.DoesNotExist:
- raise forms.ValidationError("Rack {} not found in site {} (no group)".format(rack_name, site))
- class ChildDeviceCSVForm(BaseDeviceCSVForm):
- parent = FlexibleModelChoiceField(
- queryset=Device.objects.all(),
- to_field_name='name',
- help_text='Name or ID of parent device',
- error_messages={
- 'invalid_choice': 'Parent device not found.',
- }
- )
- device_bay_name = forms.CharField(
- help_text='Name of device bay',
- )
- cluster = forms.ModelChoiceField(
- queryset=Cluster.objects.all(),
- to_field_name='name',
- required=False,
- help_text='Virtualization cluster',
- error_messages={
- 'invalid_choice': 'Invalid cluster name.',
- }
- )
- class Meta(BaseDeviceCSVForm.Meta):
- fields = [
- 'name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag', 'status',
- 'parent', 'device_bay_name', 'cluster', 'comments',
- ]
- def clean(self):
- super().clean()
- parent = self.cleaned_data.get('parent')
- device_bay_name = self.cleaned_data.get('device_bay_name')
- # Validate device bay
- if parent and device_bay_name:
- try:
- self.instance.parent_bay = DeviceBay.objects.get(device=parent, name=device_bay_name)
- # Inherit site and rack from parent device
- self.instance.site = parent.site
- self.instance.rack = parent.rack
- except DeviceBay.DoesNotExist:
- raise forms.ValidationError("Parent device/bay ({} {}) not found".format(parent, device_bay_name))
- class DeviceBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Device.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- device_type = forms.ModelChoiceField(
- queryset=DeviceType.objects.all(),
- required=False,
- label='Type',
- widget=APISelect(
- api_url="/api/dcim/device-types/",
- display_field='display_name'
- )
- )
- device_role = forms.ModelChoiceField(
- queryset=DeviceRole.objects.all(),
- required=False,
- label='Role',
- widget=APISelect(
- api_url="/api/dcim/device-roles/"
- )
- )
- tenant = forms.ModelChoiceField(
- queryset=Tenant.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/tenancy/tenants/"
- )
- )
- platform = forms.ModelChoiceField(
- queryset=Platform.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/platforms/"
- )
- )
- status = forms.ChoiceField(
- choices=add_blank_choice(DEVICE_STATUS_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- serial = forms.CharField(
- max_length=50,
- required=False,
- label='Serial Number'
- )
- class Meta:
- nullable_fields = [
- 'tenant', 'platform', 'serial',
- ]
- class DeviceFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
- model = Device
- field_order = [
- 'q', 'region', 'site', 'rack_group_id', 'rack_id', 'status', 'role', 'tenant_group', 'tenant',
- 'manufacturer_id', 'device_type_id', 'mac_address', 'has_primary_ip',
- ]
- q = forms.CharField(
- required=False,
- label='Search'
- )
- region = FilterChoiceField(
- queryset=Region.objects.all(),
- to_field_name='slug',
- required=False,
- widget=APISelectMultiple(
- api_url="/api/dcim/regions/",
- value_field="slug",
- filter_for={
- 'site': 'region'
- }
- )
- )
- site = FilterChoiceField(
- queryset=Site.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/sites/",
- value_field="slug",
- filter_for={
- 'rack_group_id': 'site',
- 'rack_id': 'site',
- }
- )
- )
- rack_group_id = FilterChoiceField(
- queryset=RackGroup.objects.select_related(
- 'site'
- ),
- label='Rack group',
- widget=APISelectMultiple(
- api_url="/api/dcim/rack-groups/",
- filter_for={
- 'rack_id': 'rack_group_id',
- }
- )
- )
- rack_id = FilterChoiceField(
- queryset=Rack.objects.all(),
- label='Rack',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/dcim/racks/",
- null_option=True,
- )
- )
- role = FilterChoiceField(
- queryset=DeviceRole.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/device-roles/",
- value_field="slug",
- )
- )
- manufacturer_id = FilterChoiceField(
- queryset=Manufacturer.objects.all(),
- label='Manufacturer',
- widget=APISelectMultiple(
- api_url="/api/dcim/manufacturers/",
- filter_for={
- 'device_type_id': 'manufacturer_id',
- }
- )
- )
- device_type_id = FilterChoiceField(
- queryset=DeviceType.objects.select_related(
- 'manufacturer'
- ),
- label='Model',
- widget=APISelectMultiple(
- api_url="/api/dcim/device-types/",
- display_field="model",
- )
- )
- platform = FilterChoiceField(
- queryset=Platform.objects.all(),
- to_field_name='slug',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/dcim/platforms/",
- value_field="slug",
- null_option=True,
- )
- )
- status = forms.MultipleChoiceField(
- choices=DEVICE_STATUS_CHOICES,
- required=False,
- widget=StaticSelect2Multiple()
- )
- mac_address = forms.CharField(
- required=False,
- label='MAC address'
- )
- has_primary_ip = forms.NullBooleanField(
- required=False,
- label='Has a primary IP',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- virtual_chassis_member = forms.NullBooleanField(
- required=False,
- label='Virtual chassis member',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- console_ports = forms.NullBooleanField(
- required=False,
- label='Has console ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- console_server_ports = forms.NullBooleanField(
- required=False,
- label='Has console server ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- power_ports = forms.NullBooleanField(
- required=False,
- label='Has power ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- power_outlets = forms.NullBooleanField(
- required=False,
- label='Has power outlets',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- interfaces = forms.NullBooleanField(
- required=False,
- label='Has interfaces',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- pass_through_ports = forms.NullBooleanField(
- required=False,
- label='Has pass-through ports',
- widget=StaticSelect2(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- #
- # Bulk device component creation
- #
- class DeviceBulkAddComponentForm(BootstrapMixin, forms.Form):
- pk = forms.ModelMultipleChoiceField(
- queryset=Device.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- name_pattern = ExpandableNameField(
- label='Name'
- )
- class DeviceBulkAddInterfaceForm(DeviceBulkAddComponentForm):
- type = forms.ChoiceField(
- choices=IFACE_TYPE_CHOICES,
- widget=StaticSelect2()
- )
- enabled = forms.BooleanField(
- required=False,
- initial=True
- )
- mtu = forms.IntegerField(
- required=False,
- min_value=1,
- max_value=32767,
- label='MTU'
- )
- mgmt_only = forms.BooleanField(
- required=False,
- label='Management only'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- #
- # Console ports
- #
- class ConsolePortForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = ConsolePort
- fields = [
- 'device', 'name', 'description', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- }
- class ConsolePortCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- tags = TagField(
- required=False
- )
- #
- # Console server ports
- #
- class ConsoleServerPortForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = ConsoleServerPort
- fields = [
- 'device', 'name', 'description', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- }
- class ConsoleServerPortCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- tags = TagField(
- required=False
- )
- class ConsoleServerPortBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=ConsoleServerPort.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- class Meta:
- nullable_fields = [
- 'description',
- ]
- class ConsoleServerPortBulkRenameForm(BulkRenameForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=ConsoleServerPort.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- class ConsoleServerPortBulkDisconnectForm(ConfirmationForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=ConsoleServerPort.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- #
- # Power ports
- #
- class PowerPortForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = PowerPort
- fields = [
- 'device', 'name', 'maximum_draw', 'allocated_draw', 'description', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- }
- class PowerPortCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- maximum_draw = forms.IntegerField(
- min_value=1,
- required=False,
- help_text="Maximum draw in watts"
- )
- allocated_draw = forms.IntegerField(
- min_value=1,
- required=False,
- help_text="Allocated draw in watts"
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- tags = TagField(
- required=False
- )
- #
- # Power outlets
- #
- class PowerOutletForm(BootstrapMixin, forms.ModelForm):
- power_port = forms.ModelChoiceField(
- queryset=PowerPort.objects.all(),
- required=False
- )
- tags = TagField(
- required=False
- )
- class Meta:
- model = PowerOutlet
- fields = [
- 'device', 'name', 'power_port', 'feed_leg', 'description', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- }
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Limit power_port choices to the local device
- if hasattr(self.instance, 'device'):
- self.fields['power_port'].queryset = PowerPort.objects.filter(
- device=self.instance.device
- )
- class PowerOutletCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- power_port = forms.ModelChoiceField(
- queryset=PowerPort.objects.all(),
- required=False
- )
- feed_leg = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_LEG_CHOICES),
- required=False
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- tags = TagField(
- required=False
- )
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Limit power_port choices to those on the parent device
- self.fields['power_port'].queryset = PowerPort.objects.filter(device=self.parent)
- class PowerOutletBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=PowerOutlet.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- feed_leg = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_LEG_CHOICES),
- required=False,
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- class Meta:
- nullable_fields = [
- 'feed_leg', 'description',
- ]
- class PowerOutletBulkRenameForm(BulkRenameForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=PowerOutlet.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- class PowerOutletBulkDisconnectForm(ConfirmationForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=PowerOutlet.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- #
- # Interfaces
- #
- class InterfaceForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = Interface
- fields = [
- 'device', 'name', 'type', 'enabled', 'lag', 'mac_address', 'mtu', 'mgmt_only', 'description',
- 'mode', 'untagged_vlan', 'tagged_vlans', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- 'type': StaticSelect2(),
- 'lag': StaticSelect2(),
- 'mode': StaticSelect2(),
- }
- labels = {
- 'mode': '802.1Q Mode',
- }
- help_texts = {
- 'mode': INTERFACE_MODE_HELP_TEXT,
- }
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Limit LAG choices to interfaces belonging to this device (or VC master)
- if self.is_bound:
- device = Device.objects.get(pk=self.data['device'])
- self.fields['lag'].queryset = Interface.objects.filter(
- device__in=[device, device.get_vc_master()], type=IFACE_TYPE_LAG
- )
- else:
- device = self.instance.device
- self.fields['lag'].queryset = Interface.objects.filter(
- device__in=[self.instance.device, self.instance.device.get_vc_master()], type=IFACE_TYPE_LAG
- )
- def clean(self):
- super().clean()
- # Validate VLAN assignments
- tagged_vlans = self.cleaned_data['tagged_vlans']
- # Untagged interfaces cannot be assigned tagged VLANs
- if self.cleaned_data['mode'] == IFACE_MODE_ACCESS and tagged_vlans:
- raise forms.ValidationError({
- 'mode': "An access interface cannot have tagged VLANs assigned."
- })
- # Remove all tagged VLAN assignments from "tagged all" interfaces
- elif self.cleaned_data['mode'] == IFACE_MODE_TAGGED_ALL:
- self.cleaned_data['tagged_vlans'] = []
- class InterfaceAssignVLANsForm(BootstrapMixin, forms.ModelForm):
- vlans = forms.MultipleChoiceField(
- choices=[],
- label='VLANs',
- widget=StaticSelect2Multiple(
- attrs={
- 'size': 20,
- }
- )
- )
- tagged = forms.BooleanField(
- required=False,
- initial=True
- )
- class Meta:
- model = Interface
- fields = []
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- if self.instance.mode == IFACE_MODE_ACCESS:
- self.initial['tagged'] = False
- # Find all VLANs already assigned to the interface for exclusion from the list
- assigned_vlans = [v.pk for v in self.instance.tagged_vlans.all()]
- if self.instance.untagged_vlan is not None:
- assigned_vlans.append(self.instance.untagged_vlan.pk)
- # Compile VLAN choices
- vlan_choices = []
- # Add non-grouped global VLANs
- global_vlans = VLAN.objects.filter(site=None, group=None).exclude(pk__in=assigned_vlans)
- vlan_choices.append(
- ('Global', [(vlan.pk, vlan) for vlan in global_vlans])
- )
- # Add grouped global VLANs
- for group in VLANGroup.objects.filter(site=None):
- global_group_vlans = VLAN.objects.filter(group=group).exclude(pk__in=assigned_vlans)
- vlan_choices.append(
- (group.name, [(vlan.pk, vlan) for vlan in global_group_vlans])
- )
- site = getattr(self.instance.parent, 'site', None)
- if site is not None:
- # Add non-grouped site VLANs
- site_vlans = VLAN.objects.filter(site=site, group=None).exclude(pk__in=assigned_vlans)
- vlan_choices.append((site.name, [(vlan.pk, vlan) for vlan in site_vlans]))
- # Add grouped site VLANs
- for group in VLANGroup.objects.filter(site=site):
- site_group_vlans = VLAN.objects.filter(group=group).exclude(pk__in=assigned_vlans)
- vlan_choices.append((
- '{} / {}'.format(group.site.name, group.name),
- [(vlan.pk, vlan) for vlan in site_group_vlans]
- ))
- self.fields['vlans'].choices = vlan_choices
- def clean(self):
- super().clean()
- # Only untagged VLANs permitted on an access interface
- if self.instance.mode == IFACE_MODE_ACCESS and len(self.cleaned_data['vlans']) > 1:
- raise forms.ValidationError("Only one VLAN may be assigned to an access interface.")
- # 'tagged' is required if more than one VLAN is selected
- if not self.cleaned_data['tagged'] and len(self.cleaned_data['vlans']) > 1:
- raise forms.ValidationError("Only one untagged VLAN may be selected.")
- def save(self, *args, **kwargs):
- if self.cleaned_data['tagged']:
- for vlan in self.cleaned_data['vlans']:
- self.instance.tagged_vlans.add(vlan)
- else:
- self.instance.untagged_vlan_id = self.cleaned_data['vlans'][0]
- return super().save(*args, **kwargs)
- class InterfaceCreateForm(ComponentForm, forms.Form):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- type = forms.ChoiceField(
- choices=IFACE_TYPE_CHOICES,
- widget=StaticSelect2(),
- )
- enabled = forms.BooleanField(
- required=False
- )
- lag = forms.ModelChoiceField(
- queryset=Interface.objects.all(),
- required=False,
- label='Parent LAG',
- widget=StaticSelect2(),
- )
- mtu = forms.IntegerField(
- required=False,
- min_value=1,
- max_value=32767,
- label='MTU'
- )
- mac_address = forms.CharField(
- required=False,
- label='MAC Address'
- )
- mgmt_only = forms.BooleanField(
- required=False,
- label='Management only',
- help_text='This interface is used only for out-of-band management'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- mode = forms.ChoiceField(
- choices=add_blank_choice(IFACE_MODE_CHOICES),
- required=False,
- widget=StaticSelect2(),
- )
- tags = TagField(
- required=False
- )
- def __init__(self, *args, **kwargs):
- # Set interfaces enabled by default
- kwargs['initial'] = kwargs.get('initial', {}).copy()
- kwargs['initial'].update({'enabled': True})
- super().__init__(*args, **kwargs)
- # Limit LAG choices to interfaces belonging to this device (or its VC master)
- if self.parent is not None:
- self.fields['lag'].queryset = Interface.objects.filter(
- device__in=[self.parent, self.parent.get_vc_master()], type=IFACE_TYPE_LAG
- )
- else:
- self.fields['lag'].queryset = Interface.objects.none()
- class InterfaceBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Interface.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(IFACE_TYPE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- enabled = forms.NullBooleanField(
- required=False,
- widget=BulkEditNullBooleanSelect()
- )
- lag = forms.ModelChoiceField(
- queryset=Interface.objects.all(),
- required=False,
- label='Parent LAG',
- widget=StaticSelect2()
- )
- mac_address = forms.CharField(
- required=False,
- label='MAC Address'
- )
- mtu = forms.IntegerField(
- required=False,
- min_value=1,
- max_value=32767,
- label='MTU'
- )
- mgmt_only = forms.NullBooleanField(
- required=False,
- widget=BulkEditNullBooleanSelect(),
- label='Management only'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- mode = forms.ChoiceField(
- choices=add_blank_choice(IFACE_MODE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- class Meta:
- nullable_fields = [
- 'lag', 'mac_address', 'mtu', 'description', 'mode',
- ]
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Limit LAG choices to interfaces which belong to the parent device (or VC master)
- device = self.parent_obj
- if device is not None:
- self.fields['lag'].queryset = Interface.objects.filter(
- device__in=[device, device.get_vc_master()],
- type=IFACE_TYPE_LAG
- )
- else:
- self.fields['lag'].choices = []
- class InterfaceBulkRenameForm(BulkRenameForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Interface.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- class InterfaceBulkDisconnectForm(ConfirmationForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Interface.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- #
- # Front pass-through ports
- #
- class FrontPortForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = FrontPort
- fields = [
- 'device', 'name', 'type', 'rear_port', 'rear_port_position', 'description', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- 'type': StaticSelect2(),
- 'rear_port': StaticSelect2(),
- }
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Limit RearPort choices to the local device
- if hasattr(self.instance, 'device'):
- self.fields['rear_port'].queryset = self.fields['rear_port'].queryset.filter(
- device=self.instance.device
- )
- # TODO: Merge with FrontPortTemplateCreateForm to remove duplicate logic
- class FrontPortCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- type = forms.ChoiceField(
- choices=PORT_TYPE_CHOICES,
- widget=StaticSelect2(),
- )
- rear_port_set = forms.MultipleChoiceField(
- choices=[],
- label='Rear ports',
- help_text='Select one rear port assignment for each front port being created.',
- )
- description = forms.CharField(
- required=False
- )
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Determine which rear port positions are occupied. These will be excluded from the list of available mappings.
- occupied_port_positions = [
- (front_port.rear_port_id, front_port.rear_port_position)
- for front_port in self.parent.frontports.all()
- ]
- # Populate rear port choices
- choices = []
- rear_ports = RearPort.objects.filter(device=self.parent)
- for rear_port in rear_ports:
- for i in range(1, rear_port.positions + 1):
- if (rear_port.pk, i) not in occupied_port_positions:
- choices.append(
- ('{}:{}'.format(rear_port.pk, i), '{}:{}'.format(rear_port.name, i))
- )
- self.fields['rear_port_set'].choices = choices
- def clean(self):
- # Validate that the number of ports being created equals the number of selected (rear port, position) tuples
- front_port_count = len(self.cleaned_data['name_pattern'])
- rear_port_count = len(self.cleaned_data['rear_port_set'])
- if front_port_count != rear_port_count:
- raise forms.ValidationError({
- 'rear_port_set': 'The provided name pattern will create {} ports, however {} rear port assignments '
- 'were selected. These counts must match.'.format(front_port_count, rear_port_count)
- })
- def get_iterative_data(self, iteration):
- # Assign rear port and position from selected set
- rear_port, position = self.cleaned_data['rear_port_set'][iteration].split(':')
- return {
- 'rear_port': int(rear_port),
- 'rear_port_position': int(position),
- }
- class FrontPortBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=FrontPort.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(PORT_TYPE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- class Meta:
- nullable_fields = [
- 'description',
- ]
- class FrontPortBulkRenameForm(BulkRenameForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=FrontPort.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- class FrontPortBulkDisconnectForm(ConfirmationForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=FrontPort.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- #
- # Rear pass-through ports
- #
- class RearPortForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = RearPort
- fields = [
- 'device', 'name', 'type', 'positions', 'description', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- 'type': StaticSelect2(),
- }
- class RearPortCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- type = forms.ChoiceField(
- choices=PORT_TYPE_CHOICES,
- widget=StaticSelect2(),
- )
- positions = forms.IntegerField(
- min_value=1,
- max_value=64,
- initial=1,
- help_text='The number of front ports which may be mapped to each rear port'
- )
- description = forms.CharField(
- required=False
- )
- class RearPortBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=RearPort.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(PORT_TYPE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- class Meta:
- nullable_fields = [
- 'description',
- ]
- class RearPortBulkRenameForm(BulkRenameForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=RearPort.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- class RearPortBulkDisconnectForm(ConfirmationForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=RearPort.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- #
- # Cables
- #
- class ConnectCableToDeviceForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
- """
- Base form for connecting a Cable to a Device component
- """
- termination_b_site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- label='Site',
- required=False,
- widget=APISelect(
- api_url='/api/dcim/sites/',
- filter_for={
- 'termination_b_rack': 'site_id',
- 'termination_b_device': 'site_id',
- }
- )
- )
- termination_b_rack = ChainedModelChoiceField(
- queryset=Rack.objects.all(),
- chains=(
- ('site', 'termination_b_site'),
- ),
- label='Rack',
- required=False,
- widget=APISelect(
- api_url='/api/dcim/racks/',
- filter_for={
- 'termination_b_device': 'rack_id',
- },
- attrs={
- 'nullable': 'true',
- }
- )
- )
- termination_b_device = ChainedModelChoiceField(
- queryset=Device.objects.all(),
- chains=(
- ('site', 'termination_b_site'),
- ('rack', 'termination_b_rack'),
- ),
- label='Device',
- required=False,
- widget=APISelect(
- api_url='/api/dcim/devices/',
- display_field='display_name',
- filter_for={
- 'termination_b_id': 'device_id',
- }
- )
- )
- class Meta:
- model = Cable
- fields = [
- 'termination_b_site', 'termination_b_rack', 'termination_b_device', 'termination_b_id', 'type', 'status',
- 'label', 'color', 'length', 'length_unit',
- ]
- class ConnectCableToConsolePortForm(ConnectCableToDeviceForm):
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/console-ports/',
- disabled_indicator='cable',
- )
- )
- class ConnectCableToConsoleServerPortForm(ConnectCableToDeviceForm):
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/console-server-ports/',
- disabled_indicator='cable',
- )
- )
- class ConnectCableToPowerPortForm(ConnectCableToDeviceForm):
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/power-ports/',
- disabled_indicator='cable',
- )
- )
- class ConnectCableToPowerOutletForm(ConnectCableToDeviceForm):
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/power-outlets/',
- disabled_indicator='cable',
- )
- )
- class ConnectCableToInterfaceForm(ConnectCableToDeviceForm):
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/interfaces/',
- disabled_indicator='cable',
- additional_query_params={
- 'kind': 'physical',
- }
- )
- )
- class ConnectCableToFrontPortForm(ConnectCableToDeviceForm):
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/front-ports/',
- disabled_indicator='cable',
- )
- )
- class ConnectCableToRearPortForm(ConnectCableToDeviceForm):
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/rear-ports/',
- disabled_indicator='cable',
- )
- )
- class ConnectCableToCircuitTerminationForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
- termination_b_provider = forms.ModelChoiceField(
- queryset=Provider.objects.all(),
- label='Provider',
- widget=APISelect(
- api_url='/api/circuits/providers/',
- filter_for={
- 'termination_b_circuit': 'provider_id',
- }
- )
- )
- termination_b_site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- label='Site',
- required=False,
- widget=APISelect(
- api_url='/api/dcim/sites/',
- filter_for={
- 'termination_b_circuit': 'site_id',
- }
- )
- )
- termination_b_circuit = ChainedModelChoiceField(
- queryset=Circuit.objects.all(),
- chains=(
- ('provider', 'termination_b_provider'),
- ),
- label='Circuit',
- widget=APISelect(
- api_url='/api/circuits/circuits/',
- display_field='cid',
- filter_for={
- 'termination_b_id': 'circuit_id',
- }
- )
- )
- termination_b_id = forms.IntegerField(
- label='Side',
- widget=APISelect(
- api_url='/api/circuits/circuit-terminations/',
- disabled_indicator='cable',
- display_field='term_side'
- )
- )
- class Meta:
- model = Cable
- fields = [
- 'termination_b_provider', 'termination_b_site', 'termination_b_circuit', 'termination_b_id', 'type',
- 'status', 'label', 'color', 'length', 'length_unit',
- ]
- class ConnectCableToPowerFeedForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
- termination_b_site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- label='Site',
- widget=APISelect(
- api_url='/api/dcim/sites/',
- display_field='cid',
- filter_for={
- 'termination_b_rackgroup': 'site_id',
- 'termination_b_powerpanel': 'site_id',
- }
- )
- )
- termination_b_rackgroup = ChainedModelChoiceField(
- queryset=RackGroup.objects.all(),
- label='Rack Group',
- chains=(
- ('site', 'termination_b_site'),
- ),
- required=False,
- widget=APISelect(
- api_url='/api/dcim/rack-groups/',
- display_field='cid',
- filter_for={
- 'termination_b_powerpanel': 'rackgroup_id',
- }
- )
- )
- termination_b_powerpanel = ChainedModelChoiceField(
- queryset=PowerPanel.objects.all(),
- chains=(
- ('site', 'termination_b_site'),
- ('rack_group', 'termination_b_rackgroup'),
- ),
- label='Power Panel',
- widget=APISelect(
- api_url='/api/dcim/power-panels/',
- filter_for={
- 'termination_b_id': 'power_panel_id',
- }
- )
- )
- termination_b_id = forms.IntegerField(
- label='Name',
- widget=APISelect(
- api_url='/api/dcim/power-feeds/',
- )
- )
- class Meta:
- model = Cable
- fields = [
- 'termination_b_rackgroup', 'termination_b_powerpanel', 'termination_b_id', 'type', 'status', 'label',
- 'color', 'length', 'length_unit',
- ]
- class CableForm(BootstrapMixin, forms.ModelForm):
- class Meta:
- model = Cable
- fields = [
- 'type', 'status', 'label', 'color', 'length', 'length_unit',
- ]
- class CableCSVForm(forms.ModelForm):
- # Termination A
- side_a_device = FlexibleModelChoiceField(
- queryset=Device.objects.all(),
- to_field_name='name',
- help_text='Side A device name or ID',
- error_messages={
- 'invalid_choice': 'Side A device not found',
- }
- )
- side_a_type = forms.ModelChoiceField(
- queryset=ContentType.objects.all(),
- limit_choices_to={
- 'model__in': CABLE_TERMINATION_TYPES,
- },
- to_field_name='model',
- help_text='Side A type'
- )
- side_a_name = forms.CharField(
- help_text='Side A component'
- )
- # Termination B
- side_b_device = FlexibleModelChoiceField(
- queryset=Device.objects.all(),
- to_field_name='name',
- help_text='Side B device name or ID',
- error_messages={
- 'invalid_choice': 'Side B device not found',
- }
- )
- side_b_type = forms.ModelChoiceField(
- queryset=ContentType.objects.all(),
- limit_choices_to={
- 'model__in': CABLE_TERMINATION_TYPES,
- },
- to_field_name='model',
- help_text='Side B type'
- )
- side_b_name = forms.CharField(
- help_text='Side B component'
- )
- # Cable attributes
- status = CSVChoiceField(
- choices=CONNECTION_STATUS_CHOICES,
- required=False,
- help_text='Connection status'
- )
- type = CSVChoiceField(
- choices=CABLE_TYPE_CHOICES,
- required=False,
- help_text='Cable type'
- )
- length_unit = CSVChoiceField(
- choices=CABLE_LENGTH_UNIT_CHOICES,
- required=False,
- help_text='Length unit'
- )
- class Meta:
- model = Cable
- fields = [
- 'side_a_device', 'side_a_type', 'side_a_name', 'side_b_device', 'side_b_type', 'side_b_name', 'type',
- 'status', 'label', 'color', 'length', 'length_unit',
- ]
- help_texts = {
- 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
- }
- # TODO: Merge the clean() methods for either end
- def clean_side_a_name(self):
- device = self.cleaned_data.get('side_a_device')
- content_type = self.cleaned_data.get('side_a_type')
- name = self.cleaned_data.get('side_a_name')
- if not device or not content_type or not name:
- return None
- model = content_type.model_class()
- try:
- termination_object = model.objects.get(
- device=device,
- name=name
- )
- if termination_object.cable is not None:
- raise forms.ValidationError(
- "Side A: {} {} is already connected".format(device, termination_object)
- )
- except ObjectDoesNotExist:
- raise forms.ValidationError(
- "A side termination not found: {} {}".format(device, name)
- )
- self.instance.termination_a = termination_object
- return termination_object
- def clean_side_b_name(self):
- device = self.cleaned_data.get('side_b_device')
- content_type = self.cleaned_data.get('side_b_type')
- name = self.cleaned_data.get('side_b_name')
- if not device or not content_type or not name:
- return None
- model = content_type.model_class()
- try:
- termination_object = model.objects.get(
- device=device,
- name=name
- )
- if termination_object.cable is not None:
- raise forms.ValidationError(
- "Side B: {} {} is already connected".format(device, termination_object)
- )
- except ObjectDoesNotExist:
- raise forms.ValidationError(
- "B side termination not found: {} {}".format(device, name)
- )
- self.instance.termination_b = termination_object
- return termination_object
- def clean_length_unit(self):
- # Avoid trying to save as NULL
- length_unit = self.cleaned_data.get('length_unit', None)
- return length_unit if length_unit is not None else ''
- class CableBulkEditForm(BootstrapMixin, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=Cable.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(CABLE_TYPE_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- status = forms.ChoiceField(
- choices=add_blank_choice(CONNECTION_STATUS_CHOICES),
- required=False,
- widget=StaticSelect2(),
- initial=''
- )
- label = forms.CharField(
- max_length=100,
- required=False
- )
- color = forms.CharField(
- max_length=6,
- required=False,
- widget=ColorSelect()
- )
- length = forms.IntegerField(
- min_value=1,
- required=False
- )
- length_unit = forms.ChoiceField(
- choices=add_blank_choice(CABLE_LENGTH_UNIT_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- class Meta:
- nullable_fields = [
- 'type', 'status', 'label', 'color', 'length',
- ]
- def clean(self):
- # Validate length/unit
- length = self.cleaned_data.get('length')
- length_unit = self.cleaned_data.get('length_unit')
- if length and not length_unit:
- raise forms.ValidationError({
- 'length_unit': "Must specify a unit when setting length"
- })
- class CableFilterForm(BootstrapMixin, forms.Form):
- model = Cable
- q = forms.CharField(
- required=False,
- label='Search'
- )
- type = forms.MultipleChoiceField(
- choices=add_blank_choice(CABLE_TYPE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- status = forms.ChoiceField(
- required=False,
- choices=add_blank_choice(CONNECTION_STATUS_CHOICES),
- widget=StaticSelect2()
- )
- color = forms.CharField(
- max_length=6,
- required=False,
- widget=ColorSelect()
- )
- device = forms.CharField(
- required=False,
- label='Device name'
- )
- #
- # Device bays
- #
- class DeviceBayForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = DeviceBay
- fields = [
- 'device', 'name', 'description', 'tags',
- ]
- widgets = {
- 'device': forms.HiddenInput(),
- }
- class DeviceBayCreateForm(ComponentForm):
- name_pattern = ExpandableNameField(
- label='Name'
- )
- tags = TagField(
- required=False
- )
- class PopulateDeviceBayForm(BootstrapMixin, forms.Form):
- installed_device = forms.ModelChoiceField(
- queryset=Device.objects.all(),
- label='Child Device',
- help_text="Child devices must first be created and assigned to the site/rack of the parent device.",
- widget=StaticSelect2(),
- )
- def __init__(self, device_bay, *args, **kwargs):
- super().__init__(*args, **kwargs)
- self.fields['installed_device'].queryset = Device.objects.filter(
- site=device_bay.device.site,
- rack=device_bay.device.rack,
- parent_bay__isnull=True,
- device_type__u_height=0,
- device_type__subdevice_role=SUBDEVICE_ROLE_CHILD
- ).exclude(pk=device_bay.device.pk)
- class DeviceBayBulkRenameForm(BulkRenameForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=DeviceBay.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- #
- # Connections
- #
- class ConsoleConnectionFilterForm(BootstrapMixin, forms.Form):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- to_field_name='slug'
- )
- device = forms.CharField(
- required=False,
- label='Device name'
- )
- class PowerConnectionFilterForm(BootstrapMixin, forms.Form):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- to_field_name='slug'
- )
- device = forms.CharField(
- required=False,
- label='Device name'
- )
- class InterfaceConnectionFilterForm(BootstrapMixin, forms.Form):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- to_field_name='slug'
- )
- device = forms.CharField(
- required=False,
- label='Device name'
- )
- #
- # Inventory items
- #
- class InventoryItemForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = InventoryItem
- fields = [
- 'name', 'manufacturer', 'part_id', 'serial', 'asset_tag', 'description', 'tags',
- ]
- widgets = {
- 'manufacturer': APISelect(
- api_url="/api/dcim/manufacturers/"
- )
- }
- class InventoryItemCSVForm(forms.ModelForm):
- device = FlexibleModelChoiceField(
- queryset=Device.objects.all(),
- to_field_name='name',
- help_text='Device name or ID',
- error_messages={
- 'invalid_choice': 'Device not found.',
- }
- )
- manufacturer = forms.ModelChoiceField(
- queryset=Manufacturer.objects.all(),
- to_field_name='name',
- required=False,
- help_text='Manufacturer name',
- error_messages={
- 'invalid_choice': 'Invalid manufacturer.',
- }
- )
- class Meta:
- model = InventoryItem
- fields = InventoryItem.csv_headers
- class InventoryItemBulkEditForm(BootstrapMixin, BulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=InventoryItem.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- manufacturer = forms.ModelChoiceField(
- queryset=Manufacturer.objects.all(),
- required=False
- )
- part_id = forms.CharField(
- max_length=50,
- required=False,
- label='Part ID'
- )
- description = forms.CharField(
- max_length=100,
- required=False
- )
- class Meta:
- nullable_fields = [
- 'manufacturer', 'part_id', 'description',
- ]
- class InventoryItemFilterForm(BootstrapMixin, forms.Form):
- model = InventoryItem
- q = forms.CharField(
- required=False,
- label='Search'
- )
- device = forms.CharField(
- required=False,
- label='Device name'
- )
- manufacturer = FilterChoiceField(
- queryset=Manufacturer.objects.all(),
- to_field_name='slug',
- null_label='-- None --'
- )
- discovered = forms.NullBooleanField(
- required=False,
- widget=forms.Select(
- choices=BOOLEAN_WITH_BLANK_CHOICES
- )
- )
- #
- # Virtual chassis
- #
- class DeviceSelectionForm(forms.Form):
- pk = forms.ModelMultipleChoiceField(
- queryset=Device.objects.all(),
- widget=forms.MultipleHiddenInput()
- )
- class VirtualChassisForm(BootstrapMixin, forms.ModelForm):
- tags = TagField(
- required=False
- )
- class Meta:
- model = VirtualChassis
- fields = [
- 'master', 'domain', 'tags',
- ]
- widgets = {
- 'master': SelectWithPK(),
- }
- class BaseVCMemberFormSet(forms.BaseModelFormSet):
- def clean(self):
- super().clean()
- # Check for duplicate VC position values
- vc_position_list = []
- for form in self.forms:
- vc_position = form.cleaned_data.get('vc_position')
- if vc_position:
- if vc_position in vc_position_list:
- error_msg = 'A virtual chassis member already exists in position {}.'.format(vc_position)
- form.add_error('vc_position', error_msg)
- vc_position_list.append(vc_position)
- class DeviceVCMembershipForm(forms.ModelForm):
- class Meta:
- model = Device
- fields = [
- 'vc_position', 'vc_priority',
- ]
- labels = {
- 'vc_position': 'Position',
- 'vc_priority': 'Priority',
- }
- def __init__(self, validate_vc_position=False, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Require VC position (only required when the Device is a VirtualChassis member)
- self.fields['vc_position'].required = True
- # Validation of vc_position is optional. This is only required when adding a new member to an existing
- # VirtualChassis. Otherwise, vc_position validation is handled by BaseVCMemberFormSet.
- self.validate_vc_position = validate_vc_position
- def clean_vc_position(self):
- vc_position = self.cleaned_data['vc_position']
- if self.validate_vc_position:
- conflicting_members = Device.objects.filter(
- virtual_chassis=self.instance.virtual_chassis,
- vc_position=vc_position
- )
- if conflicting_members.exists():
- raise forms.ValidationError(
- 'A virtual chassis member already exists in position {}.'.format(vc_position)
- )
- return vc_position
- class VCMemberSelectForm(BootstrapMixin, ChainedFieldsMixin, forms.Form):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- label='Site',
- required=False,
- widget=APISelect(
- api_url="/api/dcim/sites/",
- filter_for={
- 'rack': 'site_id',
- 'device': 'site_id',
- }
- )
- )
- rack = ChainedModelChoiceField(
- queryset=Rack.objects.all(),
- chains=(
- ('site', 'site'),
- ),
- label='Rack',
- required=False,
- widget=APISelect(
- api_url='/api/dcim/racks/',
- filter_for={
- 'device': 'rack_id'
- },
- attrs={
- 'nullable': 'true',
- }
- )
- )
- device = ChainedModelChoiceField(
- queryset=Device.objects.filter(
- virtual_chassis__isnull=True
- ),
- chains=(
- ('site', 'site'),
- ('rack', 'rack'),
- ),
- label='Device',
- widget=APISelect(
- api_url='/api/dcim/devices/',
- display_field='display_name',
- disabled_indicator='virtual_chassis'
- )
- )
- def clean_device(self):
- device = self.cleaned_data['device']
- if device.virtual_chassis is not None:
- raise forms.ValidationError(
- "Device {} is already assigned to a virtual chassis.".format(device)
- )
- return device
- class VirtualChassisFilterForm(BootstrapMixin, CustomFieldFilterForm):
- model = VirtualChassis
- q = forms.CharField(
- required=False,
- label='Search'
- )
- site = FilterChoiceField(
- queryset=Site.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/sites/",
- value_field="slug",
- )
- )
- tenant_group = FilterChoiceField(
- queryset=TenantGroup.objects.all(),
- to_field_name='slug',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/tenancy/tenant-groups/",
- value_field="slug",
- null_option=True,
- filter_for={
- 'tenant': 'group'
- }
- )
- )
- tenant = FilterChoiceField(
- queryset=Tenant.objects.all(),
- to_field_name='slug',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/tenancy/tenants/",
- value_field="slug",
- null_option=True,
- )
- )
- #
- # Power panels
- #
- class PowerPanelForm(BootstrapMixin, forms.ModelForm):
- rack_group = ChainedModelChoiceField(
- queryset=RackGroup.objects.all(),
- chains=(
- ('site', 'site'),
- ),
- required=False,
- widget=APISelect(
- api_url='/api/dcim/rack-groups/',
- )
- )
- class Meta:
- model = PowerPanel
- fields = [
- 'site', 'rack_group', 'name',
- ]
- widgets = {
- 'site': APISelect(
- api_url="/api/dcim/sites/",
- filter_for={
- 'rack_group': 'site_id',
- }
- ),
- }
- class PowerPanelCSVForm(forms.ModelForm):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- to_field_name='name',
- help_text='Name of parent site',
- error_messages={
- 'invalid_choice': 'Site not found.',
- }
- )
- rack_group_name = forms.CharField(
- required=False,
- help_text="Rack group name (optional)"
- )
- class Meta:
- model = PowerPanel
- fields = PowerPanel.csv_headers
- def clean(self):
- super().clean()
- site = self.cleaned_data.get('site')
- rack_group_name = self.cleaned_data.get('rack_group_name')
- # Validate rack group
- if rack_group_name:
- try:
- self.instance.rack_group = RackGroup.objects.get(site=site, name=rack_group_name)
- except RackGroup.DoesNotExist:
- raise forms.ValidationError(
- "Rack group {} not found in site {}".format(rack_group_name, site)
- )
- class PowerPanelFilterForm(BootstrapMixin, CustomFieldFilterForm):
- model = PowerPanel
- q = forms.CharField(
- required=False,
- label='Search'
- )
- site = FilterChoiceField(
- queryset=Site.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/sites/",
- value_field="slug",
- filter_for={
- 'rack_group_id': 'site',
- }
- )
- )
- rack_group_id = FilterChoiceField(
- queryset=RackGroup.objects.all(),
- label='Rack group (ID)',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/dcim/rack-groups/",
- null_option=True,
- )
- )
- #
- # Power feeds
- #
- class PowerFeedForm(BootstrapMixin, CustomFieldForm):
- site = ChainedModelChoiceField(
- queryset=Site.objects.all(),
- required=False,
- widget=APISelect(
- api_url='/api/dcim/sites/',
- filter_for={
- 'power_panel': 'site_id',
- 'rack': 'site_id',
- }
- )
- )
- comments = CommentField()
- tags = TagField(
- required=False
- )
- class Meta:
- model = PowerFeed
- fields = [
- 'site', 'power_panel', 'rack', 'name', 'status', 'type', 'supply', 'phase', 'voltage', 'amperage',
- 'max_utilization', 'comments', 'tags',
- ]
- widgets = {
- 'power_panel': APISelect(
- api_url="/api/dcim/power-panels/"
- ),
- 'rack': APISelect(
- api_url="/api/dcim/racks/"
- ),
- 'status': StaticSelect2(),
- 'type': StaticSelect2(),
- 'supply': StaticSelect2(),
- 'phase': StaticSelect2(),
- }
- def __init__(self, *args, **kwargs):
- super().__init__(*args, **kwargs)
- # Initialize site field
- if self.instance and hasattr(self.instance, 'power_panel'):
- self.initial['site'] = self.instance.power_panel.site
- class PowerFeedCSVForm(forms.ModelForm):
- site = forms.ModelChoiceField(
- queryset=Site.objects.all(),
- to_field_name='name',
- help_text='Name of parent site',
- error_messages={
- 'invalid_choice': 'Site not found.',
- }
- )
- panel_name = forms.ModelChoiceField(
- queryset=PowerPanel.objects.all(),
- to_field_name='name',
- help_text='Name of upstream power panel',
- error_messages={
- 'invalid_choice': 'Power panel not found.',
- }
- )
- rack_group = forms.CharField(
- required=False,
- help_text="Rack group name (optional)"
- )
- rack_name = forms.CharField(
- required=False,
- help_text="Rack name (optional)"
- )
- status = CSVChoiceField(
- choices=POWERFEED_STATUS_CHOICES,
- required=False,
- help_text='Operational status'
- )
- type = CSVChoiceField(
- choices=POWERFEED_TYPE_CHOICES,
- required=False,
- help_text='Primary or redundant'
- )
- supply = CSVChoiceField(
- choices=POWERFEED_SUPPLY_CHOICES,
- required=False,
- help_text='AC/DC'
- )
- phase = CSVChoiceField(
- choices=POWERFEED_PHASE_CHOICES,
- required=False,
- help_text='Single or three-phase'
- )
- class Meta:
- model = PowerFeed
- fields = PowerFeed.csv_headers
- def clean(self):
- super().clean()
- site = self.cleaned_data.get('site')
- panel_name = self.cleaned_data.get('panel_name')
- rack_group = self.cleaned_data.get('rack_group')
- rack_name = self.cleaned_data.get('rack_name')
- # Validate power panel
- if panel_name:
- try:
- self.instance.power_panel = PowerPanel.objects.get(site=site, name=panel_name)
- except Rack.DoesNotExist:
- raise forms.ValidationError(
- "Power panel {} not found in site {}".format(panel_name, site)
- )
- # Validate rack
- if rack_name:
- try:
- self.instance.rack = Rack.objects.get(site=site, group__name=rack_group, name=rack_name)
- except Rack.DoesNotExist:
- raise forms.ValidationError(
- "Rack {} not found in site {}, group {}".format(rack_name, site, rack_group)
- )
- class PowerFeedBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
- pk = forms.ModelMultipleChoiceField(
- queryset=PowerFeed.objects.all(),
- widget=forms.MultipleHiddenInput
- )
- powerpanel = forms.ModelChoiceField(
- queryset=PowerPanel.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/power-panels/",
- filter_for={
- 'rackgroup': 'site_id',
- }
- )
- )
- rack = forms.ModelChoiceField(
- queryset=Rack.objects.all(),
- required=False,
- widget=APISelect(
- api_url="/api/dcim/racks",
- )
- )
- status = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_STATUS_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_TYPE_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- supply = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_SUPPLY_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- phase = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_PHASE_CHOICES),
- required=False,
- initial='',
- widget=StaticSelect2()
- )
- voltage = forms.IntegerField(
- required=False
- )
- amperage = forms.IntegerField(
- required=False
- )
- max_utilization = forms.IntegerField(
- required=False
- )
- comments = forms.CharField(
- required=False
- )
- class Meta:
- nullable_fields = [
- 'rackgroup', 'comments',
- ]
- class PowerFeedFilterForm(BootstrapMixin, CustomFieldFilterForm):
- model = PowerFeed
- q = forms.CharField(
- required=False,
- label='Search'
- )
- site = FilterChoiceField(
- queryset=Site.objects.all(),
- to_field_name='slug',
- widget=APISelectMultiple(
- api_url="/api/dcim/sites/",
- value_field="slug",
- filter_for={
- 'power_panel_id': 'site',
- 'rack_id': 'site',
- }
- )
- )
- power_panel_id = FilterChoiceField(
- queryset=PowerPanel.objects.all(),
- label='Power panel',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/dcim/power-panels/",
- null_option=True,
- )
- )
- rack_id = FilterChoiceField(
- queryset=Rack.objects.all(),
- label='Rack',
- null_label='-- None --',
- widget=APISelectMultiple(
- api_url="/api/dcim/racks/",
- null_option=True,
- )
- )
- status = forms.MultipleChoiceField(
- choices=POWERFEED_STATUS_CHOICES,
- required=False,
- widget=StaticSelect2Multiple()
- )
- type = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_TYPE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- supply = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_SUPPLY_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- phase = forms.ChoiceField(
- choices=add_blank_choice(POWERFEED_PHASE_CHOICES),
- required=False,
- widget=StaticSelect2()
- )
- voltage = forms.IntegerField(
- required=False
- )
- amperage = forms.IntegerField(
- required=False
- )
- max_utilization = forms.IntegerField(
- required=False
- )
|