forms.py 125 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512
  1. import re
  2. from django import forms
  3. from django.contrib.auth.models import User
  4. from django.contrib.contenttypes.models import ContentType
  5. from django.contrib.postgres.forms.array import SimpleArrayField
  6. from django.core.exceptions import ObjectDoesNotExist
  7. from django.db.models import Q
  8. from mptt.forms import TreeNodeChoiceField
  9. from netaddr import EUI
  10. from netaddr.core import AddrFormatError
  11. from taggit.forms import TagField
  12. from timezone_field import TimeZoneFormField
  13. from circuits.models import Circuit, Provider
  14. from extras.forms import (
  15. AddRemoveTagsForm, CustomFieldForm, CustomFieldBulkEditForm, CustomFieldFilterForm, LocalConfigContextFilterForm
  16. )
  17. from ipam.models import IPAddress, VLAN, VLANGroup
  18. from tenancy.forms import TenancyFilterForm, TenancyForm
  19. from tenancy.models import Tenant, TenantGroup
  20. from utilities.forms import (
  21. APISelect, APISelectMultiple, add_blank_choice, ArrayFieldSelectMultiple, BootstrapMixin, BulkEditForm,
  22. BulkEditNullBooleanSelect, ChainedFieldsMixin, ChainedModelChoiceField, ColorSelect, CommentField, ComponentForm,
  23. ConfirmationForm, CSVChoiceField, ExpandableNameField, FilterChoiceField, FlexibleModelChoiceField, JSONField,
  24. SelectWithPK, SmallTextarea, SlugField, StaticSelect2, StaticSelect2Multiple, BOOLEAN_WITH_BLANK_CHOICES,
  25. )
  26. from virtualization.models import Cluster, ClusterGroup, VirtualMachine
  27. from .choices import *
  28. from .constants import *
  29. from .models import (
  30. Cable, DeviceBay, DeviceBayTemplate, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate,
  31. Device, DeviceRole, DeviceType, FrontPort, FrontPortTemplate, Interface, InterfaceTemplate, Manufacturer,
  32. InventoryItem, Platform, PowerFeed, PowerOutlet, PowerOutletTemplate, PowerPanel, PowerPort, PowerPortTemplate,
  33. Rack, RackGroup, RackReservation, RackRole, RearPort, RearPortTemplate, Region, Site, VirtualChassis,
  34. )
  35. DEVICE_BY_PK_RE = r'{\d+\}'
  36. INTERFACE_MODE_HELP_TEXT = """
  37. Access: One untagged VLAN<br />
  38. Tagged: One untagged VLAN and/or one or more tagged VLANs<br />
  39. Tagged All: Implies all VLANs are available (w/optional untagged VLAN)
  40. """
  41. def get_device_by_name_or_pk(name):
  42. """
  43. Attempt to retrieve a device by either its name or primary key ('{pk}').
  44. """
  45. if re.match(DEVICE_BY_PK_RE, name):
  46. pk = name.strip('{}')
  47. device = Device.objects.get(pk=pk)
  48. else:
  49. device = Device.objects.get(name=name)
  50. return device
  51. class DeviceComponentFilterForm(BootstrapMixin, forms.Form):
  52. field_order = [
  53. 'q', 'region', 'site'
  54. ]
  55. q = forms.CharField(
  56. required=False,
  57. label='Search'
  58. )
  59. region = TreeNodeChoiceField(
  60. queryset=Region.objects.all(),
  61. required=False,
  62. widget=APISelect(
  63. api_url="/api/dcim/regions/"
  64. )
  65. )
  66. site = forms.ModelChoiceField(
  67. queryset=Site.objects.all(),
  68. to_field_name='slug',
  69. required=False,
  70. help_text='Name of parent site',
  71. error_messages={
  72. 'invalid_choice': 'Site not found.',
  73. }
  74. )
  75. class InterfaceCommonForm:
  76. def clean(self):
  77. super().clean()
  78. # Validate VLAN assignments
  79. tagged_vlans = self.cleaned_data['tagged_vlans']
  80. # Untagged interfaces cannot be assigned tagged VLANs
  81. if self.cleaned_data['mode'] == InterfaceModeChoices.MODE_ACCESS and tagged_vlans:
  82. raise forms.ValidationError({
  83. 'mode': "An access interface cannot have tagged VLANs assigned."
  84. })
  85. # Remove all tagged VLAN assignments from "tagged all" interfaces
  86. elif self.cleaned_data['mode'] == InterfaceModeChoices.MODE_TAGGED_ALL:
  87. self.cleaned_data['tagged_vlans'] = []
  88. class BulkRenameForm(forms.Form):
  89. """
  90. An extendable form to be used for renaming device components in bulk.
  91. """
  92. find = forms.CharField()
  93. replace = forms.CharField()
  94. use_regex = forms.BooleanField(
  95. required=False,
  96. initial=True,
  97. label='Use regular expressions'
  98. )
  99. def clean(self):
  100. # Validate regular expression in "find" field
  101. if self.cleaned_data['use_regex']:
  102. try:
  103. re.compile(self.cleaned_data['find'])
  104. except re.error:
  105. raise forms.ValidationError({
  106. 'find': "Invalid regular expression"
  107. })
  108. #
  109. # Fields
  110. #
  111. class MACAddressField(forms.Field):
  112. widget = forms.CharField
  113. default_error_messages = {
  114. 'invalid': 'MAC address must be in EUI-48 format',
  115. }
  116. def to_python(self, value):
  117. value = super().to_python(value)
  118. # Validate MAC address format
  119. try:
  120. value = EUI(value.strip())
  121. except AddrFormatError:
  122. raise forms.ValidationError(self.error_messages['invalid'], code='invalid')
  123. return value
  124. #
  125. # Regions
  126. #
  127. class RegionForm(BootstrapMixin, forms.ModelForm):
  128. slug = SlugField()
  129. class Meta:
  130. model = Region
  131. fields = [
  132. 'parent', 'name', 'slug',
  133. ]
  134. widgets = {
  135. 'parent': APISelect(
  136. api_url="/api/dcim/regions/"
  137. )
  138. }
  139. class RegionCSVForm(forms.ModelForm):
  140. parent = forms.ModelChoiceField(
  141. queryset=Region.objects.all(),
  142. required=False,
  143. to_field_name='name',
  144. help_text='Name of parent region',
  145. error_messages={
  146. 'invalid_choice': 'Region not found.',
  147. }
  148. )
  149. class Meta:
  150. model = Region
  151. fields = Region.csv_headers
  152. help_texts = {
  153. 'name': 'Region name',
  154. 'slug': 'URL-friendly slug',
  155. }
  156. class RegionFilterForm(BootstrapMixin, forms.Form):
  157. model = Site
  158. q = forms.CharField(
  159. required=False,
  160. label='Search'
  161. )
  162. #
  163. # Sites
  164. #
  165. class SiteForm(BootstrapMixin, TenancyForm, CustomFieldForm):
  166. region = TreeNodeChoiceField(
  167. queryset=Region.objects.all(),
  168. required=False,
  169. widget=APISelect(
  170. api_url="/api/dcim/regions/"
  171. )
  172. )
  173. slug = SlugField()
  174. comments = CommentField()
  175. tags = TagField(
  176. required=False
  177. )
  178. class Meta:
  179. model = Site
  180. fields = [
  181. 'name', 'slug', 'status', 'region', 'tenant_group', 'tenant', 'facility', 'asn', 'time_zone', 'description',
  182. 'physical_address', 'shipping_address', 'latitude', 'longitude', 'contact_name', 'contact_phone',
  183. 'contact_email', 'comments', 'tags',
  184. ]
  185. widgets = {
  186. 'physical_address': SmallTextarea(
  187. attrs={
  188. 'rows': 3,
  189. }
  190. ),
  191. 'shipping_address': SmallTextarea(
  192. attrs={
  193. 'rows': 3,
  194. }
  195. ),
  196. 'status': StaticSelect2(),
  197. 'time_zone': StaticSelect2(),
  198. }
  199. help_texts = {
  200. 'name': "Full name of the site",
  201. 'facility': "Data center provider and facility (e.g. Equinix NY7)",
  202. 'asn': "BGP autonomous system number",
  203. 'time_zone': "Local time zone",
  204. 'description': "Short description (will appear in sites list)",
  205. 'physical_address': "Physical location of the building (e.g. for GPS)",
  206. 'shipping_address': "If different from the physical address",
  207. 'latitude': "Latitude in decimal format (xx.yyyyyy)",
  208. 'longitude': "Longitude in decimal format (xx.yyyyyy)"
  209. }
  210. class SiteCSVForm(forms.ModelForm):
  211. status = CSVChoiceField(
  212. choices=SiteStatusChoices,
  213. required=False,
  214. help_text='Operational status'
  215. )
  216. region = forms.ModelChoiceField(
  217. queryset=Region.objects.all(),
  218. required=False,
  219. to_field_name='name',
  220. help_text='Name of assigned region',
  221. error_messages={
  222. 'invalid_choice': 'Region not found.',
  223. }
  224. )
  225. tenant = forms.ModelChoiceField(
  226. queryset=Tenant.objects.all(),
  227. required=False,
  228. to_field_name='name',
  229. help_text='Name of assigned tenant',
  230. error_messages={
  231. 'invalid_choice': 'Tenant not found.',
  232. }
  233. )
  234. class Meta:
  235. model = Site
  236. fields = Site.csv_headers
  237. help_texts = {
  238. 'name': 'Site name',
  239. 'slug': 'URL-friendly slug',
  240. 'asn': '32-bit autonomous system number',
  241. }
  242. class SiteBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
  243. pk = forms.ModelMultipleChoiceField(
  244. queryset=Site.objects.all(),
  245. widget=forms.MultipleHiddenInput
  246. )
  247. status = forms.ChoiceField(
  248. choices=add_blank_choice(SiteStatusChoices),
  249. required=False,
  250. initial='',
  251. widget=StaticSelect2()
  252. )
  253. region = TreeNodeChoiceField(
  254. queryset=Region.objects.all(),
  255. required=False,
  256. widget=APISelect(
  257. api_url="/api/dcim/regions/"
  258. )
  259. )
  260. tenant = forms.ModelChoiceField(
  261. queryset=Tenant.objects.all(),
  262. required=False,
  263. widget=APISelect(
  264. api_url="/api/tenancy/tenants",
  265. )
  266. )
  267. asn = forms.IntegerField(
  268. min_value=1,
  269. max_value=4294967295,
  270. required=False,
  271. label='ASN'
  272. )
  273. description = forms.CharField(
  274. max_length=100,
  275. required=False
  276. )
  277. time_zone = TimeZoneFormField(
  278. choices=add_blank_choice(TimeZoneFormField().choices),
  279. required=False,
  280. widget=StaticSelect2()
  281. )
  282. class Meta:
  283. nullable_fields = [
  284. 'region', 'tenant', 'asn', 'description', 'time_zone',
  285. ]
  286. class SiteFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
  287. model = Site
  288. field_order = ['q', 'status', 'region', 'tenant_group', 'tenant']
  289. q = forms.CharField(
  290. required=False,
  291. label='Search'
  292. )
  293. status = forms.MultipleChoiceField(
  294. choices=SiteStatusChoices,
  295. required=False,
  296. widget=StaticSelect2Multiple()
  297. )
  298. region = forms.ModelMultipleChoiceField(
  299. queryset=Region.objects.all(),
  300. to_field_name='slug',
  301. required=False,
  302. widget=APISelectMultiple(
  303. api_url="/api/dcim/regions/",
  304. value_field="slug",
  305. )
  306. )
  307. #
  308. # Rack groups
  309. #
  310. class RackGroupForm(BootstrapMixin, forms.ModelForm):
  311. slug = SlugField()
  312. class Meta:
  313. model = RackGroup
  314. fields = [
  315. 'site', 'name', 'slug',
  316. ]
  317. widgets = {
  318. 'site': APISelect(
  319. api_url="/api/dcim/sites/"
  320. )
  321. }
  322. class RackGroupCSVForm(forms.ModelForm):
  323. site = forms.ModelChoiceField(
  324. queryset=Site.objects.all(),
  325. to_field_name='name',
  326. help_text='Name of parent site',
  327. error_messages={
  328. 'invalid_choice': 'Site not found.',
  329. }
  330. )
  331. class Meta:
  332. model = RackGroup
  333. fields = RackGroup.csv_headers
  334. help_texts = {
  335. 'name': 'Name of rack group',
  336. 'slug': 'URL-friendly slug',
  337. }
  338. class RackGroupFilterForm(BootstrapMixin, forms.Form):
  339. region = FilterChoiceField(
  340. queryset=Region.objects.all(),
  341. to_field_name='slug',
  342. required=False,
  343. widget=APISelectMultiple(
  344. api_url="/api/dcim/regions/",
  345. value_field="slug",
  346. filter_for={
  347. 'site': 'region'
  348. }
  349. )
  350. )
  351. site = FilterChoiceField(
  352. queryset=Site.objects.all(),
  353. to_field_name='slug',
  354. widget=APISelectMultiple(
  355. api_url="/api/dcim/sites/",
  356. value_field="slug",
  357. )
  358. )
  359. #
  360. # Rack roles
  361. #
  362. class RackRoleForm(BootstrapMixin, forms.ModelForm):
  363. slug = SlugField()
  364. class Meta:
  365. model = RackRole
  366. fields = [
  367. 'name', 'slug', 'color', 'description',
  368. ]
  369. class RackRoleCSVForm(forms.ModelForm):
  370. slug = SlugField()
  371. class Meta:
  372. model = RackRole
  373. fields = RackRole.csv_headers
  374. help_texts = {
  375. 'name': 'Name of rack role',
  376. 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
  377. }
  378. #
  379. # Racks
  380. #
  381. class RackForm(BootstrapMixin, TenancyForm, CustomFieldForm):
  382. group = ChainedModelChoiceField(
  383. queryset=RackGroup.objects.all(),
  384. chains=(
  385. ('site', 'site'),
  386. ),
  387. required=False,
  388. widget=APISelect(
  389. api_url='/api/dcim/rack-groups/',
  390. )
  391. )
  392. comments = CommentField()
  393. tags = TagField(
  394. required=False
  395. )
  396. class Meta:
  397. model = Rack
  398. fields = [
  399. 'site', 'group', 'name', 'facility_id', 'tenant_group', 'tenant', 'status', 'role', 'serial', 'asset_tag',
  400. 'type', 'width', 'u_height', 'desc_units', 'outer_width', 'outer_depth', 'outer_unit', 'comments', 'tags',
  401. ]
  402. help_texts = {
  403. 'site': "The site at which the rack exists",
  404. 'name': "Organizational rack name",
  405. 'facility_id': "The unique rack ID assigned by the facility",
  406. 'u_height': "Height in rack units",
  407. }
  408. widgets = {
  409. 'site': APISelect(
  410. api_url="/api/dcim/sites/",
  411. filter_for={
  412. 'group': 'site_id',
  413. }
  414. ),
  415. 'status': StaticSelect2(),
  416. 'role': APISelect(
  417. api_url="/api/dcim/rack-roles/"
  418. ),
  419. 'type': StaticSelect2(),
  420. 'width': StaticSelect2(),
  421. 'outer_unit': StaticSelect2(),
  422. }
  423. class RackCSVForm(forms.ModelForm):
  424. site = forms.ModelChoiceField(
  425. queryset=Site.objects.all(),
  426. to_field_name='name',
  427. help_text='Name of parent site',
  428. error_messages={
  429. 'invalid_choice': 'Site not found.',
  430. }
  431. )
  432. group_name = forms.CharField(
  433. help_text='Name of rack group',
  434. required=False
  435. )
  436. tenant = forms.ModelChoiceField(
  437. queryset=Tenant.objects.all(),
  438. required=False,
  439. to_field_name='name',
  440. help_text='Name of assigned tenant',
  441. error_messages={
  442. 'invalid_choice': 'Tenant not found.',
  443. }
  444. )
  445. status = CSVChoiceField(
  446. choices=RackStatusChoices,
  447. required=False,
  448. help_text='Operational status'
  449. )
  450. role = forms.ModelChoiceField(
  451. queryset=RackRole.objects.all(),
  452. required=False,
  453. to_field_name='name',
  454. help_text='Name of assigned role',
  455. error_messages={
  456. 'invalid_choice': 'Role not found.',
  457. }
  458. )
  459. type = CSVChoiceField(
  460. choices=RackTypeChoices,
  461. required=False,
  462. help_text='Rack type'
  463. )
  464. width = forms.ChoiceField(
  465. choices=RackWidthChoices,
  466. help_text='Rail-to-rail width (in inches)'
  467. )
  468. outer_unit = CSVChoiceField(
  469. choices=RackDimensionUnitChoices,
  470. required=False,
  471. help_text='Unit for outer dimensions'
  472. )
  473. class Meta:
  474. model = Rack
  475. fields = Rack.csv_headers
  476. help_texts = {
  477. 'name': 'Rack name',
  478. 'u_height': 'Height in rack units',
  479. }
  480. def clean(self):
  481. super().clean()
  482. site = self.cleaned_data.get('site')
  483. group_name = self.cleaned_data.get('group_name')
  484. name = self.cleaned_data.get('name')
  485. facility_id = self.cleaned_data.get('facility_id')
  486. # Validate rack group
  487. if group_name:
  488. try:
  489. self.instance.group = RackGroup.objects.get(site=site, name=group_name)
  490. except RackGroup.DoesNotExist:
  491. raise forms.ValidationError("Rack group {} not found for site {}".format(group_name, site))
  492. # Validate uniqueness of rack name within group
  493. if Rack.objects.filter(group=self.instance.group, name=name).exists():
  494. raise forms.ValidationError(
  495. "A rack named {} already exists within group {}".format(name, group_name)
  496. )
  497. # Validate uniqueness of facility ID within group
  498. if facility_id and Rack.objects.filter(group=self.instance.group, facility_id=facility_id).exists():
  499. raise forms.ValidationError(
  500. "A rack with the facility ID {} already exists within group {}".format(facility_id, group_name)
  501. )
  502. class RackBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
  503. pk = forms.ModelMultipleChoiceField(
  504. queryset=Rack.objects.all(),
  505. widget=forms.MultipleHiddenInput
  506. )
  507. site = forms.ModelChoiceField(
  508. queryset=Site.objects.all(),
  509. required=False,
  510. widget=APISelect(
  511. api_url="/api/dcim/sites",
  512. filter_for={
  513. 'group': 'site_id',
  514. }
  515. )
  516. )
  517. group = forms.ModelChoiceField(
  518. queryset=RackGroup.objects.all(),
  519. required=False,
  520. widget=APISelect(
  521. api_url="/api/dcim/rack-groups",
  522. )
  523. )
  524. tenant = forms.ModelChoiceField(
  525. queryset=Tenant.objects.all(),
  526. required=False,
  527. widget=APISelect(
  528. api_url="/api/tenancy/tenants",
  529. )
  530. )
  531. status = forms.ChoiceField(
  532. choices=add_blank_choice(RackStatusChoices),
  533. required=False,
  534. initial='',
  535. widget=StaticSelect2()
  536. )
  537. role = forms.ModelChoiceField(
  538. queryset=RackRole.objects.all(),
  539. required=False,
  540. widget=APISelect(
  541. api_url="/api/dcim/rack-roles",
  542. )
  543. )
  544. serial = forms.CharField(
  545. max_length=50,
  546. required=False,
  547. label='Serial Number'
  548. )
  549. asset_tag = forms.CharField(
  550. max_length=50,
  551. required=False
  552. )
  553. type = forms.ChoiceField(
  554. choices=add_blank_choice(RackTypeChoices),
  555. required=False,
  556. widget=StaticSelect2()
  557. )
  558. width = forms.ChoiceField(
  559. choices=add_blank_choice(RackWidthChoices),
  560. required=False,
  561. widget=StaticSelect2()
  562. )
  563. u_height = forms.IntegerField(
  564. required=False,
  565. label='Height (U)'
  566. )
  567. desc_units = forms.NullBooleanField(
  568. required=False,
  569. widget=BulkEditNullBooleanSelect,
  570. label='Descending units'
  571. )
  572. outer_width = forms.IntegerField(
  573. required=False,
  574. min_value=1
  575. )
  576. outer_depth = forms.IntegerField(
  577. required=False,
  578. min_value=1
  579. )
  580. outer_unit = forms.ChoiceField(
  581. choices=add_blank_choice(RackDimensionUnitChoices),
  582. required=False,
  583. widget=StaticSelect2()
  584. )
  585. comments = CommentField(
  586. widget=SmallTextarea
  587. )
  588. class Meta:
  589. nullable_fields = [
  590. 'group', 'tenant', 'role', 'serial', 'asset_tag', 'outer_width', 'outer_depth', 'outer_unit', 'comments',
  591. ]
  592. class RackFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldFilterForm):
  593. model = Rack
  594. field_order = ['q', 'region', 'site', 'group_id', 'status', 'role', 'tenant_group', 'tenant']
  595. q = forms.CharField(
  596. required=False,
  597. label='Search'
  598. )
  599. region = FilterChoiceField(
  600. queryset=Region.objects.all(),
  601. to_field_name='slug',
  602. required=False,
  603. widget=APISelectMultiple(
  604. api_url="/api/dcim/regions/",
  605. value_field="slug",
  606. filter_for={
  607. 'site': 'region'
  608. }
  609. )
  610. )
  611. site = FilterChoiceField(
  612. queryset=Site.objects.all(),
  613. to_field_name='slug',
  614. widget=APISelectMultiple(
  615. api_url="/api/dcim/sites/",
  616. value_field="slug",
  617. filter_for={
  618. 'group_id': 'site'
  619. }
  620. )
  621. )
  622. group_id = FilterChoiceField(
  623. queryset=RackGroup.objects.prefetch_related(
  624. 'site'
  625. ),
  626. label='Rack group',
  627. null_label='-- None --',
  628. widget=APISelectMultiple(
  629. api_url="/api/dcim/rack-groups/",
  630. null_option=True
  631. )
  632. )
  633. status = forms.MultipleChoiceField(
  634. choices=RackStatusChoices,
  635. required=False,
  636. widget=StaticSelect2Multiple()
  637. )
  638. role = FilterChoiceField(
  639. queryset=RackRole.objects.all(),
  640. to_field_name='slug',
  641. null_label='-- None --',
  642. widget=APISelectMultiple(
  643. api_url="/api/dcim/rack-roles/",
  644. value_field="slug",
  645. null_option=True,
  646. )
  647. )
  648. #
  649. # Rack reservations
  650. #
  651. class RackReservationForm(BootstrapMixin, TenancyForm, forms.ModelForm):
  652. units = SimpleArrayField(
  653. base_field=forms.IntegerField(),
  654. widget=ArrayFieldSelectMultiple(
  655. attrs={
  656. 'size': 10,
  657. }
  658. )
  659. )
  660. user = forms.ModelChoiceField(
  661. queryset=User.objects.order_by(
  662. 'username'
  663. ),
  664. widget=StaticSelect2()
  665. )
  666. class Meta:
  667. model = RackReservation
  668. fields = [
  669. 'units', 'user', 'tenant_group', 'tenant', 'description',
  670. ]
  671. def __init__(self, *args, **kwargs):
  672. super().__init__(*args, **kwargs)
  673. # Populate rack unit choices
  674. self.fields['units'].widget.choices = self._get_unit_choices()
  675. def _get_unit_choices(self):
  676. rack = self.instance.rack
  677. reserved_units = []
  678. for resv in rack.reservations.exclude(pk=self.instance.pk):
  679. for u in resv.units:
  680. reserved_units.append(u)
  681. unit_choices = [(u, {'label': str(u), 'disabled': u in reserved_units}) for u in rack.units]
  682. return unit_choices
  683. class RackReservationBulkEditForm(BootstrapMixin, BulkEditForm):
  684. pk = forms.ModelMultipleChoiceField(
  685. queryset=RackReservation.objects.all(),
  686. widget=forms.MultipleHiddenInput()
  687. )
  688. user = forms.ModelChoiceField(
  689. queryset=User.objects.order_by(
  690. 'username'
  691. ),
  692. required=False,
  693. widget=StaticSelect2()
  694. )
  695. tenant = forms.ModelChoiceField(
  696. queryset=Tenant.objects.all(),
  697. required=False,
  698. widget=APISelect(
  699. api_url="/api/tenancy/tenant",
  700. )
  701. )
  702. description = forms.CharField(
  703. max_length=100,
  704. required=False
  705. )
  706. class Meta:
  707. nullable_fields = []
  708. class RackReservationFilterForm(BootstrapMixin, TenancyFilterForm):
  709. field_order = ['q', 'site', 'group_id', 'tenant_group', 'tenant']
  710. q = forms.CharField(
  711. required=False,
  712. label='Search'
  713. )
  714. site = FilterChoiceField(
  715. queryset=Site.objects.all(),
  716. to_field_name='slug',
  717. widget=APISelectMultiple(
  718. api_url="/api/dcim/sites/",
  719. value_field="slug",
  720. )
  721. )
  722. group_id = FilterChoiceField(
  723. queryset=RackGroup.objects.prefetch_related('site'),
  724. label='Rack group',
  725. null_label='-- None --',
  726. widget=APISelectMultiple(
  727. api_url="/api/dcim/rack-groups/",
  728. null_option=True,
  729. )
  730. )
  731. #
  732. # Manufacturers
  733. #
  734. class ManufacturerForm(BootstrapMixin, forms.ModelForm):
  735. slug = SlugField()
  736. class Meta:
  737. model = Manufacturer
  738. fields = [
  739. 'name', 'slug',
  740. ]
  741. class ManufacturerCSVForm(forms.ModelForm):
  742. class Meta:
  743. model = Manufacturer
  744. fields = Manufacturer.csv_headers
  745. help_texts = {
  746. 'name': 'Manufacturer name',
  747. 'slug': 'URL-friendly slug',
  748. }
  749. #
  750. # Device types
  751. #
  752. class DeviceTypeForm(BootstrapMixin, CustomFieldForm):
  753. slug = SlugField(
  754. slug_source='model'
  755. )
  756. comments = CommentField()
  757. tags = TagField(
  758. required=False
  759. )
  760. class Meta:
  761. model = DeviceType
  762. fields = [
  763. 'manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'subdevice_role', 'comments',
  764. 'tags',
  765. ]
  766. widgets = {
  767. 'manufacturer': APISelect(
  768. api_url="/api/dcim/manufacturers/"
  769. ),
  770. 'subdevice_role': StaticSelect2()
  771. }
  772. class DeviceTypeImportForm(BootstrapMixin, forms.ModelForm):
  773. manufacturer = forms.ModelChoiceField(
  774. queryset=Manufacturer.objects.all(),
  775. to_field_name='name'
  776. )
  777. class Meta:
  778. model = DeviceType
  779. fields = [
  780. 'manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'subdevice_role',
  781. ]
  782. class DeviceTypeBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
  783. pk = forms.ModelMultipleChoiceField(
  784. queryset=DeviceType.objects.all(),
  785. widget=forms.MultipleHiddenInput()
  786. )
  787. manufacturer = forms.ModelChoiceField(
  788. queryset=Manufacturer.objects.all(),
  789. required=False,
  790. widget=APISelect(
  791. api_url="/api/dcim/manufactureres"
  792. )
  793. )
  794. u_height = forms.IntegerField(
  795. min_value=1,
  796. required=False
  797. )
  798. is_full_depth = forms.NullBooleanField(
  799. required=False,
  800. widget=BulkEditNullBooleanSelect(),
  801. label='Is full depth'
  802. )
  803. class Meta:
  804. nullable_fields = []
  805. class DeviceTypeFilterForm(BootstrapMixin, CustomFieldFilterForm):
  806. model = DeviceType
  807. q = forms.CharField(
  808. required=False,
  809. label='Search'
  810. )
  811. manufacturer = FilterChoiceField(
  812. queryset=Manufacturer.objects.all(),
  813. to_field_name='slug',
  814. widget=APISelectMultiple(
  815. api_url="/api/dcim/manufacturers/",
  816. value_field="slug",
  817. )
  818. )
  819. subdevice_role = forms.MultipleChoiceField(
  820. choices=add_blank_choice(SubdeviceRoleChoices),
  821. required=False,
  822. widget=StaticSelect2Multiple()
  823. )
  824. console_ports = forms.NullBooleanField(
  825. required=False,
  826. label='Has console ports',
  827. widget=StaticSelect2(
  828. choices=BOOLEAN_WITH_BLANK_CHOICES
  829. )
  830. )
  831. console_server_ports = forms.NullBooleanField(
  832. required=False,
  833. label='Has console server ports',
  834. widget=StaticSelect2(
  835. choices=BOOLEAN_WITH_BLANK_CHOICES
  836. )
  837. )
  838. power_ports = forms.NullBooleanField(
  839. required=False,
  840. label='Has power ports',
  841. widget=StaticSelect2(
  842. choices=BOOLEAN_WITH_BLANK_CHOICES
  843. )
  844. )
  845. power_outlets = forms.NullBooleanField(
  846. required=False,
  847. label='Has power outlets',
  848. widget=StaticSelect2(
  849. choices=BOOLEAN_WITH_BLANK_CHOICES
  850. )
  851. )
  852. interfaces = forms.NullBooleanField(
  853. required=False,
  854. label='Has interfaces',
  855. widget=StaticSelect2(
  856. choices=BOOLEAN_WITH_BLANK_CHOICES
  857. )
  858. )
  859. pass_through_ports = forms.NullBooleanField(
  860. required=False,
  861. label='Has pass-through ports',
  862. widget=StaticSelect2(
  863. choices=BOOLEAN_WITH_BLANK_CHOICES
  864. )
  865. )
  866. #
  867. # Device component templates
  868. #
  869. class ConsolePortTemplateForm(BootstrapMixin, forms.ModelForm):
  870. class Meta:
  871. model = ConsolePortTemplate
  872. fields = [
  873. 'device_type', 'name', 'type',
  874. ]
  875. widgets = {
  876. 'device_type': forms.HiddenInput(),
  877. }
  878. class ConsolePortTemplateCreateForm(ComponentForm):
  879. name_pattern = ExpandableNameField(
  880. label='Name'
  881. )
  882. type = forms.ChoiceField(
  883. choices=ConsolePortTypeChoices,
  884. widget=StaticSelect2()
  885. )
  886. class ConsoleServerPortTemplateForm(BootstrapMixin, forms.ModelForm):
  887. class Meta:
  888. model = ConsoleServerPortTemplate
  889. fields = [
  890. 'device_type', 'name', 'type',
  891. ]
  892. widgets = {
  893. 'device_type': forms.HiddenInput(),
  894. }
  895. class ConsoleServerPortTemplateCreateForm(ComponentForm):
  896. name_pattern = ExpandableNameField(
  897. label='Name'
  898. )
  899. type = forms.ChoiceField(
  900. choices=add_blank_choice(ConsolePortTypeChoices),
  901. widget=StaticSelect2()
  902. )
  903. class PowerPortTemplateForm(BootstrapMixin, forms.ModelForm):
  904. class Meta:
  905. model = PowerPortTemplate
  906. fields = [
  907. 'device_type', 'name', 'type', 'maximum_draw', 'allocated_draw',
  908. ]
  909. widgets = {
  910. 'device_type': forms.HiddenInput(),
  911. }
  912. class PowerPortTemplateCreateForm(ComponentForm):
  913. name_pattern = ExpandableNameField(
  914. label='Name'
  915. )
  916. type = forms.ChoiceField(
  917. choices=add_blank_choice(PowerPortTypeChoices),
  918. required=False
  919. )
  920. maximum_draw = forms.IntegerField(
  921. min_value=1,
  922. required=False,
  923. help_text="Maximum power draw (watts)"
  924. )
  925. allocated_draw = forms.IntegerField(
  926. min_value=1,
  927. required=False,
  928. help_text="Allocated power draw (watts)"
  929. )
  930. class PowerOutletTemplateForm(BootstrapMixin, forms.ModelForm):
  931. class Meta:
  932. model = PowerOutletTemplate
  933. fields = [
  934. 'device_type', 'name', 'type', 'power_port', 'feed_leg',
  935. ]
  936. widgets = {
  937. 'device_type': forms.HiddenInput(),
  938. }
  939. def __init__(self, *args, **kwargs):
  940. super().__init__(*args, **kwargs)
  941. # Limit power_port choices to current DeviceType
  942. if hasattr(self.instance, 'device_type'):
  943. self.fields['power_port'].queryset = PowerPortTemplate.objects.filter(
  944. device_type=self.instance.device_type
  945. )
  946. class PowerOutletTemplateCreateForm(ComponentForm):
  947. name_pattern = ExpandableNameField(
  948. label='Name'
  949. )
  950. type = forms.ChoiceField(
  951. choices=add_blank_choice(PowerOutletTypeChoices),
  952. required=False
  953. )
  954. power_port = forms.ModelChoiceField(
  955. queryset=PowerPortTemplate.objects.all(),
  956. required=False
  957. )
  958. feed_leg = forms.ChoiceField(
  959. choices=add_blank_choice(PowerOutletFeedLegChoices),
  960. required=False,
  961. widget=StaticSelect2()
  962. )
  963. def __init__(self, *args, **kwargs):
  964. super().__init__(*args, **kwargs)
  965. # Limit power_port choices to current DeviceType
  966. self.fields['power_port'].queryset = PowerPortTemplate.objects.filter(
  967. device_type=self.parent
  968. )
  969. class InterfaceTemplateForm(BootstrapMixin, forms.ModelForm):
  970. class Meta:
  971. model = InterfaceTemplate
  972. fields = [
  973. 'device_type', 'name', 'type', 'mgmt_only',
  974. ]
  975. widgets = {
  976. 'device_type': forms.HiddenInput(),
  977. 'type': StaticSelect2(),
  978. }
  979. class InterfaceTemplateCreateForm(ComponentForm):
  980. name_pattern = ExpandableNameField(
  981. label='Name'
  982. )
  983. type = forms.ChoiceField(
  984. choices=InterfaceTypeChoices,
  985. widget=StaticSelect2()
  986. )
  987. mgmt_only = forms.BooleanField(
  988. required=False,
  989. label='Management only'
  990. )
  991. class InterfaceTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  992. pk = forms.ModelMultipleChoiceField(
  993. queryset=InterfaceTemplate.objects.all(),
  994. widget=forms.MultipleHiddenInput()
  995. )
  996. type = forms.ChoiceField(
  997. choices=add_blank_choice(InterfaceTypeChoices),
  998. required=False,
  999. widget=StaticSelect2()
  1000. )
  1001. mgmt_only = forms.NullBooleanField(
  1002. required=False,
  1003. widget=BulkEditNullBooleanSelect,
  1004. label='Management only'
  1005. )
  1006. class Meta:
  1007. nullable_fields = []
  1008. class FrontPortTemplateForm(BootstrapMixin, forms.ModelForm):
  1009. class Meta:
  1010. model = FrontPortTemplate
  1011. fields = [
  1012. 'device_type', 'name', 'type', 'rear_port', 'rear_port_position',
  1013. ]
  1014. widgets = {
  1015. 'device_type': forms.HiddenInput(),
  1016. 'rear_port': StaticSelect2(),
  1017. }
  1018. def __init__(self, *args, **kwargs):
  1019. super().__init__(*args, **kwargs)
  1020. # Limit rear_port choices to current DeviceType
  1021. if hasattr(self.instance, 'device_type'):
  1022. self.fields['rear_port'].queryset = RearPortTemplate.objects.filter(
  1023. device_type=self.instance.device_type
  1024. )
  1025. class FrontPortTemplateCreateForm(ComponentForm):
  1026. name_pattern = ExpandableNameField(
  1027. label='Name'
  1028. )
  1029. type = forms.ChoiceField(
  1030. choices=PortTypeChoices,
  1031. widget=StaticSelect2()
  1032. )
  1033. rear_port_set = forms.MultipleChoiceField(
  1034. choices=[],
  1035. label='Rear ports',
  1036. help_text='Select one rear port assignment for each front port being created.',
  1037. )
  1038. def __init__(self, *args, **kwargs):
  1039. super().__init__(*args, **kwargs)
  1040. # Determine which rear port positions are occupied. These will be excluded from the list of available mappings.
  1041. occupied_port_positions = [
  1042. (front_port.rear_port_id, front_port.rear_port_position)
  1043. for front_port in self.parent.frontport_templates.all()
  1044. ]
  1045. # Populate rear port choices
  1046. choices = []
  1047. rear_ports = RearPortTemplate.objects.filter(device_type=self.parent)
  1048. for rear_port in rear_ports:
  1049. for i in range(1, rear_port.positions + 1):
  1050. if (rear_port.pk, i) not in occupied_port_positions:
  1051. choices.append(
  1052. ('{}:{}'.format(rear_port.pk, i), '{}:{}'.format(rear_port.name, i))
  1053. )
  1054. self.fields['rear_port_set'].choices = choices
  1055. def clean(self):
  1056. # Validate that the number of ports being created equals the number of selected (rear port, position) tuples
  1057. front_port_count = len(self.cleaned_data['name_pattern'])
  1058. rear_port_count = len(self.cleaned_data['rear_port_set'])
  1059. if front_port_count != rear_port_count:
  1060. raise forms.ValidationError({
  1061. 'rear_port_set': 'The provided name pattern will create {} ports, however {} rear port assignments '
  1062. 'were selected. These counts must match.'.format(front_port_count, rear_port_count)
  1063. })
  1064. def get_iterative_data(self, iteration):
  1065. # Assign rear port and position from selected set
  1066. rear_port, position = self.cleaned_data['rear_port_set'][iteration].split(':')
  1067. return {
  1068. 'rear_port': int(rear_port),
  1069. 'rear_port_position': int(position),
  1070. }
  1071. class RearPortTemplateForm(BootstrapMixin, forms.ModelForm):
  1072. class Meta:
  1073. model = RearPortTemplate
  1074. fields = [
  1075. 'device_type', 'name', 'type', 'positions',
  1076. ]
  1077. widgets = {
  1078. 'device_type': forms.HiddenInput(),
  1079. 'type': StaticSelect2(),
  1080. }
  1081. class RearPortTemplateCreateForm(ComponentForm):
  1082. name_pattern = ExpandableNameField(
  1083. label='Name'
  1084. )
  1085. type = forms.ChoiceField(
  1086. choices=PortTypeChoices,
  1087. widget=StaticSelect2(),
  1088. )
  1089. positions = forms.IntegerField(
  1090. min_value=1,
  1091. max_value=64,
  1092. initial=1,
  1093. help_text='The number of front ports which may be mapped to each rear port'
  1094. )
  1095. class DeviceBayTemplateForm(BootstrapMixin, forms.ModelForm):
  1096. class Meta:
  1097. model = DeviceBayTemplate
  1098. fields = [
  1099. 'device_type', 'name',
  1100. ]
  1101. widgets = {
  1102. 'device_type': forms.HiddenInput(),
  1103. }
  1104. class DeviceBayTemplateCreateForm(ComponentForm):
  1105. name_pattern = ExpandableNameField(
  1106. label='Name'
  1107. )
  1108. #
  1109. # Component template import forms
  1110. #
  1111. class ComponentTemplateImportForm(BootstrapMixin, forms.ModelForm):
  1112. def __init__(self, device_type, data=None, *args, **kwargs):
  1113. # Must pass the parent DeviceType on form initialization
  1114. data.update({
  1115. 'device_type': device_type.pk,
  1116. })
  1117. super().__init__(data, *args, **kwargs)
  1118. def clean_device_type(self):
  1119. data = self.cleaned_data['device_type']
  1120. # Limit fields referencing other components to the parent DeviceType
  1121. for field_name, field in self.fields.items():
  1122. if isinstance(field, forms.ModelChoiceField) and field_name != 'device_type':
  1123. field.queryset = field.queryset.filter(device_type=data)
  1124. return data
  1125. class ConsolePortTemplateImportForm(ComponentTemplateImportForm):
  1126. class Meta:
  1127. model = ConsolePortTemplate
  1128. fields = [
  1129. 'device_type', 'name', 'type',
  1130. ]
  1131. class ConsoleServerPortTemplateImportForm(ComponentTemplateImportForm):
  1132. class Meta:
  1133. model = ConsoleServerPortTemplate
  1134. fields = [
  1135. 'device_type', 'name', 'type',
  1136. ]
  1137. class PowerPortTemplateImportForm(ComponentTemplateImportForm):
  1138. class Meta:
  1139. model = PowerPortTemplate
  1140. fields = [
  1141. 'device_type', 'name', 'type', 'maximum_draw', 'allocated_draw',
  1142. ]
  1143. class PowerOutletTemplateImportForm(ComponentTemplateImportForm):
  1144. power_port = forms.ModelChoiceField(
  1145. queryset=PowerPortTemplate.objects.all(),
  1146. to_field_name='name',
  1147. required=False
  1148. )
  1149. class Meta:
  1150. model = PowerOutletTemplate
  1151. fields = [
  1152. 'device_type', 'name', 'type', 'power_port', 'feed_leg',
  1153. ]
  1154. class InterfaceTemplateImportForm(ComponentTemplateImportForm):
  1155. type = forms.ChoiceField(
  1156. choices=InterfaceTypeChoices.CHOICES
  1157. )
  1158. class Meta:
  1159. model = InterfaceTemplate
  1160. fields = [
  1161. 'device_type', 'name', 'type', 'mgmt_only',
  1162. ]
  1163. class FrontPortTemplateImportForm(ComponentTemplateImportForm):
  1164. type = forms.ChoiceField(
  1165. choices=PortTypeChoices.CHOICES
  1166. )
  1167. rear_port = forms.ModelChoiceField(
  1168. queryset=RearPortTemplate.objects.all(),
  1169. to_field_name='name',
  1170. required=False
  1171. )
  1172. class Meta:
  1173. model = FrontPortTemplate
  1174. fields = [
  1175. 'device_type', 'name', 'type', 'rear_port', 'rear_port_position',
  1176. ]
  1177. class RearPortTemplateImportForm(ComponentTemplateImportForm):
  1178. type = forms.ChoiceField(
  1179. choices=PortTypeChoices.CHOICES
  1180. )
  1181. class Meta:
  1182. model = RearPortTemplate
  1183. fields = [
  1184. 'device_type', 'name', 'type', 'positions',
  1185. ]
  1186. class DeviceBayTemplateImportForm(ComponentTemplateImportForm):
  1187. class Meta:
  1188. model = DeviceBayTemplate
  1189. fields = [
  1190. 'device_type', 'name',
  1191. ]
  1192. #
  1193. # Device roles
  1194. #
  1195. class DeviceRoleForm(BootstrapMixin, forms.ModelForm):
  1196. slug = SlugField()
  1197. class Meta:
  1198. model = DeviceRole
  1199. fields = [
  1200. 'name', 'slug', 'color', 'vm_role', 'description',
  1201. ]
  1202. class DeviceRoleCSVForm(forms.ModelForm):
  1203. slug = SlugField()
  1204. class Meta:
  1205. model = DeviceRole
  1206. fields = DeviceRole.csv_headers
  1207. help_texts = {
  1208. 'name': 'Name of device role',
  1209. 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
  1210. }
  1211. #
  1212. # Platforms
  1213. #
  1214. class PlatformForm(BootstrapMixin, forms.ModelForm):
  1215. slug = SlugField(
  1216. max_length=64
  1217. )
  1218. class Meta:
  1219. model = Platform
  1220. fields = [
  1221. 'name', 'slug', 'manufacturer', 'napalm_driver', 'napalm_args',
  1222. ]
  1223. widgets = {
  1224. 'manufacturer': APISelect(
  1225. api_url="/api/dcim/manufacturers/"
  1226. ),
  1227. 'napalm_args': SmallTextarea(),
  1228. }
  1229. class PlatformCSVForm(forms.ModelForm):
  1230. slug = SlugField()
  1231. manufacturer = forms.ModelChoiceField(
  1232. queryset=Manufacturer.objects.all(),
  1233. required=False,
  1234. to_field_name='name',
  1235. help_text='Manufacturer name',
  1236. error_messages={
  1237. 'invalid_choice': 'Manufacturer not found.',
  1238. }
  1239. )
  1240. class Meta:
  1241. model = Platform
  1242. fields = Platform.csv_headers
  1243. help_texts = {
  1244. 'name': 'Platform name',
  1245. }
  1246. #
  1247. # Devices
  1248. #
  1249. class DeviceForm(BootstrapMixin, TenancyForm, CustomFieldForm):
  1250. site = forms.ModelChoiceField(
  1251. queryset=Site.objects.all(),
  1252. widget=APISelect(
  1253. api_url="/api/dcim/sites/",
  1254. filter_for={
  1255. 'rack': 'site_id'
  1256. }
  1257. )
  1258. )
  1259. rack = ChainedModelChoiceField(
  1260. queryset=Rack.objects.all(),
  1261. chains=(
  1262. ('site', 'site'),
  1263. ),
  1264. required=False,
  1265. widget=APISelect(
  1266. api_url='/api/dcim/racks/',
  1267. display_field='display_name'
  1268. )
  1269. )
  1270. position = forms.TypedChoiceField(
  1271. required=False,
  1272. empty_value=None,
  1273. help_text="The lowest-numbered unit occupied by the device",
  1274. widget=APISelect(
  1275. api_url='/api/dcim/racks/{{rack}}/elevation/',
  1276. disabled_indicator='device'
  1277. )
  1278. )
  1279. manufacturer = forms.ModelChoiceField(
  1280. queryset=Manufacturer.objects.all(),
  1281. widget=APISelect(
  1282. api_url="/api/dcim/manufacturers/",
  1283. filter_for={
  1284. 'device_type': 'manufacturer_id',
  1285. 'platform': 'manufacturer_id'
  1286. }
  1287. )
  1288. )
  1289. device_type = ChainedModelChoiceField(
  1290. queryset=DeviceType.objects.all(),
  1291. chains=(
  1292. ('manufacturer', 'manufacturer'),
  1293. ),
  1294. label='Device type',
  1295. widget=APISelect(
  1296. api_url='/api/dcim/device-types/',
  1297. display_field='model'
  1298. )
  1299. )
  1300. cluster_group = forms.ModelChoiceField(
  1301. queryset=ClusterGroup.objects.all(),
  1302. required=False,
  1303. widget=APISelect(
  1304. api_url="/api/virtualization/cluster-groups/",
  1305. filter_for={
  1306. 'cluster': 'group_id'
  1307. },
  1308. attrs={
  1309. 'nullable': 'true'
  1310. }
  1311. )
  1312. )
  1313. cluster = ChainedModelChoiceField(
  1314. queryset=Cluster.objects.all(),
  1315. chains=(
  1316. ('group', 'cluster_group'),
  1317. ),
  1318. required=False,
  1319. widget=APISelect(
  1320. api_url='/api/virtualization/clusters/',
  1321. )
  1322. )
  1323. comments = CommentField()
  1324. tags = TagField(required=False)
  1325. local_context_data = JSONField(
  1326. required=False,
  1327. label=''
  1328. )
  1329. class Meta:
  1330. model = Device
  1331. fields = [
  1332. 'name', 'device_role', 'device_type', 'serial', 'asset_tag', 'site', 'rack', 'position', 'face',
  1333. 'status', 'platform', 'primary_ip4', 'primary_ip6', 'cluster_group', 'cluster', 'tenant_group', 'tenant',
  1334. 'comments', 'tags', 'local_context_data'
  1335. ]
  1336. help_texts = {
  1337. 'device_role': "The function this device serves",
  1338. 'serial': "Chassis serial number",
  1339. 'local_context_data': "Local config context data overwrites all source contexts in the final rendered "
  1340. "config context",
  1341. }
  1342. widgets = {
  1343. 'face': StaticSelect2(
  1344. filter_for={
  1345. 'position': 'face'
  1346. }
  1347. ),
  1348. 'device_role': APISelect(
  1349. api_url='/api/dcim/device-roles/'
  1350. ),
  1351. 'status': StaticSelect2(),
  1352. 'platform': APISelect(
  1353. api_url="/api/dcim/platforms/",
  1354. additional_query_params={
  1355. "manufacturer_id": "null"
  1356. }
  1357. ),
  1358. 'primary_ip4': StaticSelect2(),
  1359. 'primary_ip6': StaticSelect2(),
  1360. }
  1361. def __init__(self, *args, **kwargs):
  1362. # Initialize helper selectors
  1363. instance = kwargs.get('instance')
  1364. if 'initial' not in kwargs:
  1365. kwargs['initial'] = {}
  1366. # Using hasattr() instead of "is not None" to avoid RelatedObjectDoesNotExist on required field
  1367. if instance and hasattr(instance, 'device_type'):
  1368. kwargs['initial']['manufacturer'] = instance.device_type.manufacturer
  1369. if instance and instance.cluster is not None:
  1370. kwargs['initial']['cluster_group'] = instance.cluster.group
  1371. super().__init__(*args, **kwargs)
  1372. if self.instance.pk:
  1373. # Compile list of choices for primary IPv4 and IPv6 addresses
  1374. for family in [4, 6]:
  1375. ip_choices = [(None, '---------')]
  1376. # Gather PKs of all interfaces belonging to this Device or a peer VirtualChassis member
  1377. interface_ids = self.instance.vc_interfaces.values('pk')
  1378. # Collect interface IPs
  1379. interface_ips = IPAddress.objects.prefetch_related('interface').filter(
  1380. family=family, interface_id__in=interface_ids
  1381. )
  1382. if interface_ips:
  1383. ip_list = [(ip.id, '{} ({})'.format(ip.address, ip.interface)) for ip in interface_ips]
  1384. ip_choices.append(('Interface IPs', ip_list))
  1385. # Collect NAT IPs
  1386. nat_ips = IPAddress.objects.prefetch_related('nat_inside').filter(
  1387. family=family, nat_inside__interface__in=interface_ids
  1388. )
  1389. if nat_ips:
  1390. ip_list = [(ip.id, '{} ({})'.format(ip.address, ip.nat_inside.address)) for ip in nat_ips]
  1391. ip_choices.append(('NAT IPs', ip_list))
  1392. self.fields['primary_ip{}'.format(family)].choices = ip_choices
  1393. # If editing an existing device, exclude it from the list of occupied rack units. This ensures that a device
  1394. # can be flipped from one face to another.
  1395. self.fields['position'].widget.add_additional_query_param('exclude', self.instance.pk)
  1396. # Limit platform by manufacturer
  1397. self.fields['platform'].queryset = Platform.objects.filter(
  1398. Q(manufacturer__isnull=True) | Q(manufacturer=self.instance.device_type.manufacturer)
  1399. )
  1400. else:
  1401. # An object that doesn't exist yet can't have any IPs assigned to it
  1402. self.fields['primary_ip4'].choices = []
  1403. self.fields['primary_ip4'].widget.attrs['readonly'] = True
  1404. self.fields['primary_ip6'].choices = []
  1405. self.fields['primary_ip6'].widget.attrs['readonly'] = True
  1406. # Rack position
  1407. pk = self.instance.pk if self.instance.pk else None
  1408. try:
  1409. if self.is_bound and self.data.get('rack') and str(self.data.get('face')):
  1410. position_choices = Rack.objects.get(pk=self.data['rack']) \
  1411. .get_rack_units(face=self.data.get('face'), exclude=pk)
  1412. elif self.initial.get('rack') and str(self.initial.get('face')):
  1413. position_choices = Rack.objects.get(pk=self.initial['rack']) \
  1414. .get_rack_units(face=self.initial.get('face'), exclude=pk)
  1415. else:
  1416. position_choices = []
  1417. except Rack.DoesNotExist:
  1418. position_choices = []
  1419. self.fields['position'].choices = [('', '---------')] + [
  1420. (p['id'], {
  1421. 'label': p['name'],
  1422. 'disabled': bool(p['device'] and p['id'] != self.initial.get('position')),
  1423. }) for p in position_choices
  1424. ]
  1425. # Disable rack assignment if this is a child device installed in a parent device
  1426. if pk and self.instance.device_type.is_child_device and hasattr(self.instance, 'parent_bay'):
  1427. self.fields['site'].disabled = True
  1428. self.fields['rack'].disabled = True
  1429. self.initial['site'] = self.instance.parent_bay.device.site_id
  1430. self.initial['rack'] = self.instance.parent_bay.device.rack_id
  1431. class BaseDeviceCSVForm(forms.ModelForm):
  1432. device_role = forms.ModelChoiceField(
  1433. queryset=DeviceRole.objects.all(),
  1434. to_field_name='name',
  1435. help_text='Name of assigned role',
  1436. error_messages={
  1437. 'invalid_choice': 'Invalid device role.',
  1438. }
  1439. )
  1440. tenant = forms.ModelChoiceField(
  1441. queryset=Tenant.objects.all(),
  1442. required=False,
  1443. to_field_name='name',
  1444. help_text='Name of assigned tenant',
  1445. error_messages={
  1446. 'invalid_choice': 'Tenant not found.',
  1447. }
  1448. )
  1449. manufacturer = forms.ModelChoiceField(
  1450. queryset=Manufacturer.objects.all(),
  1451. to_field_name='name',
  1452. help_text='Device type manufacturer',
  1453. error_messages={
  1454. 'invalid_choice': 'Invalid manufacturer.',
  1455. }
  1456. )
  1457. model_name = forms.CharField(
  1458. help_text='Device type model name'
  1459. )
  1460. platform = forms.ModelChoiceField(
  1461. queryset=Platform.objects.all(),
  1462. required=False,
  1463. to_field_name='name',
  1464. help_text='Name of assigned platform',
  1465. error_messages={
  1466. 'invalid_choice': 'Invalid platform.',
  1467. }
  1468. )
  1469. status = CSVChoiceField(
  1470. choices=DeviceStatusChoices,
  1471. help_text='Operational status'
  1472. )
  1473. class Meta:
  1474. fields = []
  1475. model = Device
  1476. help_texts = {
  1477. 'name': 'Device name',
  1478. }
  1479. def clean(self):
  1480. super().clean()
  1481. manufacturer = self.cleaned_data.get('manufacturer')
  1482. model_name = self.cleaned_data.get('model_name')
  1483. # Validate device type
  1484. if manufacturer and model_name:
  1485. try:
  1486. self.instance.device_type = DeviceType.objects.get(manufacturer=manufacturer, model=model_name)
  1487. except DeviceType.DoesNotExist:
  1488. raise forms.ValidationError("Device type {} {} not found".format(manufacturer, model_name))
  1489. class DeviceCSVForm(BaseDeviceCSVForm):
  1490. site = forms.ModelChoiceField(
  1491. queryset=Site.objects.all(),
  1492. to_field_name='name',
  1493. help_text='Name of parent site',
  1494. error_messages={
  1495. 'invalid_choice': 'Invalid site name.',
  1496. }
  1497. )
  1498. rack_group = forms.CharField(
  1499. required=False,
  1500. help_text='Parent rack\'s group (if any)'
  1501. )
  1502. rack_name = forms.CharField(
  1503. required=False,
  1504. help_text='Name of parent rack'
  1505. )
  1506. face = CSVChoiceField(
  1507. choices=DeviceFaceChoices,
  1508. required=False,
  1509. help_text='Mounted rack face'
  1510. )
  1511. cluster = forms.ModelChoiceField(
  1512. queryset=Cluster.objects.all(),
  1513. to_field_name='name',
  1514. required=False,
  1515. help_text='Virtualization cluster',
  1516. error_messages={
  1517. 'invalid_choice': 'Invalid cluster name.',
  1518. }
  1519. )
  1520. class Meta(BaseDeviceCSVForm.Meta):
  1521. fields = [
  1522. 'name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag', 'status',
  1523. 'site', 'rack_group', 'rack_name', 'position', 'face', 'cluster', 'comments',
  1524. ]
  1525. def clean(self):
  1526. super().clean()
  1527. site = self.cleaned_data.get('site')
  1528. rack_group = self.cleaned_data.get('rack_group')
  1529. rack_name = self.cleaned_data.get('rack_name')
  1530. # Validate rack
  1531. if site and rack_group and rack_name:
  1532. try:
  1533. self.instance.rack = Rack.objects.get(site=site, group__name=rack_group, name=rack_name)
  1534. except Rack.DoesNotExist:
  1535. raise forms.ValidationError("Rack {} not found in site {} group {}".format(rack_name, site, rack_group))
  1536. elif site and rack_name:
  1537. try:
  1538. self.instance.rack = Rack.objects.get(site=site, group__isnull=True, name=rack_name)
  1539. except Rack.DoesNotExist:
  1540. raise forms.ValidationError("Rack {} not found in site {} (no group)".format(rack_name, site))
  1541. class ChildDeviceCSVForm(BaseDeviceCSVForm):
  1542. parent = FlexibleModelChoiceField(
  1543. queryset=Device.objects.all(),
  1544. to_field_name='name',
  1545. help_text='Name or ID of parent device',
  1546. error_messages={
  1547. 'invalid_choice': 'Parent device not found.',
  1548. }
  1549. )
  1550. device_bay_name = forms.CharField(
  1551. help_text='Name of device bay',
  1552. )
  1553. cluster = forms.ModelChoiceField(
  1554. queryset=Cluster.objects.all(),
  1555. to_field_name='name',
  1556. required=False,
  1557. help_text='Virtualization cluster',
  1558. error_messages={
  1559. 'invalid_choice': 'Invalid cluster name.',
  1560. }
  1561. )
  1562. class Meta(BaseDeviceCSVForm.Meta):
  1563. fields = [
  1564. 'name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag', 'status',
  1565. 'parent', 'device_bay_name', 'cluster', 'comments',
  1566. ]
  1567. def clean(self):
  1568. super().clean()
  1569. parent = self.cleaned_data.get('parent')
  1570. device_bay_name = self.cleaned_data.get('device_bay_name')
  1571. # Validate device bay
  1572. if parent and device_bay_name:
  1573. try:
  1574. self.instance.parent_bay = DeviceBay.objects.get(device=parent, name=device_bay_name)
  1575. # Inherit site and rack from parent device
  1576. self.instance.site = parent.site
  1577. self.instance.rack = parent.rack
  1578. except DeviceBay.DoesNotExist:
  1579. raise forms.ValidationError("Parent device/bay ({} {}) not found".format(parent, device_bay_name))
  1580. class DeviceBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
  1581. pk = forms.ModelMultipleChoiceField(
  1582. queryset=Device.objects.all(),
  1583. widget=forms.MultipleHiddenInput()
  1584. )
  1585. device_type = forms.ModelChoiceField(
  1586. queryset=DeviceType.objects.all(),
  1587. required=False,
  1588. label='Type',
  1589. widget=APISelect(
  1590. api_url="/api/dcim/device-types/",
  1591. display_field='display_name'
  1592. )
  1593. )
  1594. device_role = forms.ModelChoiceField(
  1595. queryset=DeviceRole.objects.all(),
  1596. required=False,
  1597. label='Role',
  1598. widget=APISelect(
  1599. api_url="/api/dcim/device-roles/"
  1600. )
  1601. )
  1602. tenant = forms.ModelChoiceField(
  1603. queryset=Tenant.objects.all(),
  1604. required=False,
  1605. widget=APISelect(
  1606. api_url="/api/tenancy/tenants/"
  1607. )
  1608. )
  1609. platform = forms.ModelChoiceField(
  1610. queryset=Platform.objects.all(),
  1611. required=False,
  1612. widget=APISelect(
  1613. api_url="/api/dcim/platforms/"
  1614. )
  1615. )
  1616. status = forms.ChoiceField(
  1617. choices=add_blank_choice(DeviceStatusChoices),
  1618. required=False,
  1619. initial='',
  1620. widget=StaticSelect2()
  1621. )
  1622. serial = forms.CharField(
  1623. max_length=50,
  1624. required=False,
  1625. label='Serial Number'
  1626. )
  1627. class Meta:
  1628. nullable_fields = [
  1629. 'tenant', 'platform', 'serial',
  1630. ]
  1631. class DeviceFilterForm(BootstrapMixin, LocalConfigContextFilterForm, TenancyFilterForm, CustomFieldFilterForm):
  1632. model = Device
  1633. field_order = [
  1634. 'q', 'region', 'site', 'group_id', 'rack_id', 'status', 'role', 'tenant_group', 'tenant',
  1635. 'manufacturer_id', 'device_type_id', 'mac_address', 'has_primary_ip',
  1636. ]
  1637. q = forms.CharField(
  1638. required=False,
  1639. label='Search'
  1640. )
  1641. region = FilterChoiceField(
  1642. queryset=Region.objects.all(),
  1643. to_field_name='slug',
  1644. required=False,
  1645. widget=APISelectMultiple(
  1646. api_url="/api/dcim/regions/",
  1647. value_field="slug",
  1648. filter_for={
  1649. 'site': 'region'
  1650. }
  1651. )
  1652. )
  1653. site = FilterChoiceField(
  1654. queryset=Site.objects.all(),
  1655. to_field_name='slug',
  1656. widget=APISelectMultiple(
  1657. api_url="/api/dcim/sites/",
  1658. value_field="slug",
  1659. filter_for={
  1660. 'group_id': 'site',
  1661. 'rack_id': 'site',
  1662. }
  1663. )
  1664. )
  1665. group_id = FilterChoiceField(
  1666. queryset=RackGroup.objects.prefetch_related(
  1667. 'site'
  1668. ),
  1669. label='Rack group',
  1670. widget=APISelectMultiple(
  1671. api_url="/api/dcim/rack-groups/",
  1672. filter_for={
  1673. 'rack_id': 'group_id',
  1674. }
  1675. )
  1676. )
  1677. rack_id = FilterChoiceField(
  1678. queryset=Rack.objects.all(),
  1679. label='Rack',
  1680. null_label='-- None --',
  1681. widget=APISelectMultiple(
  1682. api_url="/api/dcim/racks/",
  1683. null_option=True,
  1684. )
  1685. )
  1686. role = FilterChoiceField(
  1687. queryset=DeviceRole.objects.all(),
  1688. to_field_name='slug',
  1689. widget=APISelectMultiple(
  1690. api_url="/api/dcim/device-roles/",
  1691. value_field="slug",
  1692. )
  1693. )
  1694. manufacturer_id = FilterChoiceField(
  1695. queryset=Manufacturer.objects.all(),
  1696. label='Manufacturer',
  1697. widget=APISelectMultiple(
  1698. api_url="/api/dcim/manufacturers/",
  1699. filter_for={
  1700. 'device_type_id': 'manufacturer_id',
  1701. }
  1702. )
  1703. )
  1704. device_type_id = FilterChoiceField(
  1705. queryset=DeviceType.objects.prefetch_related(
  1706. 'manufacturer'
  1707. ),
  1708. label='Model',
  1709. widget=APISelectMultiple(
  1710. api_url="/api/dcim/device-types/",
  1711. display_field="model",
  1712. )
  1713. )
  1714. platform = FilterChoiceField(
  1715. queryset=Platform.objects.all(),
  1716. to_field_name='slug',
  1717. null_label='-- None --',
  1718. widget=APISelectMultiple(
  1719. api_url="/api/dcim/platforms/",
  1720. value_field="slug",
  1721. null_option=True,
  1722. )
  1723. )
  1724. status = forms.MultipleChoiceField(
  1725. choices=DeviceStatusChoices,
  1726. required=False,
  1727. widget=StaticSelect2Multiple()
  1728. )
  1729. mac_address = forms.CharField(
  1730. required=False,
  1731. label='MAC address'
  1732. )
  1733. has_primary_ip = forms.NullBooleanField(
  1734. required=False,
  1735. label='Has a primary IP',
  1736. widget=StaticSelect2(
  1737. choices=BOOLEAN_WITH_BLANK_CHOICES
  1738. )
  1739. )
  1740. virtual_chassis_member = forms.NullBooleanField(
  1741. required=False,
  1742. label='Virtual chassis member',
  1743. widget=StaticSelect2(
  1744. choices=BOOLEAN_WITH_BLANK_CHOICES
  1745. )
  1746. )
  1747. console_ports = forms.NullBooleanField(
  1748. required=False,
  1749. label='Has console ports',
  1750. widget=StaticSelect2(
  1751. choices=BOOLEAN_WITH_BLANK_CHOICES
  1752. )
  1753. )
  1754. console_server_ports = forms.NullBooleanField(
  1755. required=False,
  1756. label='Has console server ports',
  1757. widget=StaticSelect2(
  1758. choices=BOOLEAN_WITH_BLANK_CHOICES
  1759. )
  1760. )
  1761. power_ports = forms.NullBooleanField(
  1762. required=False,
  1763. label='Has power ports',
  1764. widget=StaticSelect2(
  1765. choices=BOOLEAN_WITH_BLANK_CHOICES
  1766. )
  1767. )
  1768. power_outlets = forms.NullBooleanField(
  1769. required=False,
  1770. label='Has power outlets',
  1771. widget=StaticSelect2(
  1772. choices=BOOLEAN_WITH_BLANK_CHOICES
  1773. )
  1774. )
  1775. interfaces = forms.NullBooleanField(
  1776. required=False,
  1777. label='Has interfaces',
  1778. widget=StaticSelect2(
  1779. choices=BOOLEAN_WITH_BLANK_CHOICES
  1780. )
  1781. )
  1782. pass_through_ports = forms.NullBooleanField(
  1783. required=False,
  1784. label='Has pass-through ports',
  1785. widget=StaticSelect2(
  1786. choices=BOOLEAN_WITH_BLANK_CHOICES
  1787. )
  1788. )
  1789. #
  1790. # Bulk device component creation
  1791. #
  1792. class DeviceBulkAddComponentForm(BootstrapMixin, forms.Form):
  1793. pk = forms.ModelMultipleChoiceField(
  1794. queryset=Device.objects.all(),
  1795. widget=forms.MultipleHiddenInput()
  1796. )
  1797. name_pattern = ExpandableNameField(
  1798. label='Name'
  1799. )
  1800. class DeviceBulkAddInterfaceForm(DeviceBulkAddComponentForm):
  1801. type = forms.ChoiceField(
  1802. choices=InterfaceTypeChoices,
  1803. widget=StaticSelect2()
  1804. )
  1805. enabled = forms.BooleanField(
  1806. required=False,
  1807. initial=True
  1808. )
  1809. mtu = forms.IntegerField(
  1810. required=False,
  1811. min_value=1,
  1812. max_value=32767,
  1813. label='MTU'
  1814. )
  1815. mgmt_only = forms.BooleanField(
  1816. required=False,
  1817. label='Management only'
  1818. )
  1819. description = forms.CharField(
  1820. max_length=100,
  1821. required=False
  1822. )
  1823. #
  1824. # Console ports
  1825. #
  1826. class ConsolePortFilterForm(DeviceComponentFilterForm):
  1827. model = ConsolePort
  1828. class ConsolePortForm(BootstrapMixin, forms.ModelForm):
  1829. tags = TagField(
  1830. required=False
  1831. )
  1832. class Meta:
  1833. model = ConsolePort
  1834. fields = [
  1835. 'device', 'name', 'type', 'description', 'tags',
  1836. ]
  1837. widgets = {
  1838. 'device': forms.HiddenInput(),
  1839. }
  1840. class ConsolePortCreateForm(ComponentForm):
  1841. name_pattern = ExpandableNameField(
  1842. label='Name'
  1843. )
  1844. type = forms.ChoiceField(
  1845. choices=add_blank_choice(ConsolePortTypeChoices),
  1846. required=False,
  1847. widget=StaticSelect2()
  1848. )
  1849. description = forms.CharField(
  1850. max_length=100,
  1851. required=False
  1852. )
  1853. tags = TagField(
  1854. required=False
  1855. )
  1856. class ConsolePortCSVForm(forms.ModelForm):
  1857. device = FlexibleModelChoiceField(
  1858. queryset=Device.objects.all(),
  1859. to_field_name='name',
  1860. help_text='Name or ID of device',
  1861. error_messages={
  1862. 'invalid_choice': 'Device not found.',
  1863. }
  1864. )
  1865. class Meta:
  1866. model = ConsolePort
  1867. fields = ConsolePort.csv_headers
  1868. #
  1869. # Console server ports
  1870. #
  1871. class ConsoleServerPortFilterForm(DeviceComponentFilterForm):
  1872. model = ConsoleServerPort
  1873. class ConsoleServerPortForm(BootstrapMixin, forms.ModelForm):
  1874. tags = TagField(
  1875. required=False
  1876. )
  1877. class Meta:
  1878. model = ConsoleServerPort
  1879. fields = [
  1880. 'device', 'name', 'type', 'description', 'tags',
  1881. ]
  1882. widgets = {
  1883. 'device': forms.HiddenInput(),
  1884. }
  1885. class ConsoleServerPortCreateForm(ComponentForm):
  1886. name_pattern = ExpandableNameField(
  1887. label='Name'
  1888. )
  1889. type = forms.ChoiceField(
  1890. choices=add_blank_choice(ConsolePortTypeChoices),
  1891. required=False,
  1892. widget=StaticSelect2()
  1893. )
  1894. description = forms.CharField(
  1895. max_length=100,
  1896. required=False
  1897. )
  1898. tags = TagField(
  1899. required=False
  1900. )
  1901. class ConsoleServerPortBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
  1902. pk = forms.ModelMultipleChoiceField(
  1903. queryset=ConsoleServerPort.objects.all(),
  1904. widget=forms.MultipleHiddenInput()
  1905. )
  1906. type = forms.ChoiceField(
  1907. choices=add_blank_choice(ConsolePortTypeChoices),
  1908. required=False,
  1909. widget=StaticSelect2()
  1910. )
  1911. description = forms.CharField(
  1912. max_length=100,
  1913. required=False
  1914. )
  1915. class Meta:
  1916. nullable_fields = [
  1917. 'description',
  1918. ]
  1919. class ConsoleServerPortBulkRenameForm(BulkRenameForm):
  1920. pk = forms.ModelMultipleChoiceField(
  1921. queryset=ConsoleServerPort.objects.all(),
  1922. widget=forms.MultipleHiddenInput()
  1923. )
  1924. class ConsoleServerPortBulkDisconnectForm(ConfirmationForm):
  1925. pk = forms.ModelMultipleChoiceField(
  1926. queryset=ConsoleServerPort.objects.all(),
  1927. widget=forms.MultipleHiddenInput()
  1928. )
  1929. class ConsoleServerPortCSVForm(forms.ModelForm):
  1930. device = FlexibleModelChoiceField(
  1931. queryset=Device.objects.all(),
  1932. to_field_name='name',
  1933. help_text='Name or ID of device',
  1934. error_messages={
  1935. 'invalid_choice': 'Device not found.',
  1936. }
  1937. )
  1938. class Meta:
  1939. model = ConsoleServerPort
  1940. fields = ConsoleServerPort.csv_headers
  1941. #
  1942. # Power ports
  1943. #
  1944. class PowerPortFilterForm(DeviceComponentFilterForm):
  1945. model = PowerPort
  1946. class PowerPortForm(BootstrapMixin, forms.ModelForm):
  1947. tags = TagField(
  1948. required=False
  1949. )
  1950. class Meta:
  1951. model = PowerPort
  1952. fields = [
  1953. 'device', 'name', 'type', 'maximum_draw', 'allocated_draw', 'description', 'tags',
  1954. ]
  1955. widgets = {
  1956. 'device': forms.HiddenInput(),
  1957. }
  1958. class PowerPortCreateForm(ComponentForm):
  1959. name_pattern = ExpandableNameField(
  1960. label='Name'
  1961. )
  1962. type = forms.ChoiceField(
  1963. choices=add_blank_choice(PowerPortTypeChoices),
  1964. required=False,
  1965. widget=StaticSelect2()
  1966. )
  1967. maximum_draw = forms.IntegerField(
  1968. min_value=1,
  1969. required=False,
  1970. help_text="Maximum draw in watts"
  1971. )
  1972. allocated_draw = forms.IntegerField(
  1973. min_value=1,
  1974. required=False,
  1975. help_text="Allocated draw in watts"
  1976. )
  1977. description = forms.CharField(
  1978. max_length=100,
  1979. required=False
  1980. )
  1981. tags = TagField(
  1982. required=False
  1983. )
  1984. class PowerPortCSVForm(forms.ModelForm):
  1985. device = FlexibleModelChoiceField(
  1986. queryset=Device.objects.all(),
  1987. to_field_name='name',
  1988. help_text='Name or ID of device',
  1989. error_messages={
  1990. 'invalid_choice': 'Device not found.',
  1991. }
  1992. )
  1993. class Meta:
  1994. model = PowerPort
  1995. fields = PowerPort.csv_headers
  1996. #
  1997. # Power outlets
  1998. #
  1999. class PowerOutletFilterForm(DeviceComponentFilterForm):
  2000. model = PowerOutlet
  2001. class PowerOutletForm(BootstrapMixin, forms.ModelForm):
  2002. power_port = forms.ModelChoiceField(
  2003. queryset=PowerPort.objects.all(),
  2004. required=False
  2005. )
  2006. tags = TagField(
  2007. required=False
  2008. )
  2009. class Meta:
  2010. model = PowerOutlet
  2011. fields = [
  2012. 'device', 'name', 'type', 'power_port', 'feed_leg', 'description', 'tags',
  2013. ]
  2014. widgets = {
  2015. 'device': forms.HiddenInput(),
  2016. }
  2017. def __init__(self, *args, **kwargs):
  2018. super().__init__(*args, **kwargs)
  2019. # Limit power_port choices to the local device
  2020. if hasattr(self.instance, 'device'):
  2021. self.fields['power_port'].queryset = PowerPort.objects.filter(
  2022. device=self.instance.device
  2023. )
  2024. class PowerOutletCreateForm(ComponentForm):
  2025. name_pattern = ExpandableNameField(
  2026. label='Name'
  2027. )
  2028. type = forms.ChoiceField(
  2029. choices=add_blank_choice(PowerOutletTypeChoices),
  2030. required=False,
  2031. widget=StaticSelect2()
  2032. )
  2033. power_port = forms.ModelChoiceField(
  2034. queryset=PowerPort.objects.all(),
  2035. required=False
  2036. )
  2037. feed_leg = forms.ChoiceField(
  2038. choices=add_blank_choice(PowerOutletFeedLegChoices),
  2039. required=False
  2040. )
  2041. description = forms.CharField(
  2042. max_length=100,
  2043. required=False
  2044. )
  2045. tags = TagField(
  2046. required=False
  2047. )
  2048. def __init__(self, *args, **kwargs):
  2049. super().__init__(*args, **kwargs)
  2050. # Limit power_port choices to those on the parent device
  2051. self.fields['power_port'].queryset = PowerPort.objects.filter(device=self.parent)
  2052. class PowerOutletCSVForm(forms.ModelForm):
  2053. device = FlexibleModelChoiceField(
  2054. queryset=Device.objects.all(),
  2055. to_field_name='name',
  2056. help_text='Name or ID of device',
  2057. error_messages={
  2058. 'invalid_choice': 'Device not found.',
  2059. }
  2060. )
  2061. power_port = FlexibleModelChoiceField(
  2062. queryset=PowerPort.objects.all(),
  2063. required=False,
  2064. to_field_name='name',
  2065. help_text='Name or ID of Power Port',
  2066. error_messages={
  2067. 'invalid_choice': 'Power Port not found.',
  2068. }
  2069. )
  2070. feed_leg = CSVChoiceField(
  2071. choices=PowerOutletFeedLegChoices,
  2072. required=False,
  2073. )
  2074. class Meta:
  2075. model = PowerOutlet
  2076. fields = PowerOutlet.csv_headers
  2077. def __init__(self, *args, **kwargs):
  2078. super().__init__(*args, **kwargs)
  2079. # Limit PowerPort choices to those belonging to this device (or VC master)
  2080. if self.is_bound:
  2081. try:
  2082. device = self.fields['device'].to_python(self.data['device'])
  2083. except forms.ValidationError:
  2084. device = None
  2085. else:
  2086. try:
  2087. device = self.instance.device
  2088. except Device.DoesNotExist:
  2089. device = None
  2090. if device:
  2091. self.fields['power_port'].queryset = PowerPort.objects.filter(
  2092. device__in=[device, device.get_vc_master()]
  2093. )
  2094. else:
  2095. self.fields['power_port'].queryset = PowerPort.objects.none()
  2096. class PowerOutletBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
  2097. pk = forms.ModelMultipleChoiceField(
  2098. queryset=PowerOutlet.objects.all(),
  2099. widget=forms.MultipleHiddenInput()
  2100. )
  2101. type = forms.ChoiceField(
  2102. choices=PowerOutletTypeChoices,
  2103. required=False
  2104. )
  2105. feed_leg = forms.ChoiceField(
  2106. choices=add_blank_choice(PowerOutletFeedLegChoices),
  2107. required=False,
  2108. )
  2109. power_port = forms.ModelChoiceField(
  2110. queryset=PowerPort.objects.all(),
  2111. required=False
  2112. )
  2113. description = forms.CharField(
  2114. max_length=100,
  2115. required=False
  2116. )
  2117. class Meta:
  2118. nullable_fields = [
  2119. 'type', 'feed_leg', 'power_port', 'description',
  2120. ]
  2121. def __init__(self, *args, **kwargs):
  2122. super().__init__(*args, **kwargs)
  2123. # Limit power_port queryset to PowerPorts which belong to the parent Device
  2124. self.fields['power_port'].queryset = PowerPort.objects.filter(device=self.parent_obj)
  2125. class PowerOutletBulkRenameForm(BulkRenameForm):
  2126. pk = forms.ModelMultipleChoiceField(
  2127. queryset=PowerOutlet.objects.all(),
  2128. widget=forms.MultipleHiddenInput
  2129. )
  2130. class PowerOutletBulkDisconnectForm(ConfirmationForm):
  2131. pk = forms.ModelMultipleChoiceField(
  2132. queryset=PowerOutlet.objects.all(),
  2133. widget=forms.MultipleHiddenInput
  2134. )
  2135. #
  2136. # Interfaces
  2137. #
  2138. class InterfaceFilterForm(DeviceComponentFilterForm):
  2139. model = Interface
  2140. class InterfaceForm(InterfaceCommonForm, BootstrapMixin, forms.ModelForm):
  2141. untagged_vlan = forms.ModelChoiceField(
  2142. queryset=VLAN.objects.all(),
  2143. required=False,
  2144. widget=APISelect(
  2145. api_url="/api/ipam/vlans/",
  2146. display_field='display_name',
  2147. full=True
  2148. )
  2149. )
  2150. tagged_vlans = forms.ModelMultipleChoiceField(
  2151. queryset=VLAN.objects.all(),
  2152. required=False,
  2153. widget=APISelectMultiple(
  2154. api_url="/api/ipam/vlans/",
  2155. display_field='display_name',
  2156. full=True
  2157. )
  2158. )
  2159. tags = TagField(
  2160. required=False
  2161. )
  2162. class Meta:
  2163. model = Interface
  2164. fields = [
  2165. 'device', 'name', 'type', 'enabled', 'lag', 'mac_address', 'mtu', 'mgmt_only', 'description',
  2166. 'mode', 'untagged_vlan', 'tagged_vlans', 'tags',
  2167. ]
  2168. widgets = {
  2169. 'device': forms.HiddenInput(),
  2170. 'type': StaticSelect2(),
  2171. 'lag': StaticSelect2(),
  2172. 'mode': StaticSelect2(),
  2173. }
  2174. labels = {
  2175. 'mode': '802.1Q Mode',
  2176. }
  2177. help_texts = {
  2178. 'mode': INTERFACE_MODE_HELP_TEXT,
  2179. }
  2180. def __init__(self, *args, **kwargs):
  2181. super().__init__(*args, **kwargs)
  2182. # Limit LAG choices to interfaces belonging to this device (or VC master)
  2183. if self.is_bound:
  2184. device = Device.objects.get(pk=self.data['device'])
  2185. self.fields['lag'].queryset = Interface.objects.filter(
  2186. device__in=[device, device.get_vc_master()],
  2187. type=InterfaceTypeChoices.TYPE_LAG
  2188. )
  2189. else:
  2190. device = self.instance.device
  2191. self.fields['lag'].queryset = Interface.objects.filter(
  2192. device__in=[self.instance.device, self.instance.device.get_vc_master()],
  2193. type=InterfaceTypeChoices.TYPE_LAG
  2194. )
  2195. # Limit VLan choices to those in: global vlans, global groups, the current site's group, the current site
  2196. vlan_choices = []
  2197. global_vlans = VLAN.objects.filter(site=None, group=None)
  2198. vlan_choices.append(
  2199. ('Global', [(vlan.pk, vlan) for vlan in global_vlans])
  2200. )
  2201. for group in VLANGroup.objects.filter(site=None):
  2202. global_group_vlans = VLAN.objects.filter(group=group)
  2203. vlan_choices.append(
  2204. (group.name, [(vlan.pk, vlan) for vlan in global_group_vlans])
  2205. )
  2206. site = getattr(self.instance.parent, 'site', None)
  2207. if site is not None:
  2208. # Add non-grouped site VLANs
  2209. site_vlans = VLAN.objects.filter(site=site, group=None)
  2210. vlan_choices.append((site.name, [(vlan.pk, vlan) for vlan in site_vlans]))
  2211. # Add grouped site VLANs
  2212. for group in VLANGroup.objects.filter(site=site):
  2213. site_group_vlans = VLAN.objects.filter(group=group)
  2214. vlan_choices.append((
  2215. '{} / {}'.format(group.site.name, group.name),
  2216. [(vlan.pk, vlan) for vlan in site_group_vlans]
  2217. ))
  2218. self.fields['untagged_vlan'].choices = [(None, '---------')] + vlan_choices
  2219. self.fields['tagged_vlans'].choices = vlan_choices
  2220. class InterfaceCreateForm(InterfaceCommonForm, ComponentForm, forms.Form):
  2221. name_pattern = ExpandableNameField(
  2222. label='Name'
  2223. )
  2224. type = forms.ChoiceField(
  2225. choices=InterfaceTypeChoices,
  2226. widget=StaticSelect2(),
  2227. )
  2228. enabled = forms.BooleanField(
  2229. required=False
  2230. )
  2231. lag = forms.ModelChoiceField(
  2232. queryset=Interface.objects.all(),
  2233. required=False,
  2234. label='Parent LAG',
  2235. widget=StaticSelect2(),
  2236. )
  2237. mtu = forms.IntegerField(
  2238. required=False,
  2239. min_value=1,
  2240. max_value=32767,
  2241. label='MTU'
  2242. )
  2243. mac_address = forms.CharField(
  2244. required=False,
  2245. label='MAC Address'
  2246. )
  2247. mgmt_only = forms.BooleanField(
  2248. required=False,
  2249. label='Management only',
  2250. help_text='This interface is used only for out-of-band management'
  2251. )
  2252. description = forms.CharField(
  2253. max_length=100,
  2254. required=False
  2255. )
  2256. mode = forms.ChoiceField(
  2257. choices=add_blank_choice(InterfaceModeChoices),
  2258. required=False,
  2259. widget=StaticSelect2(),
  2260. )
  2261. tags = TagField(
  2262. required=False
  2263. )
  2264. untagged_vlan = forms.ModelChoiceField(
  2265. queryset=VLAN.objects.all(),
  2266. required=False,
  2267. widget=APISelect(
  2268. api_url="/api/ipam/vlans/",
  2269. display_field='display_name',
  2270. full=True
  2271. )
  2272. )
  2273. tagged_vlans = forms.ModelMultipleChoiceField(
  2274. queryset=VLAN.objects.all(),
  2275. required=False,
  2276. widget=APISelectMultiple(
  2277. api_url="/api/ipam/vlans/",
  2278. display_field='display_name',
  2279. full=True
  2280. )
  2281. )
  2282. def __init__(self, *args, **kwargs):
  2283. # Set interfaces enabled by default
  2284. kwargs['initial'] = kwargs.get('initial', {}).copy()
  2285. kwargs['initial'].update({'enabled': True})
  2286. super().__init__(*args, **kwargs)
  2287. # Limit LAG choices to interfaces belonging to this device (or its VC master)
  2288. if self.parent is not None:
  2289. self.fields['lag'].queryset = Interface.objects.filter(
  2290. device__in=[self.parent, self.parent.get_vc_master()],
  2291. type=InterfaceTypeChoices.TYPE_LAG
  2292. )
  2293. else:
  2294. self.fields['lag'].queryset = Interface.objects.none()
  2295. # Limit VLan choices to those in: global vlans, global groups, the current site's group, the current site
  2296. vlan_choices = []
  2297. global_vlans = VLAN.objects.filter(site=None, group=None)
  2298. vlan_choices.append(
  2299. ('Global', [(vlan.pk, vlan) for vlan in global_vlans])
  2300. )
  2301. for group in VLANGroup.objects.filter(site=None):
  2302. global_group_vlans = VLAN.objects.filter(group=group)
  2303. vlan_choices.append(
  2304. (group.name, [(vlan.pk, vlan) for vlan in global_group_vlans])
  2305. )
  2306. site = getattr(self.parent, 'site', None)
  2307. if site is not None:
  2308. # Add non-grouped site VLANs
  2309. site_vlans = VLAN.objects.filter(site=site, group=None)
  2310. vlan_choices.append((site.name, [(vlan.pk, vlan) for vlan in site_vlans]))
  2311. # Add grouped site VLANs
  2312. for group in VLANGroup.objects.filter(site=site):
  2313. site_group_vlans = VLAN.objects.filter(group=group)
  2314. vlan_choices.append((
  2315. '{} / {}'.format(group.site.name, group.name),
  2316. [(vlan.pk, vlan) for vlan in site_group_vlans]
  2317. ))
  2318. self.fields['untagged_vlan'].choices = [(None, '---------')] + vlan_choices
  2319. self.fields['tagged_vlans'].choices = vlan_choices
  2320. class InterfaceCSVForm(forms.ModelForm):
  2321. device = FlexibleModelChoiceField(
  2322. queryset=Device.objects.all(),
  2323. required=False,
  2324. to_field_name='name',
  2325. help_text='Name or ID of device',
  2326. error_messages={
  2327. 'invalid_choice': 'Device not found.',
  2328. }
  2329. )
  2330. virtual_machine = FlexibleModelChoiceField(
  2331. queryset=VirtualMachine.objects.all(),
  2332. required=False,
  2333. to_field_name='name',
  2334. help_text='Name or ID of virtual machine',
  2335. error_messages={
  2336. 'invalid_choice': 'Virtual machine not found.',
  2337. }
  2338. )
  2339. lag = FlexibleModelChoiceField(
  2340. queryset=Interface.objects.all(),
  2341. required=False,
  2342. to_field_name='name',
  2343. help_text='Name or ID of LAG interface',
  2344. error_messages={
  2345. 'invalid_choice': 'LAG interface not found.',
  2346. }
  2347. )
  2348. type = CSVChoiceField(
  2349. choices=InterfaceTypeChoices,
  2350. )
  2351. mode = CSVChoiceField(
  2352. choices=InterfaceModeChoices,
  2353. required=False,
  2354. )
  2355. class Meta:
  2356. model = Interface
  2357. fields = Interface.csv_headers
  2358. def __init__(self, *args, **kwargs):
  2359. super().__init__(*args, **kwargs)
  2360. # Limit LAG choices to interfaces belonging to this device (or VC master)
  2361. if self.is_bound:
  2362. try:
  2363. device = self.fields['device'].to_python(self.data['device'])
  2364. except forms.ValidationError:
  2365. device = None
  2366. else:
  2367. device = self.instance.device
  2368. if device:
  2369. self.fields['lag'].queryset = Interface.objects.filter(
  2370. device__in=[device, device.get_vc_master()], type=InterfaceTypeChoices.TYPE_LAG
  2371. )
  2372. else:
  2373. self.fields['lag'].queryset = Interface.objects.none()
  2374. def clean_enabled(self):
  2375. # Make sure enabled is True when it's not included in the uploaded data
  2376. if 'enabled' not in self.data:
  2377. return True
  2378. else:
  2379. return self.cleaned_data['enabled']
  2380. class InterfaceBulkEditForm(InterfaceCommonForm, BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
  2381. pk = forms.ModelMultipleChoiceField(
  2382. queryset=Interface.objects.all(),
  2383. widget=forms.MultipleHiddenInput()
  2384. )
  2385. type = forms.ChoiceField(
  2386. choices=add_blank_choice(InterfaceTypeChoices),
  2387. required=False,
  2388. widget=StaticSelect2()
  2389. )
  2390. enabled = forms.NullBooleanField(
  2391. required=False,
  2392. widget=BulkEditNullBooleanSelect()
  2393. )
  2394. lag = forms.ModelChoiceField(
  2395. queryset=Interface.objects.all(),
  2396. required=False,
  2397. label='Parent LAG',
  2398. widget=StaticSelect2()
  2399. )
  2400. mac_address = forms.CharField(
  2401. required=False,
  2402. label='MAC Address'
  2403. )
  2404. mtu = forms.IntegerField(
  2405. required=False,
  2406. min_value=1,
  2407. max_value=32767,
  2408. label='MTU'
  2409. )
  2410. mgmt_only = forms.NullBooleanField(
  2411. required=False,
  2412. widget=BulkEditNullBooleanSelect(),
  2413. label='Management only'
  2414. )
  2415. description = forms.CharField(
  2416. max_length=100,
  2417. required=False
  2418. )
  2419. mode = forms.ChoiceField(
  2420. choices=add_blank_choice(InterfaceModeChoices),
  2421. required=False,
  2422. widget=StaticSelect2()
  2423. )
  2424. untagged_vlan = forms.ModelChoiceField(
  2425. queryset=VLAN.objects.all(),
  2426. required=False,
  2427. widget=APISelect(
  2428. api_url="/api/ipam/vlans/",
  2429. display_field='display_name',
  2430. full=True
  2431. )
  2432. )
  2433. tagged_vlans = forms.ModelMultipleChoiceField(
  2434. queryset=VLAN.objects.all(),
  2435. required=False,
  2436. widget=APISelectMultiple(
  2437. api_url="/api/ipam/vlans/",
  2438. display_field='display_name',
  2439. full=True
  2440. )
  2441. )
  2442. class Meta:
  2443. nullable_fields = [
  2444. 'lag', 'mac_address', 'mtu', 'description', 'mode', 'untagged_vlan', 'tagged_vlans'
  2445. ]
  2446. def __init__(self, *args, **kwargs):
  2447. super().__init__(*args, **kwargs)
  2448. # Limit LAG choices to interfaces which belong to the parent device (or VC master)
  2449. device = self.parent_obj
  2450. if device is not None:
  2451. self.fields['lag'].queryset = Interface.objects.filter(
  2452. device__in=[device, device.get_vc_master()],
  2453. type=InterfaceTypeChoices.TYPE_LAG
  2454. )
  2455. else:
  2456. self.fields['lag'].choices = []
  2457. # Limit VLan choices to those in: global vlans, global groups, the current site's group, the current site
  2458. vlan_choices = []
  2459. global_vlans = VLAN.objects.filter(site=None, group=None)
  2460. vlan_choices.append(
  2461. ('Global', [(vlan.pk, vlan) for vlan in global_vlans])
  2462. )
  2463. for group in VLANGroup.objects.filter(site=None):
  2464. global_group_vlans = VLAN.objects.filter(group=group)
  2465. vlan_choices.append(
  2466. (group.name, [(vlan.pk, vlan) for vlan in global_group_vlans])
  2467. )
  2468. if self.parent_obj is not None:
  2469. site = getattr(self.parent_obj, 'site', None)
  2470. if site is not None:
  2471. # Add non-grouped site VLANs
  2472. site_vlans = VLAN.objects.filter(site=site, group=None)
  2473. vlan_choices.append((site.name, [(vlan.pk, vlan) for vlan in site_vlans]))
  2474. # Add grouped site VLANs
  2475. for group in VLANGroup.objects.filter(site=site):
  2476. site_group_vlans = VLAN.objects.filter(group=group)
  2477. vlan_choices.append((
  2478. '{} / {}'.format(group.site.name, group.name),
  2479. [(vlan.pk, vlan) for vlan in site_group_vlans]
  2480. ))
  2481. self.fields['untagged_vlan'].choices = [(None, '---------')] + vlan_choices
  2482. self.fields['tagged_vlans'].choices = vlan_choices
  2483. class InterfaceBulkRenameForm(BulkRenameForm):
  2484. pk = forms.ModelMultipleChoiceField(
  2485. queryset=Interface.objects.all(),
  2486. widget=forms.MultipleHiddenInput()
  2487. )
  2488. class InterfaceBulkDisconnectForm(ConfirmationForm):
  2489. pk = forms.ModelMultipleChoiceField(
  2490. queryset=Interface.objects.all(),
  2491. widget=forms.MultipleHiddenInput()
  2492. )
  2493. #
  2494. # Front pass-through ports
  2495. #
  2496. class FrontPortFilterForm(DeviceComponentFilterForm):
  2497. model = FrontPort
  2498. class FrontPortForm(BootstrapMixin, forms.ModelForm):
  2499. tags = TagField(
  2500. required=False
  2501. )
  2502. class Meta:
  2503. model = FrontPort
  2504. fields = [
  2505. 'device', 'name', 'type', 'rear_port', 'rear_port_position', 'description', 'tags',
  2506. ]
  2507. widgets = {
  2508. 'device': forms.HiddenInput(),
  2509. 'type': StaticSelect2(),
  2510. 'rear_port': StaticSelect2(),
  2511. }
  2512. def __init__(self, *args, **kwargs):
  2513. super().__init__(*args, **kwargs)
  2514. # Limit RearPort choices to the local device
  2515. if hasattr(self.instance, 'device'):
  2516. self.fields['rear_port'].queryset = self.fields['rear_port'].queryset.filter(
  2517. device=self.instance.device
  2518. )
  2519. # TODO: Merge with FrontPortTemplateCreateForm to remove duplicate logic
  2520. class FrontPortCreateForm(ComponentForm):
  2521. name_pattern = ExpandableNameField(
  2522. label='Name'
  2523. )
  2524. type = forms.ChoiceField(
  2525. choices=PortTypeChoices,
  2526. widget=StaticSelect2(),
  2527. )
  2528. rear_port_set = forms.MultipleChoiceField(
  2529. choices=[],
  2530. label='Rear ports',
  2531. help_text='Select one rear port assignment for each front port being created.',
  2532. )
  2533. description = forms.CharField(
  2534. required=False
  2535. )
  2536. def __init__(self, *args, **kwargs):
  2537. super().__init__(*args, **kwargs)
  2538. # Determine which rear port positions are occupied. These will be excluded from the list of available mappings.
  2539. occupied_port_positions = [
  2540. (front_port.rear_port_id, front_port.rear_port_position)
  2541. for front_port in self.parent.frontports.all()
  2542. ]
  2543. # Populate rear port choices
  2544. choices = []
  2545. rear_ports = RearPort.objects.filter(device=self.parent)
  2546. for rear_port in rear_ports:
  2547. for i in range(1, rear_port.positions + 1):
  2548. if (rear_port.pk, i) not in occupied_port_positions:
  2549. choices.append(
  2550. ('{}:{}'.format(rear_port.pk, i), '{}:{}'.format(rear_port.name, i))
  2551. )
  2552. self.fields['rear_port_set'].choices = choices
  2553. def clean(self):
  2554. # Validate that the number of ports being created equals the number of selected (rear port, position) tuples
  2555. front_port_count = len(self.cleaned_data['name_pattern'])
  2556. rear_port_count = len(self.cleaned_data['rear_port_set'])
  2557. if front_port_count != rear_port_count:
  2558. raise forms.ValidationError({
  2559. 'rear_port_set': 'The provided name pattern will create {} ports, however {} rear port assignments '
  2560. 'were selected. These counts must match.'.format(front_port_count, rear_port_count)
  2561. })
  2562. def get_iterative_data(self, iteration):
  2563. # Assign rear port and position from selected set
  2564. rear_port, position = self.cleaned_data['rear_port_set'][iteration].split(':')
  2565. return {
  2566. 'rear_port': int(rear_port),
  2567. 'rear_port_position': int(position),
  2568. }
  2569. class FrontPortCSVForm(forms.ModelForm):
  2570. device = FlexibleModelChoiceField(
  2571. queryset=Device.objects.all(),
  2572. to_field_name='name',
  2573. help_text='Name or ID of device',
  2574. error_messages={
  2575. 'invalid_choice': 'Device not found.',
  2576. }
  2577. )
  2578. rear_port = FlexibleModelChoiceField(
  2579. queryset=RearPort.objects.all(),
  2580. to_field_name='name',
  2581. help_text='Name or ID of Rear Port',
  2582. error_messages={
  2583. 'invalid_choice': 'Rear Port not found.',
  2584. }
  2585. )
  2586. type = CSVChoiceField(
  2587. choices=PortTypeChoices,
  2588. )
  2589. class Meta:
  2590. model = FrontPort
  2591. fields = FrontPort.csv_headers
  2592. def __init__(self, *args, **kwargs):
  2593. super().__init__(*args, **kwargs)
  2594. # Limit RearPort choices to those belonging to this device (or VC master)
  2595. if self.is_bound:
  2596. try:
  2597. device = self.fields['device'].to_python(self.data['device'])
  2598. except forms.ValidationError:
  2599. device = None
  2600. else:
  2601. try:
  2602. device = self.instance.device
  2603. except Device.DoesNotExist:
  2604. device = None
  2605. if device:
  2606. self.fields['rear_port'].queryset = RearPort.objects.filter(
  2607. device__in=[device, device.get_vc_master()]
  2608. )
  2609. else:
  2610. self.fields['rear_port'].queryset = RearPort.objects.none()
  2611. class FrontPortBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
  2612. pk = forms.ModelMultipleChoiceField(
  2613. queryset=FrontPort.objects.all(),
  2614. widget=forms.MultipleHiddenInput()
  2615. )
  2616. type = forms.ChoiceField(
  2617. choices=add_blank_choice(PortTypeChoices),
  2618. required=False,
  2619. widget=StaticSelect2()
  2620. )
  2621. description = forms.CharField(
  2622. max_length=100,
  2623. required=False
  2624. )
  2625. class Meta:
  2626. nullable_fields = [
  2627. 'description',
  2628. ]
  2629. class FrontPortBulkRenameForm(BulkRenameForm):
  2630. pk = forms.ModelMultipleChoiceField(
  2631. queryset=FrontPort.objects.all(),
  2632. widget=forms.MultipleHiddenInput
  2633. )
  2634. class FrontPortBulkDisconnectForm(ConfirmationForm):
  2635. pk = forms.ModelMultipleChoiceField(
  2636. queryset=FrontPort.objects.all(),
  2637. widget=forms.MultipleHiddenInput
  2638. )
  2639. #
  2640. # Rear pass-through ports
  2641. #
  2642. class RearPortFilterForm(DeviceComponentFilterForm):
  2643. model = RearPort
  2644. class RearPortForm(BootstrapMixin, forms.ModelForm):
  2645. tags = TagField(
  2646. required=False
  2647. )
  2648. class Meta:
  2649. model = RearPort
  2650. fields = [
  2651. 'device', 'name', 'type', 'positions', 'description', 'tags',
  2652. ]
  2653. widgets = {
  2654. 'device': forms.HiddenInput(),
  2655. 'type': StaticSelect2(),
  2656. }
  2657. class RearPortCreateForm(ComponentForm):
  2658. name_pattern = ExpandableNameField(
  2659. label='Name'
  2660. )
  2661. type = forms.ChoiceField(
  2662. choices=PortTypeChoices,
  2663. widget=StaticSelect2(),
  2664. )
  2665. positions = forms.IntegerField(
  2666. min_value=1,
  2667. max_value=64,
  2668. initial=1,
  2669. help_text='The number of front ports which may be mapped to each rear port'
  2670. )
  2671. description = forms.CharField(
  2672. required=False
  2673. )
  2674. class RearPortCSVForm(forms.ModelForm):
  2675. device = FlexibleModelChoiceField(
  2676. queryset=Device.objects.all(),
  2677. to_field_name='name',
  2678. help_text='Name or ID of device',
  2679. error_messages={
  2680. 'invalid_choice': 'Device not found.',
  2681. }
  2682. )
  2683. type = CSVChoiceField(
  2684. choices=PortTypeChoices,
  2685. )
  2686. class Meta:
  2687. model = RearPort
  2688. fields = RearPort.csv_headers
  2689. class RearPortBulkEditForm(BootstrapMixin, AddRemoveTagsForm, BulkEditForm):
  2690. pk = forms.ModelMultipleChoiceField(
  2691. queryset=RearPort.objects.all(),
  2692. widget=forms.MultipleHiddenInput()
  2693. )
  2694. type = forms.ChoiceField(
  2695. choices=add_blank_choice(PortTypeChoices),
  2696. required=False,
  2697. widget=StaticSelect2()
  2698. )
  2699. description = forms.CharField(
  2700. max_length=100,
  2701. required=False
  2702. )
  2703. class Meta:
  2704. nullable_fields = [
  2705. 'description',
  2706. ]
  2707. class RearPortBulkRenameForm(BulkRenameForm):
  2708. pk = forms.ModelMultipleChoiceField(
  2709. queryset=RearPort.objects.all(),
  2710. widget=forms.MultipleHiddenInput
  2711. )
  2712. class RearPortBulkDisconnectForm(ConfirmationForm):
  2713. pk = forms.ModelMultipleChoiceField(
  2714. queryset=RearPort.objects.all(),
  2715. widget=forms.MultipleHiddenInput
  2716. )
  2717. #
  2718. # Cables
  2719. #
  2720. class ConnectCableToDeviceForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
  2721. """
  2722. Base form for connecting a Cable to a Device component
  2723. """
  2724. termination_b_site = forms.ModelChoiceField(
  2725. queryset=Site.objects.all(),
  2726. label='Site',
  2727. required=False,
  2728. widget=APISelect(
  2729. api_url='/api/dcim/sites/',
  2730. filter_for={
  2731. 'termination_b_rack': 'site_id',
  2732. 'termination_b_device': 'site_id',
  2733. }
  2734. )
  2735. )
  2736. termination_b_rack = ChainedModelChoiceField(
  2737. queryset=Rack.objects.all(),
  2738. chains=(
  2739. ('site', 'termination_b_site'),
  2740. ),
  2741. label='Rack',
  2742. required=False,
  2743. widget=APISelect(
  2744. api_url='/api/dcim/racks/',
  2745. filter_for={
  2746. 'termination_b_device': 'rack_id',
  2747. },
  2748. attrs={
  2749. 'nullable': 'true',
  2750. }
  2751. )
  2752. )
  2753. termination_b_device = ChainedModelChoiceField(
  2754. queryset=Device.objects.all(),
  2755. chains=(
  2756. ('site', 'termination_b_site'),
  2757. ('rack', 'termination_b_rack'),
  2758. ),
  2759. label='Device',
  2760. required=False,
  2761. widget=APISelect(
  2762. api_url='/api/dcim/devices/',
  2763. display_field='display_name',
  2764. filter_for={
  2765. 'termination_b_id': 'device_id',
  2766. }
  2767. )
  2768. )
  2769. class Meta:
  2770. model = Cable
  2771. fields = [
  2772. 'termination_b_site', 'termination_b_rack', 'termination_b_device', 'termination_b_id', 'type', 'status',
  2773. 'label', 'color', 'length', 'length_unit',
  2774. ]
  2775. class ConnectCableToConsolePortForm(ConnectCableToDeviceForm):
  2776. termination_b_id = forms.IntegerField(
  2777. label='Name',
  2778. widget=APISelect(
  2779. api_url='/api/dcim/console-ports/',
  2780. disabled_indicator='cable',
  2781. )
  2782. )
  2783. class ConnectCableToConsoleServerPortForm(ConnectCableToDeviceForm):
  2784. termination_b_id = forms.IntegerField(
  2785. label='Name',
  2786. widget=APISelect(
  2787. api_url='/api/dcim/console-server-ports/',
  2788. disabled_indicator='cable',
  2789. )
  2790. )
  2791. class ConnectCableToPowerPortForm(ConnectCableToDeviceForm):
  2792. termination_b_id = forms.IntegerField(
  2793. label='Name',
  2794. widget=APISelect(
  2795. api_url='/api/dcim/power-ports/',
  2796. disabled_indicator='cable',
  2797. )
  2798. )
  2799. class ConnectCableToPowerOutletForm(ConnectCableToDeviceForm):
  2800. termination_b_id = forms.IntegerField(
  2801. label='Name',
  2802. widget=APISelect(
  2803. api_url='/api/dcim/power-outlets/',
  2804. disabled_indicator='cable',
  2805. )
  2806. )
  2807. class ConnectCableToInterfaceForm(ConnectCableToDeviceForm):
  2808. termination_b_id = forms.IntegerField(
  2809. label='Name',
  2810. widget=APISelect(
  2811. api_url='/api/dcim/interfaces/',
  2812. disabled_indicator='cable',
  2813. additional_query_params={
  2814. 'kind': 'physical',
  2815. }
  2816. )
  2817. )
  2818. class ConnectCableToFrontPortForm(ConnectCableToDeviceForm):
  2819. termination_b_id = forms.IntegerField(
  2820. label='Name',
  2821. widget=APISelect(
  2822. api_url='/api/dcim/front-ports/',
  2823. disabled_indicator='cable',
  2824. )
  2825. )
  2826. class ConnectCableToRearPortForm(ConnectCableToDeviceForm):
  2827. termination_b_id = forms.IntegerField(
  2828. label='Name',
  2829. widget=APISelect(
  2830. api_url='/api/dcim/rear-ports/',
  2831. disabled_indicator='cable',
  2832. )
  2833. )
  2834. class ConnectCableToCircuitTerminationForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
  2835. termination_b_provider = forms.ModelChoiceField(
  2836. queryset=Provider.objects.all(),
  2837. label='Provider',
  2838. widget=APISelect(
  2839. api_url='/api/circuits/providers/',
  2840. filter_for={
  2841. 'termination_b_circuit': 'provider_id',
  2842. }
  2843. )
  2844. )
  2845. termination_b_site = forms.ModelChoiceField(
  2846. queryset=Site.objects.all(),
  2847. label='Site',
  2848. required=False,
  2849. widget=APISelect(
  2850. api_url='/api/dcim/sites/',
  2851. filter_for={
  2852. 'termination_b_circuit': 'site_id',
  2853. }
  2854. )
  2855. )
  2856. termination_b_circuit = ChainedModelChoiceField(
  2857. queryset=Circuit.objects.all(),
  2858. chains=(
  2859. ('provider', 'termination_b_provider'),
  2860. ),
  2861. label='Circuit',
  2862. widget=APISelect(
  2863. api_url='/api/circuits/circuits/',
  2864. display_field='cid',
  2865. filter_for={
  2866. 'termination_b_id': 'circuit_id',
  2867. }
  2868. )
  2869. )
  2870. termination_b_id = forms.IntegerField(
  2871. label='Side',
  2872. widget=APISelect(
  2873. api_url='/api/circuits/circuit-terminations/',
  2874. disabled_indicator='cable',
  2875. display_field='term_side'
  2876. )
  2877. )
  2878. class Meta:
  2879. model = Cable
  2880. fields = [
  2881. 'termination_b_provider', 'termination_b_site', 'termination_b_circuit', 'termination_b_id', 'type',
  2882. 'status', 'label', 'color', 'length', 'length_unit',
  2883. ]
  2884. class ConnectCableToPowerFeedForm(BootstrapMixin, ChainedFieldsMixin, forms.ModelForm):
  2885. termination_b_site = forms.ModelChoiceField(
  2886. queryset=Site.objects.all(),
  2887. label='Site',
  2888. widget=APISelect(
  2889. api_url='/api/dcim/sites/',
  2890. display_field='cid',
  2891. filter_for={
  2892. 'termination_b_rackgroup': 'site_id',
  2893. 'termination_b_powerpanel': 'site_id',
  2894. }
  2895. )
  2896. )
  2897. termination_b_rackgroup = ChainedModelChoiceField(
  2898. queryset=RackGroup.objects.all(),
  2899. label='Rack Group',
  2900. chains=(
  2901. ('site', 'termination_b_site'),
  2902. ),
  2903. required=False,
  2904. widget=APISelect(
  2905. api_url='/api/dcim/rack-groups/',
  2906. display_field='cid',
  2907. filter_for={
  2908. 'termination_b_powerpanel': 'rackgroup_id',
  2909. }
  2910. )
  2911. )
  2912. termination_b_powerpanel = ChainedModelChoiceField(
  2913. queryset=PowerPanel.objects.all(),
  2914. chains=(
  2915. ('site', 'termination_b_site'),
  2916. ('rack_group', 'termination_b_rackgroup'),
  2917. ),
  2918. label='Power Panel',
  2919. widget=APISelect(
  2920. api_url='/api/dcim/power-panels/',
  2921. filter_for={
  2922. 'termination_b_id': 'power_panel_id',
  2923. }
  2924. )
  2925. )
  2926. termination_b_id = forms.IntegerField(
  2927. label='Name',
  2928. widget=APISelect(
  2929. api_url='/api/dcim/power-feeds/',
  2930. )
  2931. )
  2932. class Meta:
  2933. model = Cable
  2934. fields = [
  2935. 'termination_b_rackgroup', 'termination_b_powerpanel', 'termination_b_id', 'type', 'status', 'label',
  2936. 'color', 'length', 'length_unit',
  2937. ]
  2938. class CableForm(BootstrapMixin, forms.ModelForm):
  2939. class Meta:
  2940. model = Cable
  2941. fields = [
  2942. 'type', 'status', 'label', 'color', 'length', 'length_unit',
  2943. ]
  2944. class CableCSVForm(forms.ModelForm):
  2945. # Termination A
  2946. side_a_device = FlexibleModelChoiceField(
  2947. queryset=Device.objects.all(),
  2948. to_field_name='name',
  2949. help_text='Side A device name or ID',
  2950. error_messages={
  2951. 'invalid_choice': 'Side A device not found',
  2952. }
  2953. )
  2954. side_a_type = forms.ModelChoiceField(
  2955. queryset=ContentType.objects.all(),
  2956. limit_choices_to={
  2957. 'model__in': CABLE_TERMINATION_TYPES,
  2958. },
  2959. to_field_name='model',
  2960. help_text='Side A type'
  2961. )
  2962. side_a_name = forms.CharField(
  2963. help_text='Side A component'
  2964. )
  2965. # Termination B
  2966. side_b_device = FlexibleModelChoiceField(
  2967. queryset=Device.objects.all(),
  2968. to_field_name='name',
  2969. help_text='Side B device name or ID',
  2970. error_messages={
  2971. 'invalid_choice': 'Side B device not found',
  2972. }
  2973. )
  2974. side_b_type = forms.ModelChoiceField(
  2975. queryset=ContentType.objects.all(),
  2976. limit_choices_to={
  2977. 'model__in': CABLE_TERMINATION_TYPES,
  2978. },
  2979. to_field_name='model',
  2980. help_text='Side B type'
  2981. )
  2982. side_b_name = forms.CharField(
  2983. help_text='Side B component'
  2984. )
  2985. # Cable attributes
  2986. status = CSVChoiceField(
  2987. choices=CableStatusChoices,
  2988. required=False,
  2989. help_text='Connection status'
  2990. )
  2991. type = CSVChoiceField(
  2992. choices=CableTypeChoices,
  2993. required=False,
  2994. help_text='Cable type'
  2995. )
  2996. length_unit = CSVChoiceField(
  2997. choices=CableLengthUnitChoices,
  2998. required=False,
  2999. help_text='Length unit'
  3000. )
  3001. class Meta:
  3002. model = Cable
  3003. fields = [
  3004. 'side_a_device', 'side_a_type', 'side_a_name', 'side_b_device', 'side_b_type', 'side_b_name', 'type',
  3005. 'status', 'label', 'color', 'length', 'length_unit',
  3006. ]
  3007. help_texts = {
  3008. 'color': 'RGB color in hexadecimal (e.g. 00ff00)'
  3009. }
  3010. # TODO: Merge the clean() methods for either end
  3011. def clean_side_a_name(self):
  3012. device = self.cleaned_data.get('side_a_device')
  3013. content_type = self.cleaned_data.get('side_a_type')
  3014. name = self.cleaned_data.get('side_a_name')
  3015. if not device or not content_type or not name:
  3016. return None
  3017. model = content_type.model_class()
  3018. try:
  3019. termination_object = model.objects.get(
  3020. device=device,
  3021. name=name
  3022. )
  3023. if termination_object.cable is not None:
  3024. raise forms.ValidationError(
  3025. "Side A: {} {} is already connected".format(device, termination_object)
  3026. )
  3027. except ObjectDoesNotExist:
  3028. raise forms.ValidationError(
  3029. "A side termination not found: {} {}".format(device, name)
  3030. )
  3031. self.instance.termination_a = termination_object
  3032. return termination_object
  3033. def clean_side_b_name(self):
  3034. device = self.cleaned_data.get('side_b_device')
  3035. content_type = self.cleaned_data.get('side_b_type')
  3036. name = self.cleaned_data.get('side_b_name')
  3037. if not device or not content_type or not name:
  3038. return None
  3039. model = content_type.model_class()
  3040. try:
  3041. termination_object = model.objects.get(
  3042. device=device,
  3043. name=name
  3044. )
  3045. if termination_object.cable is not None:
  3046. raise forms.ValidationError(
  3047. "Side B: {} {} is already connected".format(device, termination_object)
  3048. )
  3049. except ObjectDoesNotExist:
  3050. raise forms.ValidationError(
  3051. "B side termination not found: {} {}".format(device, name)
  3052. )
  3053. self.instance.termination_b = termination_object
  3054. return termination_object
  3055. def clean_length_unit(self):
  3056. # Avoid trying to save as NULL
  3057. length_unit = self.cleaned_data.get('length_unit', None)
  3058. return length_unit if length_unit is not None else ''
  3059. class CableBulkEditForm(BootstrapMixin, BulkEditForm):
  3060. pk = forms.ModelMultipleChoiceField(
  3061. queryset=Cable.objects.all(),
  3062. widget=forms.MultipleHiddenInput
  3063. )
  3064. type = forms.ChoiceField(
  3065. choices=add_blank_choice(CableTypeChoices),
  3066. required=False,
  3067. initial='',
  3068. widget=StaticSelect2()
  3069. )
  3070. status = forms.ChoiceField(
  3071. choices=add_blank_choice(CableStatusChoices),
  3072. required=False,
  3073. widget=StaticSelect2(),
  3074. initial=''
  3075. )
  3076. label = forms.CharField(
  3077. max_length=100,
  3078. required=False
  3079. )
  3080. color = forms.CharField(
  3081. max_length=6,
  3082. required=False,
  3083. widget=ColorSelect()
  3084. )
  3085. length = forms.IntegerField(
  3086. min_value=1,
  3087. required=False
  3088. )
  3089. length_unit = forms.ChoiceField(
  3090. choices=add_blank_choice(CableLengthUnitChoices),
  3091. required=False,
  3092. initial='',
  3093. widget=StaticSelect2()
  3094. )
  3095. class Meta:
  3096. nullable_fields = [
  3097. 'type', 'status', 'label', 'color', 'length',
  3098. ]
  3099. def clean(self):
  3100. # Validate length/unit
  3101. length = self.cleaned_data.get('length')
  3102. length_unit = self.cleaned_data.get('length_unit')
  3103. if length and not length_unit:
  3104. raise forms.ValidationError({
  3105. 'length_unit': "Must specify a unit when setting length"
  3106. })
  3107. class CableFilterForm(BootstrapMixin, forms.Form):
  3108. model = Cable
  3109. q = forms.CharField(
  3110. required=False,
  3111. label='Search'
  3112. )
  3113. site = FilterChoiceField(
  3114. queryset=Site.objects.all(),
  3115. to_field_name='slug',
  3116. widget=APISelectMultiple(
  3117. api_url="/api/dcim/sites/",
  3118. value_field="slug",
  3119. filter_for={
  3120. 'rack_id': 'site',
  3121. }
  3122. )
  3123. )
  3124. rack_id = FilterChoiceField(
  3125. queryset=Rack.objects.all(),
  3126. label='Rack',
  3127. null_label='-- None --',
  3128. widget=APISelectMultiple(
  3129. api_url="/api/dcim/racks/",
  3130. null_option=True,
  3131. )
  3132. )
  3133. type = forms.MultipleChoiceField(
  3134. choices=add_blank_choice(CableTypeChoices),
  3135. required=False,
  3136. widget=StaticSelect2()
  3137. )
  3138. status = forms.ChoiceField(
  3139. required=False,
  3140. choices=add_blank_choice(CableStatusChoices),
  3141. widget=StaticSelect2()
  3142. )
  3143. color = forms.CharField(
  3144. max_length=6,
  3145. required=False,
  3146. widget=ColorSelect()
  3147. )
  3148. device_id = FilterChoiceField(
  3149. queryset=Device.objects.all(),
  3150. required=False,
  3151. label='Device',
  3152. widget=APISelectMultiple(
  3153. api_url='/api/dcim/devices/',
  3154. )
  3155. )
  3156. #
  3157. # Device bays
  3158. #
  3159. class DeviceBayFilterForm(DeviceComponentFilterForm):
  3160. model = DeviceBay
  3161. class DeviceBayForm(BootstrapMixin, forms.ModelForm):
  3162. tags = TagField(
  3163. required=False
  3164. )
  3165. class Meta:
  3166. model = DeviceBay
  3167. fields = [
  3168. 'device', 'name', 'description', 'tags',
  3169. ]
  3170. widgets = {
  3171. 'device': forms.HiddenInput(),
  3172. }
  3173. class DeviceBayCreateForm(ComponentForm):
  3174. name_pattern = ExpandableNameField(
  3175. label='Name'
  3176. )
  3177. tags = TagField(
  3178. required=False
  3179. )
  3180. class PopulateDeviceBayForm(BootstrapMixin, forms.Form):
  3181. installed_device = forms.ModelChoiceField(
  3182. queryset=Device.objects.all(),
  3183. label='Child Device',
  3184. help_text="Child devices must first be created and assigned to the site/rack of the parent device.",
  3185. widget=StaticSelect2(),
  3186. )
  3187. def __init__(self, device_bay, *args, **kwargs):
  3188. super().__init__(*args, **kwargs)
  3189. self.fields['installed_device'].queryset = Device.objects.filter(
  3190. site=device_bay.device.site,
  3191. rack=device_bay.device.rack,
  3192. parent_bay__isnull=True,
  3193. device_type__u_height=0,
  3194. device_type__subdevice_role=SubdeviceRoleChoices.ROLE_CHILD
  3195. ).exclude(pk=device_bay.device.pk)
  3196. class DeviceBayCSVForm(forms.ModelForm):
  3197. device = FlexibleModelChoiceField(
  3198. queryset=Device.objects.all(),
  3199. to_field_name='name',
  3200. help_text='Name or ID of device',
  3201. error_messages={
  3202. 'invalid_choice': 'Device not found.',
  3203. }
  3204. )
  3205. installed_device = FlexibleModelChoiceField(
  3206. queryset=Device.objects.all(),
  3207. required=False,
  3208. to_field_name='name',
  3209. help_text='Name or ID of device',
  3210. error_messages={
  3211. 'invalid_choice': 'Child device not found.',
  3212. }
  3213. )
  3214. class Meta:
  3215. model = DeviceBay
  3216. fields = DeviceBay.csv_headers
  3217. def __init__(self, *args, **kwargs):
  3218. super().__init__(*args, **kwargs)
  3219. # Limit installed device choices to devices of the correct type and location
  3220. if self.is_bound:
  3221. try:
  3222. device = self.fields['device'].to_python(self.data['device'])
  3223. except forms.ValidationError:
  3224. device = None
  3225. else:
  3226. try:
  3227. device = self.instance.device
  3228. except Device.DoesNotExist:
  3229. device = None
  3230. if device:
  3231. self.fields['installed_device'].queryset = Device.objects.filter(
  3232. site=device.site,
  3233. rack=device.rack,
  3234. parent_bay__isnull=True,
  3235. device_type__u_height=0,
  3236. device_type__subdevice_role=SubdeviceRoleChoices.ROLE_CHILD
  3237. ).exclude(pk=device.pk)
  3238. else:
  3239. self.fields['installed_device'].queryset = Interface.objects.none()
  3240. class DeviceBayBulkRenameForm(BulkRenameForm):
  3241. pk = forms.ModelMultipleChoiceField(
  3242. queryset=DeviceBay.objects.all(),
  3243. widget=forms.MultipleHiddenInput()
  3244. )
  3245. #
  3246. # Connections
  3247. #
  3248. class ConsoleConnectionFilterForm(BootstrapMixin, forms.Form):
  3249. site = FilterChoiceField(
  3250. queryset=Site.objects.all(),
  3251. to_field_name='slug',
  3252. widget=APISelectMultiple(
  3253. api_url="/api/dcim/sites/",
  3254. value_field="slug",
  3255. )
  3256. )
  3257. device_id = FilterChoiceField(
  3258. queryset=Device.objects.all(),
  3259. required=False,
  3260. label='Device',
  3261. widget=APISelectMultiple(
  3262. api_url='/api/dcim/devices/',
  3263. )
  3264. )
  3265. class PowerConnectionFilterForm(BootstrapMixin, forms.Form):
  3266. site = FilterChoiceField(
  3267. queryset=Site.objects.all(),
  3268. to_field_name='slug',
  3269. widget=APISelectMultiple(
  3270. api_url="/api/dcim/sites/",
  3271. value_field="slug",
  3272. )
  3273. )
  3274. device_id = FilterChoiceField(
  3275. queryset=Device.objects.all(),
  3276. required=False,
  3277. label='Device',
  3278. widget=APISelectMultiple(
  3279. api_url='/api/dcim/devices/',
  3280. )
  3281. )
  3282. class InterfaceConnectionFilterForm(BootstrapMixin, forms.Form):
  3283. site = FilterChoiceField(
  3284. queryset=Site.objects.all(),
  3285. to_field_name='slug',
  3286. widget=APISelectMultiple(
  3287. api_url="/api/dcim/sites/",
  3288. value_field="slug",
  3289. )
  3290. )
  3291. device_id = FilterChoiceField(
  3292. queryset=Device.objects.all(),
  3293. required=False,
  3294. label='Device',
  3295. widget=APISelectMultiple(
  3296. api_url='/api/dcim/devices/',
  3297. )
  3298. )
  3299. #
  3300. # Inventory items
  3301. #
  3302. class InventoryItemForm(BootstrapMixin, forms.ModelForm):
  3303. tags = TagField(
  3304. required=False
  3305. )
  3306. class Meta:
  3307. model = InventoryItem
  3308. fields = [
  3309. 'name', 'device', 'manufacturer', 'part_id', 'serial', 'asset_tag', 'description', 'tags',
  3310. ]
  3311. widgets = {
  3312. 'device': APISelect(
  3313. api_url="/api/dcim/devices/"
  3314. ),
  3315. 'manufacturer': APISelect(
  3316. api_url="/api/dcim/manufacturers/"
  3317. )
  3318. }
  3319. class InventoryItemCSVForm(forms.ModelForm):
  3320. device = FlexibleModelChoiceField(
  3321. queryset=Device.objects.all(),
  3322. to_field_name='name',
  3323. help_text='Device name or ID',
  3324. error_messages={
  3325. 'invalid_choice': 'Device not found.',
  3326. }
  3327. )
  3328. manufacturer = forms.ModelChoiceField(
  3329. queryset=Manufacturer.objects.all(),
  3330. to_field_name='name',
  3331. required=False,
  3332. help_text='Manufacturer name',
  3333. error_messages={
  3334. 'invalid_choice': 'Invalid manufacturer.',
  3335. }
  3336. )
  3337. class Meta:
  3338. model = InventoryItem
  3339. fields = InventoryItem.csv_headers
  3340. class InventoryItemBulkEditForm(BootstrapMixin, BulkEditForm):
  3341. pk = forms.ModelMultipleChoiceField(
  3342. queryset=InventoryItem.objects.all(),
  3343. widget=forms.MultipleHiddenInput()
  3344. )
  3345. device = forms.ModelChoiceField(
  3346. queryset=Device.objects.all(),
  3347. required=False,
  3348. widget=APISelect(
  3349. api_url="/api/dcim/devices/"
  3350. )
  3351. )
  3352. manufacturer = forms.ModelChoiceField(
  3353. queryset=Manufacturer.objects.all(),
  3354. required=False,
  3355. widget=APISelect(
  3356. api_url="/api/dcim/manufacturers/"
  3357. )
  3358. )
  3359. part_id = forms.CharField(
  3360. max_length=50,
  3361. required=False,
  3362. label='Part ID'
  3363. )
  3364. description = forms.CharField(
  3365. max_length=100,
  3366. required=False
  3367. )
  3368. class Meta:
  3369. nullable_fields = [
  3370. 'manufacturer', 'part_id', 'description',
  3371. ]
  3372. class InventoryItemFilterForm(BootstrapMixin, forms.Form):
  3373. model = InventoryItem
  3374. q = forms.CharField(
  3375. required=False,
  3376. label='Search'
  3377. )
  3378. region = FilterChoiceField(
  3379. queryset=Region.objects.all(),
  3380. to_field_name='slug',
  3381. required=False,
  3382. widget=APISelectMultiple(
  3383. api_url="/api/dcim/regions/",
  3384. value_field="slug",
  3385. filter_for={
  3386. 'site': 'region'
  3387. }
  3388. )
  3389. )
  3390. site = FilterChoiceField(
  3391. queryset=Site.objects.all(),
  3392. to_field_name='slug',
  3393. widget=APISelectMultiple(
  3394. api_url="/api/dcim/sites/",
  3395. value_field="slug",
  3396. filter_for={
  3397. 'device_id': 'site'
  3398. }
  3399. )
  3400. )
  3401. device_id = FilterChoiceField(
  3402. queryset=Device.objects.all(),
  3403. required=False,
  3404. label='Device',
  3405. widget=APISelect(
  3406. api_url='/api/dcim/devices/',
  3407. )
  3408. )
  3409. manufacturer = FilterChoiceField(
  3410. queryset=Manufacturer.objects.all(),
  3411. to_field_name='slug',
  3412. widget=APISelect(
  3413. api_url="/api/dcim/manufacturers/",
  3414. value_field="slug",
  3415. )
  3416. )
  3417. discovered = forms.NullBooleanField(
  3418. required=False,
  3419. widget=StaticSelect2(
  3420. choices=BOOLEAN_WITH_BLANK_CHOICES
  3421. )
  3422. )
  3423. #
  3424. # Virtual chassis
  3425. #
  3426. class DeviceSelectionForm(forms.Form):
  3427. pk = forms.ModelMultipleChoiceField(
  3428. queryset=Device.objects.all(),
  3429. widget=forms.MultipleHiddenInput()
  3430. )
  3431. class VirtualChassisForm(BootstrapMixin, forms.ModelForm):
  3432. tags = TagField(
  3433. required=False
  3434. )
  3435. class Meta:
  3436. model = VirtualChassis
  3437. fields = [
  3438. 'master', 'domain', 'tags',
  3439. ]
  3440. widgets = {
  3441. 'master': SelectWithPK(),
  3442. }
  3443. class BaseVCMemberFormSet(forms.BaseModelFormSet):
  3444. def clean(self):
  3445. super().clean()
  3446. # Check for duplicate VC position values
  3447. vc_position_list = []
  3448. for form in self.forms:
  3449. vc_position = form.cleaned_data.get('vc_position')
  3450. if vc_position:
  3451. if vc_position in vc_position_list:
  3452. error_msg = 'A virtual chassis member already exists in position {}.'.format(vc_position)
  3453. form.add_error('vc_position', error_msg)
  3454. vc_position_list.append(vc_position)
  3455. class DeviceVCMembershipForm(forms.ModelForm):
  3456. class Meta:
  3457. model = Device
  3458. fields = [
  3459. 'vc_position', 'vc_priority',
  3460. ]
  3461. labels = {
  3462. 'vc_position': 'Position',
  3463. 'vc_priority': 'Priority',
  3464. }
  3465. def __init__(self, validate_vc_position=False, *args, **kwargs):
  3466. super().__init__(*args, **kwargs)
  3467. # Require VC position (only required when the Device is a VirtualChassis member)
  3468. self.fields['vc_position'].required = True
  3469. # Validation of vc_position is optional. This is only required when adding a new member to an existing
  3470. # VirtualChassis. Otherwise, vc_position validation is handled by BaseVCMemberFormSet.
  3471. self.validate_vc_position = validate_vc_position
  3472. def clean_vc_position(self):
  3473. vc_position = self.cleaned_data['vc_position']
  3474. if self.validate_vc_position:
  3475. conflicting_members = Device.objects.filter(
  3476. virtual_chassis=self.instance.virtual_chassis,
  3477. vc_position=vc_position
  3478. )
  3479. if conflicting_members.exists():
  3480. raise forms.ValidationError(
  3481. 'A virtual chassis member already exists in position {}.'.format(vc_position)
  3482. )
  3483. return vc_position
  3484. class VCMemberSelectForm(BootstrapMixin, ChainedFieldsMixin, forms.Form):
  3485. site = forms.ModelChoiceField(
  3486. queryset=Site.objects.all(),
  3487. label='Site',
  3488. required=False,
  3489. widget=APISelect(
  3490. api_url="/api/dcim/sites/",
  3491. filter_for={
  3492. 'rack': 'site_id',
  3493. 'device': 'site_id',
  3494. }
  3495. )
  3496. )
  3497. rack = ChainedModelChoiceField(
  3498. queryset=Rack.objects.all(),
  3499. chains=(
  3500. ('site', 'site'),
  3501. ),
  3502. label='Rack',
  3503. required=False,
  3504. widget=APISelect(
  3505. api_url='/api/dcim/racks/',
  3506. filter_for={
  3507. 'device': 'rack_id'
  3508. },
  3509. attrs={
  3510. 'nullable': 'true',
  3511. }
  3512. )
  3513. )
  3514. device = ChainedModelChoiceField(
  3515. queryset=Device.objects.filter(
  3516. virtual_chassis__isnull=True
  3517. ),
  3518. chains=(
  3519. ('site', 'site'),
  3520. ('rack', 'rack'),
  3521. ),
  3522. label='Device',
  3523. widget=APISelect(
  3524. api_url='/api/dcim/devices/',
  3525. display_field='display_name',
  3526. disabled_indicator='virtual_chassis'
  3527. )
  3528. )
  3529. def clean_device(self):
  3530. device = self.cleaned_data['device']
  3531. if device.virtual_chassis is not None:
  3532. raise forms.ValidationError(
  3533. "Device {} is already assigned to a virtual chassis.".format(device)
  3534. )
  3535. return device
  3536. class VirtualChassisFilterForm(BootstrapMixin, CustomFieldFilterForm):
  3537. model = VirtualChassis
  3538. q = forms.CharField(
  3539. required=False,
  3540. label='Search'
  3541. )
  3542. region = FilterChoiceField(
  3543. queryset=Region.objects.all(),
  3544. to_field_name='slug',
  3545. required=False,
  3546. widget=APISelectMultiple(
  3547. api_url="/api/dcim/regions/",
  3548. value_field="slug",
  3549. filter_for={
  3550. 'site': 'region'
  3551. }
  3552. )
  3553. )
  3554. site = FilterChoiceField(
  3555. queryset=Site.objects.all(),
  3556. to_field_name='slug',
  3557. widget=APISelectMultiple(
  3558. api_url="/api/dcim/sites/",
  3559. value_field="slug",
  3560. )
  3561. )
  3562. tenant_group = FilterChoiceField(
  3563. queryset=TenantGroup.objects.all(),
  3564. to_field_name='slug',
  3565. null_label='-- None --',
  3566. widget=APISelectMultiple(
  3567. api_url="/api/tenancy/tenant-groups/",
  3568. value_field="slug",
  3569. null_option=True,
  3570. filter_for={
  3571. 'tenant': 'group'
  3572. }
  3573. )
  3574. )
  3575. tenant = FilterChoiceField(
  3576. queryset=Tenant.objects.all(),
  3577. to_field_name='slug',
  3578. null_label='-- None --',
  3579. widget=APISelectMultiple(
  3580. api_url="/api/tenancy/tenants/",
  3581. value_field="slug",
  3582. null_option=True,
  3583. )
  3584. )
  3585. #
  3586. # Power panels
  3587. #
  3588. class PowerPanelForm(BootstrapMixin, forms.ModelForm):
  3589. rack_group = ChainedModelChoiceField(
  3590. queryset=RackGroup.objects.all(),
  3591. chains=(
  3592. ('site', 'site'),
  3593. ),
  3594. required=False,
  3595. widget=APISelect(
  3596. api_url='/api/dcim/rack-groups/',
  3597. )
  3598. )
  3599. class Meta:
  3600. model = PowerPanel
  3601. fields = [
  3602. 'site', 'rack_group', 'name',
  3603. ]
  3604. widgets = {
  3605. 'site': APISelect(
  3606. api_url="/api/dcim/sites/",
  3607. filter_for={
  3608. 'rack_group': 'site_id',
  3609. }
  3610. ),
  3611. }
  3612. class PowerPanelCSVForm(forms.ModelForm):
  3613. site = forms.ModelChoiceField(
  3614. queryset=Site.objects.all(),
  3615. to_field_name='name',
  3616. help_text='Name of parent site',
  3617. error_messages={
  3618. 'invalid_choice': 'Site not found.',
  3619. }
  3620. )
  3621. rack_group_name = forms.CharField(
  3622. required=False,
  3623. help_text="Rack group name (optional)"
  3624. )
  3625. class Meta:
  3626. model = PowerPanel
  3627. fields = PowerPanel.csv_headers
  3628. def clean(self):
  3629. super().clean()
  3630. site = self.cleaned_data.get('site')
  3631. rack_group_name = self.cleaned_data.get('rack_group_name')
  3632. # Validate rack group
  3633. if rack_group_name:
  3634. try:
  3635. self.instance.rack_group = RackGroup.objects.get(site=site, name=rack_group_name)
  3636. except RackGroup.DoesNotExist:
  3637. raise forms.ValidationError(
  3638. "Rack group {} not found in site {}".format(rack_group_name, site)
  3639. )
  3640. class PowerPanelFilterForm(BootstrapMixin, CustomFieldFilterForm):
  3641. model = PowerPanel
  3642. q = forms.CharField(
  3643. required=False,
  3644. label='Search'
  3645. )
  3646. region = FilterChoiceField(
  3647. queryset=Region.objects.all(),
  3648. to_field_name='slug',
  3649. required=False,
  3650. widget=APISelectMultiple(
  3651. api_url="/api/dcim/regions/",
  3652. value_field="slug",
  3653. filter_for={
  3654. 'site': 'region'
  3655. }
  3656. )
  3657. )
  3658. site = FilterChoiceField(
  3659. queryset=Site.objects.all(),
  3660. to_field_name='slug',
  3661. widget=APISelectMultiple(
  3662. api_url="/api/dcim/sites/",
  3663. value_field="slug",
  3664. filter_for={
  3665. 'rack_group_id': 'site',
  3666. }
  3667. )
  3668. )
  3669. rack_group_id = FilterChoiceField(
  3670. queryset=RackGroup.objects.all(),
  3671. label='Rack group (ID)',
  3672. null_label='-- None --',
  3673. widget=APISelectMultiple(
  3674. api_url="/api/dcim/rack-groups/",
  3675. null_option=True,
  3676. )
  3677. )
  3678. #
  3679. # Power feeds
  3680. #
  3681. class PowerFeedForm(BootstrapMixin, CustomFieldForm):
  3682. site = ChainedModelChoiceField(
  3683. queryset=Site.objects.all(),
  3684. required=False,
  3685. widget=APISelect(
  3686. api_url='/api/dcim/sites/',
  3687. filter_for={
  3688. 'power_panel': 'site_id',
  3689. 'rack': 'site_id',
  3690. }
  3691. )
  3692. )
  3693. comments = CommentField()
  3694. tags = TagField(
  3695. required=False
  3696. )
  3697. class Meta:
  3698. model = PowerFeed
  3699. fields = [
  3700. 'site', 'power_panel', 'rack', 'name', 'status', 'type', 'supply', 'phase', 'voltage', 'amperage',
  3701. 'max_utilization', 'comments', 'tags',
  3702. ]
  3703. widgets = {
  3704. 'power_panel': APISelect(
  3705. api_url="/api/dcim/power-panels/"
  3706. ),
  3707. 'rack': APISelect(
  3708. api_url="/api/dcim/racks/"
  3709. ),
  3710. 'status': StaticSelect2(),
  3711. 'type': StaticSelect2(),
  3712. 'supply': StaticSelect2(),
  3713. 'phase': StaticSelect2(),
  3714. }
  3715. def __init__(self, *args, **kwargs):
  3716. super().__init__(*args, **kwargs)
  3717. # Initialize site field
  3718. if self.instance and hasattr(self.instance, 'power_panel'):
  3719. self.initial['site'] = self.instance.power_panel.site
  3720. class PowerFeedCSVForm(forms.ModelForm):
  3721. site = forms.ModelChoiceField(
  3722. queryset=Site.objects.all(),
  3723. to_field_name='name',
  3724. help_text='Name of parent site',
  3725. error_messages={
  3726. 'invalid_choice': 'Site not found.',
  3727. }
  3728. )
  3729. panel_name = forms.ModelChoiceField(
  3730. queryset=PowerPanel.objects.all(),
  3731. to_field_name='name',
  3732. help_text='Name of upstream power panel',
  3733. error_messages={
  3734. 'invalid_choice': 'Power panel not found.',
  3735. }
  3736. )
  3737. rack_group = forms.CharField(
  3738. required=False,
  3739. help_text="Rack group name (optional)"
  3740. )
  3741. rack_name = forms.CharField(
  3742. required=False,
  3743. help_text="Rack name (optional)"
  3744. )
  3745. status = CSVChoiceField(
  3746. choices=PowerFeedStatusChoices,
  3747. required=False,
  3748. help_text='Operational status'
  3749. )
  3750. type = CSVChoiceField(
  3751. choices=PowerFeedTypeChoices,
  3752. required=False,
  3753. help_text='Primary or redundant'
  3754. )
  3755. supply = CSVChoiceField(
  3756. choices=PowerFeedSupplyChoices,
  3757. required=False,
  3758. help_text='AC/DC'
  3759. )
  3760. phase = CSVChoiceField(
  3761. choices=PowerFeedPhaseChoices,
  3762. required=False,
  3763. help_text='Single or three-phase'
  3764. )
  3765. class Meta:
  3766. model = PowerFeed
  3767. fields = PowerFeed.csv_headers
  3768. def clean(self):
  3769. super().clean()
  3770. site = self.cleaned_data.get('site')
  3771. panel_name = self.cleaned_data.get('panel_name')
  3772. rack_group = self.cleaned_data.get('rack_group')
  3773. rack_name = self.cleaned_data.get('rack_name')
  3774. # Validate power panel
  3775. if panel_name:
  3776. try:
  3777. self.instance.power_panel = PowerPanel.objects.get(site=site, name=panel_name)
  3778. except Rack.DoesNotExist:
  3779. raise forms.ValidationError(
  3780. "Power panel {} not found in site {}".format(panel_name, site)
  3781. )
  3782. # Validate rack
  3783. if rack_name:
  3784. try:
  3785. self.instance.rack = Rack.objects.get(site=site, group__name=rack_group, name=rack_name)
  3786. except Rack.DoesNotExist:
  3787. raise forms.ValidationError(
  3788. "Rack {} not found in site {}, group {}".format(rack_name, site, rack_group)
  3789. )
  3790. class PowerFeedBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEditForm):
  3791. pk = forms.ModelMultipleChoiceField(
  3792. queryset=PowerFeed.objects.all(),
  3793. widget=forms.MultipleHiddenInput
  3794. )
  3795. powerpanel = forms.ModelChoiceField(
  3796. queryset=PowerPanel.objects.all(),
  3797. required=False,
  3798. widget=APISelect(
  3799. api_url="/api/dcim/power-panels/",
  3800. filter_for={
  3801. 'rackgroup': 'site_id',
  3802. }
  3803. )
  3804. )
  3805. rack = forms.ModelChoiceField(
  3806. queryset=Rack.objects.all(),
  3807. required=False,
  3808. widget=APISelect(
  3809. api_url="/api/dcim/racks",
  3810. )
  3811. )
  3812. status = forms.ChoiceField(
  3813. choices=add_blank_choice(PowerFeedStatusChoices),
  3814. required=False,
  3815. initial='',
  3816. widget=StaticSelect2()
  3817. )
  3818. type = forms.ChoiceField(
  3819. choices=add_blank_choice(PowerFeedTypeChoices),
  3820. required=False,
  3821. initial='',
  3822. widget=StaticSelect2()
  3823. )
  3824. supply = forms.ChoiceField(
  3825. choices=add_blank_choice(PowerFeedSupplyChoices),
  3826. required=False,
  3827. initial='',
  3828. widget=StaticSelect2()
  3829. )
  3830. phase = forms.ChoiceField(
  3831. choices=add_blank_choice(PowerFeedPhaseChoices),
  3832. required=False,
  3833. initial='',
  3834. widget=StaticSelect2()
  3835. )
  3836. voltage = forms.IntegerField(
  3837. required=False
  3838. )
  3839. amperage = forms.IntegerField(
  3840. required=False
  3841. )
  3842. max_utilization = forms.IntegerField(
  3843. required=False
  3844. )
  3845. comments = forms.CharField(
  3846. required=False
  3847. )
  3848. class Meta:
  3849. nullable_fields = [
  3850. 'rackgroup', 'comments',
  3851. ]
  3852. class PowerFeedFilterForm(BootstrapMixin, CustomFieldFilterForm):
  3853. model = PowerFeed
  3854. q = forms.CharField(
  3855. required=False,
  3856. label='Search'
  3857. )
  3858. region = FilterChoiceField(
  3859. queryset=Region.objects.all(),
  3860. to_field_name='slug',
  3861. required=False,
  3862. widget=APISelectMultiple(
  3863. api_url="/api/dcim/regions/",
  3864. value_field="slug",
  3865. filter_for={
  3866. 'site': 'region'
  3867. }
  3868. )
  3869. )
  3870. site = FilterChoiceField(
  3871. queryset=Site.objects.all(),
  3872. to_field_name='slug',
  3873. widget=APISelectMultiple(
  3874. api_url="/api/dcim/sites/",
  3875. value_field="slug",
  3876. filter_for={
  3877. 'power_panel_id': 'site',
  3878. 'rack_id': 'site',
  3879. }
  3880. )
  3881. )
  3882. power_panel_id = FilterChoiceField(
  3883. queryset=PowerPanel.objects.all(),
  3884. label='Power panel',
  3885. null_label='-- None --',
  3886. widget=APISelectMultiple(
  3887. api_url="/api/dcim/power-panels/",
  3888. null_option=True,
  3889. )
  3890. )
  3891. rack_id = FilterChoiceField(
  3892. queryset=Rack.objects.all(),
  3893. label='Rack',
  3894. null_label='-- None --',
  3895. widget=APISelectMultiple(
  3896. api_url="/api/dcim/racks/",
  3897. null_option=True,
  3898. )
  3899. )
  3900. status = forms.MultipleChoiceField(
  3901. choices=PowerFeedStatusChoices,
  3902. required=False,
  3903. widget=StaticSelect2Multiple()
  3904. )
  3905. type = forms.ChoiceField(
  3906. choices=add_blank_choice(PowerFeedTypeChoices),
  3907. required=False,
  3908. widget=StaticSelect2()
  3909. )
  3910. supply = forms.ChoiceField(
  3911. choices=add_blank_choice(PowerFeedSupplyChoices),
  3912. required=False,
  3913. widget=StaticSelect2()
  3914. )
  3915. phase = forms.ChoiceField(
  3916. choices=add_blank_choice(PowerFeedPhaseChoices),
  3917. required=False,
  3918. widget=StaticSelect2()
  3919. )
  3920. voltage = forms.IntegerField(
  3921. required=False
  3922. )
  3923. amperage = forms.IntegerField(
  3924. required=False
  3925. )
  3926. max_utilization = forms.IntegerField(
  3927. required=False
  3928. )