forms.py 159 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493
  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.utils.safestring import mark_safe
  8. from django.utils.translation import gettext as _
  9. from netaddr import EUI
  10. from netaddr.core import AddrFormatError
  11. from timezone_field import TimeZoneFormField
  12. from circuits.models import Circuit, CircuitTermination, Provider
  13. from extras.forms import (
  14. AddRemoveTagsForm, CustomFieldModelBulkEditForm, CustomFieldModelCSVForm, CustomFieldModelFilterForm, CustomFieldModelForm,
  15. CustomFieldsMixin, LocalConfigContextFilterForm,
  16. )
  17. from extras.models import Tag
  18. from ipam.constants import BGP_ASN_MAX, BGP_ASN_MIN
  19. from ipam.models import IPAddress, VLAN, VLANGroup
  20. from tenancy.forms import TenancyFilterForm, TenancyForm
  21. from tenancy.models import Tenant
  22. from utilities.forms import (
  23. APISelect, APISelectMultiple, add_blank_choice, BootstrapMixin, BulkEditForm, BulkEditNullBooleanSelect,
  24. ColorField, CommentField, CSVChoiceField, CSVContentTypeField, CSVModelChoiceField, CSVTypedChoiceField,
  25. DynamicModelChoiceField, DynamicModelMultipleChoiceField, ExpandableNameField, form_from_model, JSONField,
  26. NumericArrayField, SelectWithPK, SmallTextarea, SlugField, StaticSelect, StaticSelectMultiple, TagFilterField,
  27. BOOLEAN_WITH_BLANK_CHOICES,
  28. )
  29. from virtualization.models import Cluster, ClusterGroup
  30. from .choices import *
  31. from .constants import *
  32. from .models import *
  33. DEVICE_BY_PK_RE = r'{\d+\}'
  34. INTERFACE_MODE_HELP_TEXT = """
  35. Access: One untagged VLAN<br />
  36. Tagged: One untagged VLAN and/or one or more tagged VLANs<br />
  37. Tagged (All): Implies all VLANs are available (w/optional untagged VLAN)
  38. """
  39. def get_device_by_name_or_pk(name):
  40. """
  41. Attempt to retrieve a device by either its name or primary key ('{pk}').
  42. """
  43. if re.match(DEVICE_BY_PK_RE, name):
  44. pk = name.strip('{}')
  45. device = Device.objects.get(pk=pk)
  46. else:
  47. device = Device.objects.get(name=name)
  48. return device
  49. class DeviceComponentFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  50. field_order = [
  51. 'q', 'name', 'label', 'region_id', 'site_group_id', 'site_id',
  52. ]
  53. q = forms.CharField(
  54. required=False,
  55. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  56. label=_('Search')
  57. )
  58. name = forms.CharField(
  59. required=False
  60. )
  61. label = forms.CharField(
  62. required=False
  63. )
  64. region_id = DynamicModelMultipleChoiceField(
  65. queryset=Region.objects.all(),
  66. required=False,
  67. label=_('Region'),
  68. fetch_trigger='open'
  69. )
  70. site_group_id = DynamicModelMultipleChoiceField(
  71. queryset=SiteGroup.objects.all(),
  72. required=False,
  73. label=_('Site group'),
  74. fetch_trigger='open'
  75. )
  76. site_id = DynamicModelMultipleChoiceField(
  77. queryset=Site.objects.all(),
  78. required=False,
  79. query_params={
  80. 'region_id': '$region_id',
  81. 'group_id': '$site_group_id',
  82. },
  83. label=_('Site'),
  84. fetch_trigger='open'
  85. )
  86. location_id = DynamicModelMultipleChoiceField(
  87. queryset=Location.objects.all(),
  88. required=False,
  89. query_params={
  90. 'site_id': '$site_id',
  91. },
  92. label=_('Location'),
  93. fetch_trigger='open'
  94. )
  95. device_id = DynamicModelMultipleChoiceField(
  96. queryset=Device.objects.all(),
  97. required=False,
  98. query_params={
  99. 'site_id': '$site_id',
  100. 'location_id': '$location_id',
  101. },
  102. label=_('Device'),
  103. fetch_trigger='open'
  104. )
  105. class InterfaceCommonForm(forms.Form):
  106. mac_address = forms.CharField(
  107. empty_value=None,
  108. required=False,
  109. label='MAC address'
  110. )
  111. mtu = forms.IntegerField(
  112. required=False,
  113. min_value=INTERFACE_MTU_MIN,
  114. max_value=INTERFACE_MTU_MAX,
  115. label='MTU'
  116. )
  117. def clean(self):
  118. super().clean()
  119. parent_field = 'device' if 'device' in self.cleaned_data else 'virtual_machine'
  120. tagged_vlans = self.cleaned_data.get('tagged_vlans')
  121. # Untagged interfaces cannot be assigned tagged VLANs
  122. if self.cleaned_data['mode'] == InterfaceModeChoices.MODE_ACCESS and tagged_vlans:
  123. raise forms.ValidationError({
  124. 'mode': "An access interface cannot have tagged VLANs assigned."
  125. })
  126. # Remove all tagged VLAN assignments from "tagged all" interfaces
  127. elif self.cleaned_data['mode'] == InterfaceModeChoices.MODE_TAGGED_ALL:
  128. self.cleaned_data['tagged_vlans'] = []
  129. # Validate tagged VLANs; must be a global VLAN or in the same site
  130. elif self.cleaned_data['mode'] == InterfaceModeChoices.MODE_TAGGED and tagged_vlans:
  131. valid_sites = [None, self.cleaned_data[parent_field].site]
  132. invalid_vlans = [str(v) for v in tagged_vlans if v.site not in valid_sites]
  133. if invalid_vlans:
  134. raise forms.ValidationError({
  135. 'tagged_vlans': f"The tagged VLANs ({', '.join(invalid_vlans)}) must belong to the same site as "
  136. f"the interface's parent device/VM, or they must be global"
  137. })
  138. class ComponentForm(forms.Form):
  139. """
  140. Subclass this form when facilitating the creation of one or more device component or component templates based on
  141. a name pattern.
  142. """
  143. name_pattern = ExpandableNameField(
  144. label='Name'
  145. )
  146. label_pattern = ExpandableNameField(
  147. label='Label',
  148. required=False,
  149. help_text='Alphanumeric ranges are supported. (Must match the number of names being created.)'
  150. )
  151. def clean(self):
  152. super().clean()
  153. # Validate that the number of components being created from both the name_pattern and label_pattern are equal
  154. if self.cleaned_data['label_pattern']:
  155. name_pattern_count = len(self.cleaned_data['name_pattern'])
  156. label_pattern_count = len(self.cleaned_data['label_pattern'])
  157. if name_pattern_count != label_pattern_count:
  158. raise forms.ValidationError({
  159. 'label_pattern': f'The provided name pattern will create {name_pattern_count} components, however '
  160. f'{label_pattern_count} labels will be generated. These counts must match.'
  161. }, code='label_pattern_mismatch')
  162. #
  163. # Fields
  164. #
  165. class MACAddressField(forms.Field):
  166. widget = forms.CharField
  167. default_error_messages = {
  168. 'invalid': 'MAC address must be in EUI-48 format',
  169. }
  170. def to_python(self, value):
  171. value = super().to_python(value)
  172. # Validate MAC address format
  173. try:
  174. value = EUI(value.strip())
  175. except AddrFormatError:
  176. raise forms.ValidationError(self.error_messages['invalid'], code='invalid')
  177. return value
  178. #
  179. # Regions
  180. #
  181. class RegionForm(BootstrapMixin, CustomFieldModelForm):
  182. parent = DynamicModelChoiceField(
  183. queryset=Region.objects.all(),
  184. required=False
  185. )
  186. slug = SlugField()
  187. class Meta:
  188. model = Region
  189. fields = (
  190. 'parent', 'name', 'slug', 'description',
  191. )
  192. class RegionCSVForm(CustomFieldModelCSVForm):
  193. parent = CSVModelChoiceField(
  194. queryset=Region.objects.all(),
  195. required=False,
  196. to_field_name='name',
  197. help_text='Name of parent region'
  198. )
  199. class Meta:
  200. model = Region
  201. fields = ('name', 'slug', 'parent', 'description')
  202. class RegionBulkEditForm(BootstrapMixin, CustomFieldModelBulkEditForm):
  203. pk = forms.ModelMultipleChoiceField(
  204. queryset=Region.objects.all(),
  205. widget=forms.MultipleHiddenInput
  206. )
  207. parent = DynamicModelChoiceField(
  208. queryset=Region.objects.all(),
  209. required=False
  210. )
  211. description = forms.CharField(
  212. max_length=200,
  213. required=False
  214. )
  215. class Meta:
  216. nullable_fields = ['parent', 'description']
  217. class RegionFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  218. model = Region
  219. field_groups = [
  220. ['q'],
  221. ['parent_id'],
  222. ]
  223. q = forms.CharField(
  224. required=False,
  225. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  226. label=_('Search')
  227. )
  228. parent_id = DynamicModelMultipleChoiceField(
  229. queryset=Region.objects.all(),
  230. required=False,
  231. label=_('Parent region'),
  232. fetch_trigger='open'
  233. )
  234. #
  235. # Site groups
  236. #
  237. class SiteGroupForm(BootstrapMixin, CustomFieldModelForm):
  238. parent = DynamicModelChoiceField(
  239. queryset=SiteGroup.objects.all(),
  240. required=False
  241. )
  242. slug = SlugField()
  243. class Meta:
  244. model = SiteGroup
  245. fields = (
  246. 'parent', 'name', 'slug', 'description',
  247. )
  248. class SiteGroupCSVForm(CustomFieldModelCSVForm):
  249. parent = CSVModelChoiceField(
  250. queryset=SiteGroup.objects.all(),
  251. required=False,
  252. to_field_name='name',
  253. help_text='Name of parent site group'
  254. )
  255. class Meta:
  256. model = SiteGroup
  257. fields = ('name', 'slug', 'parent', 'description')
  258. class SiteGroupBulkEditForm(BootstrapMixin, CustomFieldModelBulkEditForm):
  259. pk = forms.ModelMultipleChoiceField(
  260. queryset=SiteGroup.objects.all(),
  261. widget=forms.MultipleHiddenInput
  262. )
  263. parent = DynamicModelChoiceField(
  264. queryset=SiteGroup.objects.all(),
  265. required=False
  266. )
  267. description = forms.CharField(
  268. max_length=200,
  269. required=False
  270. )
  271. class Meta:
  272. nullable_fields = ['parent', 'description']
  273. class SiteGroupFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  274. model = SiteGroup
  275. field_groups = [
  276. ['q'],
  277. ['parent_id'],
  278. ]
  279. q = forms.CharField(
  280. required=False,
  281. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  282. label=_('Search')
  283. )
  284. parent_id = DynamicModelMultipleChoiceField(
  285. queryset=SiteGroup.objects.all(),
  286. required=False,
  287. label=_('Parent group'),
  288. fetch_trigger='open'
  289. )
  290. #
  291. # Sites
  292. #
  293. class SiteForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
  294. region = DynamicModelChoiceField(
  295. queryset=Region.objects.all(),
  296. required=False
  297. )
  298. group = DynamicModelChoiceField(
  299. queryset=SiteGroup.objects.all(),
  300. required=False
  301. )
  302. slug = SlugField()
  303. time_zone = TimeZoneFormField(
  304. choices=add_blank_choice(TimeZoneFormField().choices),
  305. required=False,
  306. widget=StaticSelect()
  307. )
  308. comments = CommentField()
  309. tags = DynamicModelMultipleChoiceField(
  310. queryset=Tag.objects.all(),
  311. required=False
  312. )
  313. class Meta:
  314. model = Site
  315. fields = [
  316. 'name', 'slug', 'status', 'region', 'group', 'tenant_group', 'tenant', 'facility', 'asn', 'time_zone',
  317. 'description', 'physical_address', 'shipping_address', 'latitude', 'longitude', 'contact_name',
  318. 'contact_phone', 'contact_email', 'comments', 'tags',
  319. ]
  320. fieldsets = (
  321. ('Site', (
  322. 'name', 'slug', 'status', 'region', 'group', 'facility', 'asn', 'time_zone', 'description', 'tags',
  323. )),
  324. ('Tenancy', ('tenant_group', 'tenant')),
  325. ('Contact Info', (
  326. 'physical_address', 'shipping_address', 'latitude', 'longitude', 'contact_name', 'contact_phone',
  327. 'contact_email',
  328. )),
  329. )
  330. widgets = {
  331. 'physical_address': SmallTextarea(
  332. attrs={
  333. 'rows': 3,
  334. }
  335. ),
  336. 'shipping_address': SmallTextarea(
  337. attrs={
  338. 'rows': 3,
  339. }
  340. ),
  341. 'status': StaticSelect(),
  342. 'time_zone': StaticSelect(),
  343. }
  344. help_texts = {
  345. 'name': "Full name of the site",
  346. 'facility': "Data center provider and facility (e.g. Equinix NY7)",
  347. 'asn': "BGP autonomous system number",
  348. 'time_zone': "Local time zone",
  349. 'description': "Short description (will appear in sites list)",
  350. 'physical_address': "Physical location of the building (e.g. for GPS)",
  351. 'shipping_address': "If different from the physical address",
  352. 'latitude': "Latitude in decimal format (xx.yyyyyy)",
  353. 'longitude': "Longitude in decimal format (xx.yyyyyy)"
  354. }
  355. class SiteCSVForm(CustomFieldModelCSVForm):
  356. status = CSVChoiceField(
  357. choices=SiteStatusChoices,
  358. required=False,
  359. help_text='Operational status'
  360. )
  361. region = CSVModelChoiceField(
  362. queryset=Region.objects.all(),
  363. required=False,
  364. to_field_name='name',
  365. help_text='Assigned region'
  366. )
  367. group = CSVModelChoiceField(
  368. queryset=SiteGroup.objects.all(),
  369. required=False,
  370. to_field_name='name',
  371. help_text='Assigned group'
  372. )
  373. tenant = CSVModelChoiceField(
  374. queryset=Tenant.objects.all(),
  375. required=False,
  376. to_field_name='name',
  377. help_text='Assigned tenant'
  378. )
  379. class Meta:
  380. model = Site
  381. fields = (
  382. 'name', 'slug', 'status', 'region', 'group', 'tenant', 'facility', 'asn', 'time_zone', 'description',
  383. 'physical_address', 'shipping_address', 'latitude', 'longitude', 'contact_name', 'contact_phone',
  384. 'contact_email', 'comments',
  385. )
  386. help_texts = {
  387. 'time_zone': mark_safe(
  388. 'Time zone (<a href="https://en.wikipedia.org/wiki/List_of_tz_database_time_zones">available options</a>)'
  389. )
  390. }
  391. class SiteBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  392. pk = forms.ModelMultipleChoiceField(
  393. queryset=Site.objects.all(),
  394. widget=forms.MultipleHiddenInput
  395. )
  396. status = forms.ChoiceField(
  397. choices=add_blank_choice(SiteStatusChoices),
  398. required=False,
  399. initial='',
  400. widget=StaticSelect()
  401. )
  402. region = DynamicModelChoiceField(
  403. queryset=Region.objects.all(),
  404. required=False
  405. )
  406. group = DynamicModelChoiceField(
  407. queryset=SiteGroup.objects.all(),
  408. required=False
  409. )
  410. tenant = DynamicModelChoiceField(
  411. queryset=Tenant.objects.all(),
  412. required=False
  413. )
  414. asn = forms.IntegerField(
  415. min_value=BGP_ASN_MIN,
  416. max_value=BGP_ASN_MAX,
  417. required=False,
  418. label='ASN'
  419. )
  420. description = forms.CharField(
  421. max_length=100,
  422. required=False
  423. )
  424. time_zone = TimeZoneFormField(
  425. choices=add_blank_choice(TimeZoneFormField().choices),
  426. required=False,
  427. widget=StaticSelect()
  428. )
  429. class Meta:
  430. nullable_fields = [
  431. 'region', 'group', 'tenant', 'asn', 'description', 'time_zone',
  432. ]
  433. class SiteFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldModelFilterForm):
  434. model = Site
  435. field_order = ['q', 'status', 'region_id', 'tenant_group_id', 'tenant_id']
  436. field_groups = [
  437. ['q', 'tag'],
  438. ['status', 'region_id', 'group_id'],
  439. ['tenant_group_id', 'tenant_id'],
  440. ]
  441. q = forms.CharField(
  442. required=False,
  443. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  444. label=_('Search')
  445. )
  446. status = forms.MultipleChoiceField(
  447. choices=SiteStatusChoices,
  448. required=False,
  449. widget=StaticSelectMultiple(),
  450. )
  451. region_id = DynamicModelMultipleChoiceField(
  452. queryset=Region.objects.all(),
  453. required=False,
  454. label=_('Region'),
  455. fetch_trigger='open'
  456. )
  457. group_id = DynamicModelMultipleChoiceField(
  458. queryset=SiteGroup.objects.all(),
  459. required=False,
  460. label=_('Site group'),
  461. fetch_trigger='open'
  462. )
  463. tag = TagFilterField(model)
  464. #
  465. # Locations
  466. #
  467. class LocationForm(BootstrapMixin, CustomFieldModelForm):
  468. region = DynamicModelChoiceField(
  469. queryset=Region.objects.all(),
  470. required=False,
  471. initial_params={
  472. 'sites': '$site'
  473. }
  474. )
  475. site_group = DynamicModelChoiceField(
  476. queryset=SiteGroup.objects.all(),
  477. required=False,
  478. initial_params={
  479. 'sites': '$site'
  480. }
  481. )
  482. site = DynamicModelChoiceField(
  483. queryset=Site.objects.all(),
  484. query_params={
  485. 'region_id': '$region',
  486. 'group_id': '$site_group',
  487. }
  488. )
  489. parent = DynamicModelChoiceField(
  490. queryset=Location.objects.all(),
  491. required=False,
  492. query_params={
  493. 'site_id': '$site'
  494. }
  495. )
  496. slug = SlugField()
  497. class Meta:
  498. model = Location
  499. fields = (
  500. 'region', 'site_group', 'site', 'parent', 'name', 'slug', 'description',
  501. )
  502. class LocationCSVForm(CustomFieldModelCSVForm):
  503. site = CSVModelChoiceField(
  504. queryset=Site.objects.all(),
  505. to_field_name='name',
  506. help_text='Assigned site'
  507. )
  508. parent = CSVModelChoiceField(
  509. queryset=Location.objects.all(),
  510. required=False,
  511. to_field_name='name',
  512. help_text='Parent location',
  513. error_messages={
  514. 'invalid_choice': 'Location not found.',
  515. }
  516. )
  517. class Meta:
  518. model = Location
  519. fields = ('site', 'parent', 'name', 'slug', 'description')
  520. class LocationBulkEditForm(BootstrapMixin, CustomFieldModelBulkEditForm):
  521. pk = forms.ModelMultipleChoiceField(
  522. queryset=Location.objects.all(),
  523. widget=forms.MultipleHiddenInput
  524. )
  525. site = DynamicModelChoiceField(
  526. queryset=Site.objects.all(),
  527. required=False
  528. )
  529. parent = DynamicModelChoiceField(
  530. queryset=Location.objects.all(),
  531. required=False,
  532. query_params={
  533. 'site_id': '$site'
  534. }
  535. )
  536. description = forms.CharField(
  537. max_length=200,
  538. required=False
  539. )
  540. class Meta:
  541. nullable_fields = ['parent', 'description']
  542. class LocationFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  543. model = Location
  544. q = forms.CharField(
  545. required=False,
  546. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  547. label=_('Search')
  548. )
  549. region_id = DynamicModelMultipleChoiceField(
  550. queryset=Region.objects.all(),
  551. required=False,
  552. label=_('Region'),
  553. fetch_trigger='open'
  554. )
  555. site_group_id = DynamicModelMultipleChoiceField(
  556. queryset=SiteGroup.objects.all(),
  557. required=False,
  558. label=_('Site group'),
  559. fetch_trigger='open'
  560. )
  561. site_id = DynamicModelMultipleChoiceField(
  562. queryset=Site.objects.all(),
  563. required=False,
  564. query_params={
  565. 'region_id': '$region_id',
  566. 'group_id': '$site_group_id',
  567. },
  568. label=_('Site'),
  569. fetch_trigger='open'
  570. )
  571. parent_id = DynamicModelMultipleChoiceField(
  572. queryset=Location.objects.all(),
  573. required=False,
  574. query_params={
  575. 'region_id': '$region_id',
  576. 'site_id': '$site_id',
  577. },
  578. label=_('Parent'),
  579. fetch_trigger='open'
  580. )
  581. #
  582. # Rack roles
  583. #
  584. class RackRoleForm(BootstrapMixin, CustomFieldModelForm):
  585. slug = SlugField()
  586. class Meta:
  587. model = RackRole
  588. fields = [
  589. 'name', 'slug', 'color', 'description',
  590. ]
  591. class RackRoleCSVForm(CustomFieldModelCSVForm):
  592. slug = SlugField()
  593. class Meta:
  594. model = RackRole
  595. fields = ('name', 'slug', 'color', 'description')
  596. help_texts = {
  597. 'color': mark_safe('RGB color in hexadecimal (e.g. <code>00ff00</code>)'),
  598. }
  599. class RackRoleBulkEditForm(BootstrapMixin, CustomFieldModelBulkEditForm):
  600. pk = forms.ModelMultipleChoiceField(
  601. queryset=RackRole.objects.all(),
  602. widget=forms.MultipleHiddenInput
  603. )
  604. color = ColorField(
  605. required=False
  606. )
  607. description = forms.CharField(
  608. max_length=200,
  609. required=False
  610. )
  611. class Meta:
  612. nullable_fields = ['color', 'description']
  613. #
  614. # Racks
  615. #
  616. class RackForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
  617. region = DynamicModelChoiceField(
  618. queryset=Region.objects.all(),
  619. required=False,
  620. initial_params={
  621. 'sites': '$site'
  622. }
  623. )
  624. site_group = DynamicModelChoiceField(
  625. queryset=SiteGroup.objects.all(),
  626. required=False,
  627. initial_params={
  628. 'sites': '$site'
  629. }
  630. )
  631. site = DynamicModelChoiceField(
  632. queryset=Site.objects.all(),
  633. query_params={
  634. 'region_id': '$region',
  635. 'group_id': '$site_group',
  636. }
  637. )
  638. location = DynamicModelChoiceField(
  639. queryset=Location.objects.all(),
  640. required=False,
  641. query_params={
  642. 'site_id': '$site'
  643. }
  644. )
  645. role = DynamicModelChoiceField(
  646. queryset=RackRole.objects.all(),
  647. required=False
  648. )
  649. comments = CommentField()
  650. tags = DynamicModelMultipleChoiceField(
  651. queryset=Tag.objects.all(),
  652. required=False
  653. )
  654. class Meta:
  655. model = Rack
  656. fields = [
  657. 'region', 'site_group', 'site', 'location', 'name', 'facility_id', 'tenant_group', 'tenant', 'status',
  658. 'role', 'serial', 'asset_tag', 'type', 'width', 'u_height', 'desc_units', 'outer_width', 'outer_depth',
  659. 'outer_unit', 'comments', 'tags',
  660. ]
  661. help_texts = {
  662. 'site': "The site at which the rack exists",
  663. 'name': "Organizational rack name",
  664. 'facility_id': "The unique rack ID assigned by the facility",
  665. 'u_height': "Height in rack units",
  666. }
  667. widgets = {
  668. 'status': StaticSelect(),
  669. 'type': StaticSelect(),
  670. 'width': StaticSelect(),
  671. 'outer_unit': StaticSelect(),
  672. }
  673. class RackCSVForm(CustomFieldModelCSVForm):
  674. site = CSVModelChoiceField(
  675. queryset=Site.objects.all(),
  676. to_field_name='name'
  677. )
  678. location = CSVModelChoiceField(
  679. queryset=Location.objects.all(),
  680. required=False,
  681. to_field_name='name'
  682. )
  683. tenant = CSVModelChoiceField(
  684. queryset=Tenant.objects.all(),
  685. required=False,
  686. to_field_name='name',
  687. help_text='Name of assigned tenant'
  688. )
  689. status = CSVChoiceField(
  690. choices=RackStatusChoices,
  691. required=False,
  692. help_text='Operational status'
  693. )
  694. role = CSVModelChoiceField(
  695. queryset=RackRole.objects.all(),
  696. required=False,
  697. to_field_name='name',
  698. help_text='Name of assigned role'
  699. )
  700. type = CSVChoiceField(
  701. choices=RackTypeChoices,
  702. required=False,
  703. help_text='Rack type'
  704. )
  705. width = forms.ChoiceField(
  706. choices=RackWidthChoices,
  707. help_text='Rail-to-rail width (in inches)'
  708. )
  709. outer_unit = CSVChoiceField(
  710. choices=RackDimensionUnitChoices,
  711. required=False,
  712. help_text='Unit for outer dimensions'
  713. )
  714. class Meta:
  715. model = Rack
  716. fields = (
  717. 'site', 'location', 'name', 'facility_id', 'tenant', 'status', 'role', 'type', 'serial', 'asset_tag',
  718. 'width', 'u_height', 'desc_units', 'outer_width', 'outer_depth', 'outer_unit', 'comments',
  719. )
  720. def __init__(self, data=None, *args, **kwargs):
  721. super().__init__(data, *args, **kwargs)
  722. if data:
  723. # Limit location queryset by assigned site
  724. params = {f"site__{self.fields['site'].to_field_name}": data.get('site')}
  725. self.fields['location'].queryset = self.fields['location'].queryset.filter(**params)
  726. class RackBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  727. pk = forms.ModelMultipleChoiceField(
  728. queryset=Rack.objects.all(),
  729. widget=forms.MultipleHiddenInput
  730. )
  731. region = DynamicModelChoiceField(
  732. queryset=Region.objects.all(),
  733. required=False,
  734. initial_params={
  735. 'sites': '$site'
  736. }
  737. )
  738. site_group = DynamicModelChoiceField(
  739. queryset=SiteGroup.objects.all(),
  740. required=False,
  741. initial_params={
  742. 'sites': '$site'
  743. }
  744. )
  745. site = DynamicModelChoiceField(
  746. queryset=Site.objects.all(),
  747. required=False,
  748. query_params={
  749. 'region_id': '$region',
  750. 'group_id': '$site_group',
  751. }
  752. )
  753. location = DynamicModelChoiceField(
  754. queryset=Location.objects.all(),
  755. required=False,
  756. query_params={
  757. 'site_id': '$site'
  758. }
  759. )
  760. tenant = DynamicModelChoiceField(
  761. queryset=Tenant.objects.all(),
  762. required=False
  763. )
  764. status = forms.ChoiceField(
  765. choices=add_blank_choice(RackStatusChoices),
  766. required=False,
  767. initial='',
  768. widget=StaticSelect()
  769. )
  770. role = DynamicModelChoiceField(
  771. queryset=RackRole.objects.all(),
  772. required=False
  773. )
  774. serial = forms.CharField(
  775. max_length=50,
  776. required=False,
  777. label='Serial Number'
  778. )
  779. asset_tag = forms.CharField(
  780. max_length=50,
  781. required=False
  782. )
  783. type = forms.ChoiceField(
  784. choices=add_blank_choice(RackTypeChoices),
  785. required=False,
  786. widget=StaticSelect()
  787. )
  788. width = forms.ChoiceField(
  789. choices=add_blank_choice(RackWidthChoices),
  790. required=False,
  791. widget=StaticSelect()
  792. )
  793. u_height = forms.IntegerField(
  794. required=False,
  795. label='Height (U)'
  796. )
  797. desc_units = forms.NullBooleanField(
  798. required=False,
  799. widget=BulkEditNullBooleanSelect,
  800. label='Descending units'
  801. )
  802. outer_width = forms.IntegerField(
  803. required=False,
  804. min_value=1
  805. )
  806. outer_depth = forms.IntegerField(
  807. required=False,
  808. min_value=1
  809. )
  810. outer_unit = forms.ChoiceField(
  811. choices=add_blank_choice(RackDimensionUnitChoices),
  812. required=False,
  813. widget=StaticSelect()
  814. )
  815. comments = CommentField(
  816. widget=SmallTextarea,
  817. label='Comments'
  818. )
  819. class Meta:
  820. nullable_fields = [
  821. 'location', 'tenant', 'role', 'serial', 'asset_tag', 'outer_width', 'outer_depth', 'outer_unit', 'comments',
  822. ]
  823. class RackFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldModelFilterForm):
  824. model = Rack
  825. field_order = ['q', 'region_id', 'site_id', 'location_id', 'status', 'role_id', 'tenant_group_id', 'tenant_id']
  826. field_groups = [
  827. ['q', 'tag'],
  828. ['region_id', 'site_id', 'location_id'],
  829. ['status', 'role_id'],
  830. ['type', 'width', 'asset_tag'],
  831. ['tenant_group_id', 'tenant_id'],
  832. ]
  833. q = forms.CharField(
  834. required=False,
  835. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  836. label=_('Search')
  837. )
  838. region_id = DynamicModelMultipleChoiceField(
  839. queryset=Region.objects.all(),
  840. required=False,
  841. label=_('Region'),
  842. fetch_trigger='open'
  843. )
  844. site_id = DynamicModelMultipleChoiceField(
  845. queryset=Site.objects.all(),
  846. required=False,
  847. query_params={
  848. 'region_id': '$region_id'
  849. },
  850. label=_('Site'),
  851. fetch_trigger='open'
  852. )
  853. location_id = DynamicModelMultipleChoiceField(
  854. queryset=Location.objects.all(),
  855. required=False,
  856. null_option='None',
  857. query_params={
  858. 'site_id': '$site_id'
  859. },
  860. label=_('Location'),
  861. fetch_trigger='open'
  862. )
  863. status = forms.MultipleChoiceField(
  864. choices=RackStatusChoices,
  865. required=False,
  866. widget=StaticSelectMultiple()
  867. )
  868. type = forms.MultipleChoiceField(
  869. choices=RackTypeChoices,
  870. required=False,
  871. widget=StaticSelectMultiple()
  872. )
  873. width = forms.MultipleChoiceField(
  874. choices=RackWidthChoices,
  875. required=False,
  876. widget=StaticSelectMultiple()
  877. )
  878. role_id = DynamicModelMultipleChoiceField(
  879. queryset=RackRole.objects.all(),
  880. required=False,
  881. null_option='None',
  882. label=_('Role'),
  883. fetch_trigger='open'
  884. )
  885. asset_tag = forms.CharField(
  886. required=False
  887. )
  888. tag = TagFilterField(model)
  889. #
  890. # Rack elevations
  891. #
  892. class RackElevationFilterForm(RackFilterForm):
  893. field_order = [
  894. 'q', 'region_id', 'site_id', 'location_id', 'id', 'status', 'role_id', 'tenant_group_id',
  895. 'tenant_id',
  896. ]
  897. id = DynamicModelMultipleChoiceField(
  898. queryset=Rack.objects.all(),
  899. label=_('Rack'),
  900. required=False,
  901. query_params={
  902. 'site_id': '$site_id',
  903. 'location_id': '$location_id',
  904. },
  905. fetch_trigger='open'
  906. )
  907. #
  908. # Rack reservations
  909. #
  910. class RackReservationForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
  911. region = DynamicModelChoiceField(
  912. queryset=Region.objects.all(),
  913. required=False,
  914. initial_params={
  915. 'sites': '$site'
  916. },
  917. fetch_trigger='open'
  918. )
  919. site_group = DynamicModelChoiceField(
  920. queryset=SiteGroup.objects.all(),
  921. required=False,
  922. initial_params={
  923. 'sites': '$site'
  924. },
  925. fetch_trigger='open'
  926. )
  927. site = DynamicModelChoiceField(
  928. queryset=Site.objects.all(),
  929. required=False,
  930. query_params={
  931. 'region_id': '$region',
  932. 'group_id': '$site_group',
  933. },
  934. fetch_trigger='open'
  935. )
  936. location = DynamicModelChoiceField(
  937. queryset=Location.objects.all(),
  938. required=False,
  939. query_params={
  940. 'site_id': '$site'
  941. },
  942. fetch_trigger='open'
  943. )
  944. rack = DynamicModelChoiceField(
  945. queryset=Rack.objects.all(),
  946. query_params={
  947. 'site_id': '$site',
  948. 'location_id': '$location',
  949. },
  950. fetch_trigger='open'
  951. )
  952. units = NumericArrayField(
  953. base_field=forms.IntegerField(),
  954. help_text="Comma-separated list of numeric unit IDs. A range may be specified using a hyphen."
  955. )
  956. user = forms.ModelChoiceField(
  957. queryset=User.objects.order_by(
  958. 'username'
  959. ),
  960. widget=StaticSelect()
  961. )
  962. tags = DynamicModelMultipleChoiceField(
  963. queryset=Tag.objects.all(),
  964. required=False,
  965. fetch_trigger='open'
  966. )
  967. class Meta:
  968. model = RackReservation
  969. fields = [
  970. 'region', 'site_group', 'site', 'location', 'rack', 'units', 'user', 'tenant_group', 'tenant',
  971. 'description', 'tags',
  972. ]
  973. fieldsets = (
  974. ('Reservation', ('region', 'site', 'location', 'rack', 'units', 'user', 'description', 'tags')),
  975. ('Tenancy', ('tenant_group', 'tenant')),
  976. )
  977. class RackReservationCSVForm(CustomFieldModelCSVForm):
  978. site = CSVModelChoiceField(
  979. queryset=Site.objects.all(),
  980. to_field_name='name',
  981. help_text='Parent site'
  982. )
  983. location = CSVModelChoiceField(
  984. queryset=Location.objects.all(),
  985. to_field_name='name',
  986. required=False,
  987. help_text="Rack's location (if any)"
  988. )
  989. rack = CSVModelChoiceField(
  990. queryset=Rack.objects.all(),
  991. to_field_name='name',
  992. help_text='Rack'
  993. )
  994. units = SimpleArrayField(
  995. base_field=forms.IntegerField(),
  996. required=True,
  997. help_text='Comma-separated list of individual unit numbers'
  998. )
  999. tenant = CSVModelChoiceField(
  1000. queryset=Tenant.objects.all(),
  1001. required=False,
  1002. to_field_name='name',
  1003. help_text='Assigned tenant'
  1004. )
  1005. class Meta:
  1006. model = RackReservation
  1007. fields = ('site', 'location', 'rack', 'units', 'tenant', 'description')
  1008. def __init__(self, data=None, *args, **kwargs):
  1009. super().__init__(data, *args, **kwargs)
  1010. if data:
  1011. # Limit location queryset by assigned site
  1012. params = {f"site__{self.fields['site'].to_field_name}": data.get('site')}
  1013. self.fields['location'].queryset = self.fields['location'].queryset.filter(**params)
  1014. # Limit rack queryset by assigned site and group
  1015. params = {
  1016. f"site__{self.fields['site'].to_field_name}": data.get('site'),
  1017. f"location__{self.fields['location'].to_field_name}": data.get('location'),
  1018. }
  1019. self.fields['rack'].queryset = self.fields['rack'].queryset.filter(**params)
  1020. class RackReservationBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  1021. pk = forms.ModelMultipleChoiceField(
  1022. queryset=RackReservation.objects.all(),
  1023. widget=forms.MultipleHiddenInput()
  1024. )
  1025. user = forms.ModelChoiceField(
  1026. queryset=User.objects.order_by(
  1027. 'username'
  1028. ),
  1029. required=False,
  1030. widget=StaticSelect()
  1031. )
  1032. tenant = DynamicModelChoiceField(
  1033. queryset=Tenant.objects.all(),
  1034. required=False
  1035. )
  1036. description = forms.CharField(
  1037. max_length=100,
  1038. required=False
  1039. )
  1040. class Meta:
  1041. nullable_fields = []
  1042. class RackReservationFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldModelFilterForm):
  1043. model = RackReservation
  1044. field_order = ['q', 'region_id', 'site_id', 'location_id', 'user_id', 'tenant_group_id', 'tenant_id']
  1045. field_groups = [
  1046. ['q', 'tag'],
  1047. ['user_id'],
  1048. ['region_id', 'site_id', 'location_id'],
  1049. ['tenant_group_id', 'tenant_id'],
  1050. ]
  1051. q = forms.CharField(
  1052. required=False,
  1053. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  1054. label=_('Search')
  1055. )
  1056. region_id = DynamicModelMultipleChoiceField(
  1057. queryset=Region.objects.all(),
  1058. required=False,
  1059. label=_('Region'),
  1060. fetch_trigger='open'
  1061. )
  1062. site_id = DynamicModelMultipleChoiceField(
  1063. queryset=Site.objects.all(),
  1064. required=False,
  1065. query_params={
  1066. 'region_id': '$region_id'
  1067. },
  1068. label=_('Site'),
  1069. fetch_trigger='open'
  1070. )
  1071. location_id = DynamicModelMultipleChoiceField(
  1072. queryset=Location.objects.prefetch_related('site'),
  1073. required=False,
  1074. label=_('Location'),
  1075. null_option='None',
  1076. fetch_trigger='open'
  1077. )
  1078. user_id = DynamicModelMultipleChoiceField(
  1079. queryset=User.objects.all(),
  1080. required=False,
  1081. label=_('User'),
  1082. widget=APISelectMultiple(
  1083. api_url='/api/users/users/',
  1084. ),
  1085. fetch_trigger='open'
  1086. )
  1087. tag = TagFilterField(model)
  1088. #
  1089. # Manufacturers
  1090. #
  1091. class ManufacturerForm(BootstrapMixin, CustomFieldModelForm):
  1092. slug = SlugField()
  1093. class Meta:
  1094. model = Manufacturer
  1095. fields = [
  1096. 'name', 'slug', 'description',
  1097. ]
  1098. class ManufacturerCSVForm(CustomFieldModelCSVForm):
  1099. class Meta:
  1100. model = Manufacturer
  1101. fields = ('name', 'slug', 'description')
  1102. class ManufacturerBulkEditForm(BootstrapMixin, CustomFieldModelBulkEditForm):
  1103. pk = forms.ModelMultipleChoiceField(
  1104. queryset=Manufacturer.objects.all(),
  1105. widget=forms.MultipleHiddenInput
  1106. )
  1107. description = forms.CharField(
  1108. max_length=200,
  1109. required=False
  1110. )
  1111. class Meta:
  1112. nullable_fields = ['description']
  1113. #
  1114. # Device types
  1115. #
  1116. class DeviceTypeForm(BootstrapMixin, CustomFieldModelForm):
  1117. manufacturer = DynamicModelChoiceField(
  1118. queryset=Manufacturer.objects.all()
  1119. )
  1120. slug = SlugField(
  1121. slug_source='model'
  1122. )
  1123. comments = CommentField()
  1124. tags = DynamicModelMultipleChoiceField(
  1125. queryset=Tag.objects.all(),
  1126. required=False
  1127. )
  1128. class Meta:
  1129. model = DeviceType
  1130. fields = [
  1131. 'manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'subdevice_role',
  1132. 'front_image', 'rear_image', 'comments', 'tags',
  1133. ]
  1134. fieldsets = (
  1135. ('Device Type', (
  1136. 'manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'subdevice_role', 'tags',
  1137. )),
  1138. ('Images', ('front_image', 'rear_image')),
  1139. )
  1140. widgets = {
  1141. 'subdevice_role': StaticSelect(),
  1142. 'front_image': forms.ClearableFileInput(attrs={
  1143. 'accept': DEVICETYPE_IMAGE_FORMATS
  1144. }),
  1145. 'rear_image': forms.ClearableFileInput(attrs={
  1146. 'accept': DEVICETYPE_IMAGE_FORMATS
  1147. })
  1148. }
  1149. class DeviceTypeImportForm(BootstrapMixin, forms.ModelForm):
  1150. manufacturer = forms.ModelChoiceField(
  1151. queryset=Manufacturer.objects.all(),
  1152. to_field_name='name'
  1153. )
  1154. class Meta:
  1155. model = DeviceType
  1156. fields = [
  1157. 'manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'subdevice_role',
  1158. 'comments',
  1159. ]
  1160. class DeviceTypeBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  1161. pk = forms.ModelMultipleChoiceField(
  1162. queryset=DeviceType.objects.all(),
  1163. widget=forms.MultipleHiddenInput()
  1164. )
  1165. manufacturer = DynamicModelChoiceField(
  1166. queryset=Manufacturer.objects.all(),
  1167. required=False
  1168. )
  1169. u_height = forms.IntegerField(
  1170. min_value=1,
  1171. required=False
  1172. )
  1173. is_full_depth = forms.NullBooleanField(
  1174. required=False,
  1175. widget=BulkEditNullBooleanSelect(),
  1176. label='Is full depth'
  1177. )
  1178. class Meta:
  1179. nullable_fields = []
  1180. class DeviceTypeFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  1181. model = DeviceType
  1182. field_groups = [
  1183. ['q', 'tag'],
  1184. ['manufacturer_id', 'subdevice_role'],
  1185. ['console_ports', 'console_server_ports', 'power_ports', 'power_outlets', 'interfaces', 'pass_through_ports'],
  1186. ]
  1187. q = forms.CharField(
  1188. required=False,
  1189. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  1190. label=_('Search')
  1191. )
  1192. manufacturer_id = DynamicModelMultipleChoiceField(
  1193. queryset=Manufacturer.objects.all(),
  1194. required=False,
  1195. label=_('Manufacturer'),
  1196. fetch_trigger='open'
  1197. )
  1198. subdevice_role = forms.MultipleChoiceField(
  1199. choices=add_blank_choice(SubdeviceRoleChoices),
  1200. required=False,
  1201. widget=StaticSelectMultiple()
  1202. )
  1203. console_ports = forms.NullBooleanField(
  1204. required=False,
  1205. label='Has console ports',
  1206. widget=StaticSelect(
  1207. choices=BOOLEAN_WITH_BLANK_CHOICES
  1208. )
  1209. )
  1210. console_server_ports = forms.NullBooleanField(
  1211. required=False,
  1212. label='Has console server ports',
  1213. widget=StaticSelect(
  1214. choices=BOOLEAN_WITH_BLANK_CHOICES
  1215. )
  1216. )
  1217. power_ports = forms.NullBooleanField(
  1218. required=False,
  1219. label='Has power ports',
  1220. widget=StaticSelect(
  1221. choices=BOOLEAN_WITH_BLANK_CHOICES
  1222. )
  1223. )
  1224. power_outlets = forms.NullBooleanField(
  1225. required=False,
  1226. label='Has power outlets',
  1227. widget=StaticSelect(
  1228. choices=BOOLEAN_WITH_BLANK_CHOICES
  1229. )
  1230. )
  1231. interfaces = forms.NullBooleanField(
  1232. required=False,
  1233. label='Has interfaces',
  1234. widget=StaticSelect(
  1235. choices=BOOLEAN_WITH_BLANK_CHOICES
  1236. )
  1237. )
  1238. pass_through_ports = forms.NullBooleanField(
  1239. required=False,
  1240. label='Has pass-through ports',
  1241. widget=StaticSelect(
  1242. choices=BOOLEAN_WITH_BLANK_CHOICES
  1243. )
  1244. )
  1245. tag = TagFilterField(model)
  1246. #
  1247. # Device component templates
  1248. #
  1249. class ComponentTemplateCreateForm(BootstrapMixin, ComponentForm):
  1250. """
  1251. Base form for the creation of device component templates (subclassed from ComponentTemplateModel).
  1252. """
  1253. manufacturer = DynamicModelChoiceField(
  1254. queryset=Manufacturer.objects.all(),
  1255. required=False,
  1256. initial_params={
  1257. 'device_types': 'device_type'
  1258. }
  1259. )
  1260. device_type = DynamicModelChoiceField(
  1261. queryset=DeviceType.objects.all(),
  1262. query_params={
  1263. 'manufacturer_id': '$manufacturer'
  1264. }
  1265. )
  1266. description = forms.CharField(
  1267. required=False
  1268. )
  1269. class ConsolePortTemplateForm(BootstrapMixin, forms.ModelForm):
  1270. class Meta:
  1271. model = ConsolePortTemplate
  1272. fields = [
  1273. 'device_type', 'name', 'label', 'type', 'description',
  1274. ]
  1275. widgets = {
  1276. 'device_type': forms.HiddenInput(),
  1277. }
  1278. class ConsolePortTemplateCreateForm(ComponentTemplateCreateForm):
  1279. type = forms.ChoiceField(
  1280. choices=add_blank_choice(ConsolePortTypeChoices),
  1281. widget=StaticSelect()
  1282. )
  1283. field_order = ('manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'type', 'description')
  1284. class ConsolePortTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1285. pk = forms.ModelMultipleChoiceField(
  1286. queryset=ConsolePortTemplate.objects.all(),
  1287. widget=forms.MultipleHiddenInput()
  1288. )
  1289. label = forms.CharField(
  1290. max_length=64,
  1291. required=False
  1292. )
  1293. type = forms.ChoiceField(
  1294. choices=add_blank_choice(ConsolePortTypeChoices),
  1295. required=False,
  1296. widget=StaticSelect()
  1297. )
  1298. class Meta:
  1299. nullable_fields = ('label', 'type', 'description')
  1300. class ConsoleServerPortTemplateForm(BootstrapMixin, forms.ModelForm):
  1301. class Meta:
  1302. model = ConsoleServerPortTemplate
  1303. fields = [
  1304. 'device_type', 'name', 'label', 'type', 'description',
  1305. ]
  1306. widgets = {
  1307. 'device_type': forms.HiddenInput(),
  1308. }
  1309. class ConsoleServerPortTemplateCreateForm(ComponentTemplateCreateForm):
  1310. type = forms.ChoiceField(
  1311. choices=add_blank_choice(ConsolePortTypeChoices),
  1312. widget=StaticSelect()
  1313. )
  1314. field_order = ('manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'type', 'description')
  1315. class ConsoleServerPortTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1316. pk = forms.ModelMultipleChoiceField(
  1317. queryset=ConsoleServerPortTemplate.objects.all(),
  1318. widget=forms.MultipleHiddenInput()
  1319. )
  1320. label = forms.CharField(
  1321. max_length=64,
  1322. required=False
  1323. )
  1324. type = forms.ChoiceField(
  1325. choices=add_blank_choice(ConsolePortTypeChoices),
  1326. required=False,
  1327. widget=StaticSelect()
  1328. )
  1329. description = forms.CharField(
  1330. required=False
  1331. )
  1332. class Meta:
  1333. nullable_fields = ('label', 'type', 'description')
  1334. class PowerPortTemplateForm(BootstrapMixin, forms.ModelForm):
  1335. class Meta:
  1336. model = PowerPortTemplate
  1337. fields = [
  1338. 'device_type', 'name', 'label', 'type', 'maximum_draw', 'allocated_draw', 'description',
  1339. ]
  1340. widgets = {
  1341. 'device_type': forms.HiddenInput(),
  1342. }
  1343. class PowerPortTemplateCreateForm(ComponentTemplateCreateForm):
  1344. type = forms.ChoiceField(
  1345. choices=add_blank_choice(PowerPortTypeChoices),
  1346. required=False
  1347. )
  1348. maximum_draw = forms.IntegerField(
  1349. min_value=1,
  1350. required=False,
  1351. help_text="Maximum power draw (watts)"
  1352. )
  1353. allocated_draw = forms.IntegerField(
  1354. min_value=1,
  1355. required=False,
  1356. help_text="Allocated power draw (watts)"
  1357. )
  1358. field_order = (
  1359. 'manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'type', 'maximum_draw', 'allocated_draw',
  1360. 'description',
  1361. )
  1362. class PowerPortTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1363. pk = forms.ModelMultipleChoiceField(
  1364. queryset=PowerPortTemplate.objects.all(),
  1365. widget=forms.MultipleHiddenInput()
  1366. )
  1367. label = forms.CharField(
  1368. max_length=64,
  1369. required=False
  1370. )
  1371. type = forms.ChoiceField(
  1372. choices=add_blank_choice(PowerPortTypeChoices),
  1373. required=False,
  1374. widget=StaticSelect()
  1375. )
  1376. maximum_draw = forms.IntegerField(
  1377. min_value=1,
  1378. required=False,
  1379. help_text="Maximum power draw (watts)"
  1380. )
  1381. allocated_draw = forms.IntegerField(
  1382. min_value=1,
  1383. required=False,
  1384. help_text="Allocated power draw (watts)"
  1385. )
  1386. description = forms.CharField(
  1387. required=False
  1388. )
  1389. class Meta:
  1390. nullable_fields = ('label', 'type', 'maximum_draw', 'allocated_draw', 'description')
  1391. class PowerOutletTemplateForm(BootstrapMixin, forms.ModelForm):
  1392. class Meta:
  1393. model = PowerOutletTemplate
  1394. fields = [
  1395. 'device_type', 'name', 'label', 'type', 'power_port', 'feed_leg', 'description',
  1396. ]
  1397. widgets = {
  1398. 'device_type': forms.HiddenInput(),
  1399. }
  1400. def __init__(self, *args, **kwargs):
  1401. super().__init__(*args, **kwargs)
  1402. # Limit power_port choices to current DeviceType
  1403. if hasattr(self.instance, 'device_type'):
  1404. self.fields['power_port'].queryset = PowerPortTemplate.objects.filter(
  1405. device_type=self.instance.device_type
  1406. )
  1407. class PowerOutletTemplateCreateForm(ComponentTemplateCreateForm):
  1408. type = forms.ChoiceField(
  1409. choices=add_blank_choice(PowerOutletTypeChoices),
  1410. required=False
  1411. )
  1412. power_port = forms.ModelChoiceField(
  1413. queryset=PowerPortTemplate.objects.all(),
  1414. required=False
  1415. )
  1416. feed_leg = forms.ChoiceField(
  1417. choices=add_blank_choice(PowerOutletFeedLegChoices),
  1418. required=False,
  1419. widget=StaticSelect()
  1420. )
  1421. field_order = (
  1422. 'manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'type', 'power_port', 'feed_leg',
  1423. 'description',
  1424. )
  1425. def __init__(self, *args, **kwargs):
  1426. super().__init__(*args, **kwargs)
  1427. # Limit power_port choices to current DeviceType
  1428. device_type = DeviceType.objects.get(
  1429. pk=self.initial.get('device_type') or self.data.get('device_type')
  1430. )
  1431. self.fields['power_port'].queryset = PowerPortTemplate.objects.filter(
  1432. device_type=device_type
  1433. )
  1434. class PowerOutletTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1435. pk = forms.ModelMultipleChoiceField(
  1436. queryset=PowerOutletTemplate.objects.all(),
  1437. widget=forms.MultipleHiddenInput()
  1438. )
  1439. device_type = forms.ModelChoiceField(
  1440. queryset=DeviceType.objects.all(),
  1441. required=False,
  1442. disabled=True,
  1443. widget=forms.HiddenInput()
  1444. )
  1445. label = forms.CharField(
  1446. max_length=64,
  1447. required=False
  1448. )
  1449. type = forms.ChoiceField(
  1450. choices=add_blank_choice(PowerOutletTypeChoices),
  1451. required=False,
  1452. widget=StaticSelect()
  1453. )
  1454. power_port = forms.ModelChoiceField(
  1455. queryset=PowerPortTemplate.objects.all(),
  1456. required=False
  1457. )
  1458. feed_leg = forms.ChoiceField(
  1459. choices=add_blank_choice(PowerOutletFeedLegChoices),
  1460. required=False,
  1461. widget=StaticSelect()
  1462. )
  1463. description = forms.CharField(
  1464. required=False
  1465. )
  1466. class Meta:
  1467. nullable_fields = ('label', 'type', 'power_port', 'feed_leg', 'description')
  1468. def __init__(self, *args, **kwargs):
  1469. super().__init__(*args, **kwargs)
  1470. # Limit power_port queryset to PowerPortTemplates which belong to the parent DeviceType
  1471. if 'device_type' in self.initial:
  1472. device_type = DeviceType.objects.filter(pk=self.initial['device_type']).first()
  1473. self.fields['power_port'].queryset = PowerPortTemplate.objects.filter(device_type=device_type)
  1474. else:
  1475. self.fields['power_port'].choices = ()
  1476. self.fields['power_port'].widget.attrs['disabled'] = True
  1477. class InterfaceTemplateForm(BootstrapMixin, forms.ModelForm):
  1478. class Meta:
  1479. model = InterfaceTemplate
  1480. fields = [
  1481. 'device_type', 'name', 'label', 'type', 'mgmt_only', 'description',
  1482. ]
  1483. widgets = {
  1484. 'device_type': forms.HiddenInput(),
  1485. 'type': StaticSelect(),
  1486. }
  1487. class InterfaceTemplateCreateForm(ComponentTemplateCreateForm):
  1488. type = forms.ChoiceField(
  1489. choices=InterfaceTypeChoices,
  1490. widget=StaticSelect()
  1491. )
  1492. mgmt_only = forms.BooleanField(
  1493. required=False,
  1494. label='Management only'
  1495. )
  1496. field_order = ('manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'type', 'mgmt_only', 'description')
  1497. class InterfaceTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1498. pk = forms.ModelMultipleChoiceField(
  1499. queryset=InterfaceTemplate.objects.all(),
  1500. widget=forms.MultipleHiddenInput()
  1501. )
  1502. label = forms.CharField(
  1503. max_length=64,
  1504. required=False
  1505. )
  1506. type = forms.ChoiceField(
  1507. choices=add_blank_choice(InterfaceTypeChoices),
  1508. required=False,
  1509. widget=StaticSelect()
  1510. )
  1511. mgmt_only = forms.NullBooleanField(
  1512. required=False,
  1513. widget=BulkEditNullBooleanSelect,
  1514. label='Management only'
  1515. )
  1516. description = forms.CharField(
  1517. required=False
  1518. )
  1519. class Meta:
  1520. nullable_fields = ('label', 'description')
  1521. class FrontPortTemplateForm(BootstrapMixin, forms.ModelForm):
  1522. class Meta:
  1523. model = FrontPortTemplate
  1524. fields = [
  1525. 'device_type', 'name', 'label', 'type', 'color', 'rear_port', 'rear_port_position', 'description',
  1526. ]
  1527. widgets = {
  1528. 'device_type': forms.HiddenInput(),
  1529. 'rear_port': StaticSelect(),
  1530. }
  1531. def __init__(self, *args, **kwargs):
  1532. super().__init__(*args, **kwargs)
  1533. # Limit rear_port choices to current DeviceType
  1534. if hasattr(self.instance, 'device_type'):
  1535. self.fields['rear_port'].queryset = RearPortTemplate.objects.filter(
  1536. device_type=self.instance.device_type
  1537. )
  1538. class FrontPortTemplateCreateForm(ComponentTemplateCreateForm):
  1539. type = forms.ChoiceField(
  1540. choices=PortTypeChoices,
  1541. widget=StaticSelect()
  1542. )
  1543. color = ColorField(
  1544. required=False
  1545. )
  1546. rear_port_set = forms.MultipleChoiceField(
  1547. choices=[],
  1548. label='Rear ports',
  1549. help_text='Select one rear port assignment for each front port being created.',
  1550. )
  1551. field_order = (
  1552. 'manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'type', 'color', 'rear_port_set', 'description',
  1553. )
  1554. def __init__(self, *args, **kwargs):
  1555. super().__init__(*args, **kwargs)
  1556. device_type = DeviceType.objects.get(
  1557. pk=self.initial.get('device_type') or self.data.get('device_type')
  1558. )
  1559. # Determine which rear port positions are occupied. These will be excluded from the list of available mappings.
  1560. occupied_port_positions = [
  1561. (front_port.rear_port_id, front_port.rear_port_position)
  1562. for front_port in device_type.frontporttemplates.all()
  1563. ]
  1564. # Populate rear port choices
  1565. choices = []
  1566. rear_ports = RearPortTemplate.objects.filter(device_type=device_type)
  1567. for rear_port in rear_ports:
  1568. for i in range(1, rear_port.positions + 1):
  1569. if (rear_port.pk, i) not in occupied_port_positions:
  1570. choices.append(
  1571. ('{}:{}'.format(rear_port.pk, i), '{}:{}'.format(rear_port.name, i))
  1572. )
  1573. self.fields['rear_port_set'].choices = choices
  1574. def clean(self):
  1575. super().clean()
  1576. # Validate that the number of ports being created equals the number of selected (rear port, position) tuples
  1577. front_port_count = len(self.cleaned_data['name_pattern'])
  1578. rear_port_count = len(self.cleaned_data['rear_port_set'])
  1579. if front_port_count != rear_port_count:
  1580. raise forms.ValidationError({
  1581. 'rear_port_set': 'The provided name pattern will create {} ports, however {} rear port assignments '
  1582. 'were selected. These counts must match.'.format(front_port_count, rear_port_count)
  1583. })
  1584. def get_iterative_data(self, iteration):
  1585. # Assign rear port and position from selected set
  1586. rear_port, position = self.cleaned_data['rear_port_set'][iteration].split(':')
  1587. return {
  1588. 'rear_port': int(rear_port),
  1589. 'rear_port_position': int(position),
  1590. }
  1591. class FrontPortTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1592. pk = forms.ModelMultipleChoiceField(
  1593. queryset=FrontPortTemplate.objects.all(),
  1594. widget=forms.MultipleHiddenInput()
  1595. )
  1596. label = forms.CharField(
  1597. max_length=64,
  1598. required=False
  1599. )
  1600. type = forms.ChoiceField(
  1601. choices=add_blank_choice(PortTypeChoices),
  1602. required=False,
  1603. widget=StaticSelect()
  1604. )
  1605. color = ColorField(
  1606. required=False
  1607. )
  1608. description = forms.CharField(
  1609. required=False
  1610. )
  1611. class Meta:
  1612. nullable_fields = ('description',)
  1613. class RearPortTemplateForm(BootstrapMixin, forms.ModelForm):
  1614. class Meta:
  1615. model = RearPortTemplate
  1616. fields = [
  1617. 'device_type', 'name', 'label', 'type', 'color', 'positions', 'description',
  1618. ]
  1619. widgets = {
  1620. 'device_type': forms.HiddenInput(),
  1621. 'type': StaticSelect(),
  1622. }
  1623. class RearPortTemplateCreateForm(ComponentTemplateCreateForm):
  1624. type = forms.ChoiceField(
  1625. choices=PortTypeChoices,
  1626. widget=StaticSelect(),
  1627. )
  1628. color = ColorField(
  1629. required=False
  1630. )
  1631. positions = forms.IntegerField(
  1632. min_value=REARPORT_POSITIONS_MIN,
  1633. max_value=REARPORT_POSITIONS_MAX,
  1634. initial=1,
  1635. help_text='The number of front ports which may be mapped to each rear port'
  1636. )
  1637. field_order = (
  1638. 'manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'type', 'color', 'positions', 'description',
  1639. )
  1640. class RearPortTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1641. pk = forms.ModelMultipleChoiceField(
  1642. queryset=RearPortTemplate.objects.all(),
  1643. widget=forms.MultipleHiddenInput()
  1644. )
  1645. label = forms.CharField(
  1646. max_length=64,
  1647. required=False
  1648. )
  1649. type = forms.ChoiceField(
  1650. choices=add_blank_choice(PortTypeChoices),
  1651. required=False,
  1652. widget=StaticSelect()
  1653. )
  1654. color = ColorField(
  1655. required=False
  1656. )
  1657. description = forms.CharField(
  1658. required=False
  1659. )
  1660. class Meta:
  1661. nullable_fields = ('description',)
  1662. class DeviceBayTemplateForm(BootstrapMixin, forms.ModelForm):
  1663. class Meta:
  1664. model = DeviceBayTemplate
  1665. fields = [
  1666. 'device_type', 'name', 'label', 'description',
  1667. ]
  1668. widgets = {
  1669. 'device_type': forms.HiddenInput(),
  1670. }
  1671. class DeviceBayTemplateCreateForm(ComponentTemplateCreateForm):
  1672. field_order = ('manufacturer', 'device_type', 'name_pattern', 'label_pattern', 'description')
  1673. class DeviceBayTemplateBulkEditForm(BootstrapMixin, BulkEditForm):
  1674. pk = forms.ModelMultipleChoiceField(
  1675. queryset=DeviceBayTemplate.objects.all(),
  1676. widget=forms.MultipleHiddenInput()
  1677. )
  1678. label = forms.CharField(
  1679. max_length=64,
  1680. required=False
  1681. )
  1682. description = forms.CharField(
  1683. required=False
  1684. )
  1685. class Meta:
  1686. nullable_fields = ('label', 'description')
  1687. #
  1688. # Component template import forms
  1689. #
  1690. class ComponentTemplateImportForm(BootstrapMixin, forms.ModelForm):
  1691. def __init__(self, device_type, data=None, *args, **kwargs):
  1692. # Must pass the parent DeviceType on form initialization
  1693. data.update({
  1694. 'device_type': device_type.pk,
  1695. })
  1696. super().__init__(data, *args, **kwargs)
  1697. def clean_device_type(self):
  1698. data = self.cleaned_data['device_type']
  1699. # Limit fields referencing other components to the parent DeviceType
  1700. for field_name, field in self.fields.items():
  1701. if isinstance(field, forms.ModelChoiceField) and field_name != 'device_type':
  1702. field.queryset = field.queryset.filter(device_type=data)
  1703. return data
  1704. class ConsolePortTemplateImportForm(ComponentTemplateImportForm):
  1705. class Meta:
  1706. model = ConsolePortTemplate
  1707. fields = [
  1708. 'device_type', 'name', 'label', 'type', 'description',
  1709. ]
  1710. class ConsoleServerPortTemplateImportForm(ComponentTemplateImportForm):
  1711. class Meta:
  1712. model = ConsoleServerPortTemplate
  1713. fields = [
  1714. 'device_type', 'name', 'label', 'type', 'description',
  1715. ]
  1716. class PowerPortTemplateImportForm(ComponentTemplateImportForm):
  1717. class Meta:
  1718. model = PowerPortTemplate
  1719. fields = [
  1720. 'device_type', 'name', 'label', 'type', 'maximum_draw', 'allocated_draw', 'description',
  1721. ]
  1722. class PowerOutletTemplateImportForm(ComponentTemplateImportForm):
  1723. power_port = forms.ModelChoiceField(
  1724. queryset=PowerPortTemplate.objects.all(),
  1725. to_field_name='name',
  1726. required=False
  1727. )
  1728. class Meta:
  1729. model = PowerOutletTemplate
  1730. fields = [
  1731. 'device_type', 'name', 'label', 'type', 'power_port', 'feed_leg', 'description',
  1732. ]
  1733. class InterfaceTemplateImportForm(ComponentTemplateImportForm):
  1734. type = forms.ChoiceField(
  1735. choices=InterfaceTypeChoices.CHOICES
  1736. )
  1737. class Meta:
  1738. model = InterfaceTemplate
  1739. fields = [
  1740. 'device_type', 'name', 'label', 'type', 'mgmt_only', 'description',
  1741. ]
  1742. class FrontPortTemplateImportForm(ComponentTemplateImportForm):
  1743. type = forms.ChoiceField(
  1744. choices=PortTypeChoices.CHOICES
  1745. )
  1746. rear_port = forms.ModelChoiceField(
  1747. queryset=RearPortTemplate.objects.all(),
  1748. to_field_name='name'
  1749. )
  1750. class Meta:
  1751. model = FrontPortTemplate
  1752. fields = [
  1753. 'device_type', 'name', 'type', 'rear_port', 'rear_port_position', 'label', 'description',
  1754. ]
  1755. class RearPortTemplateImportForm(ComponentTemplateImportForm):
  1756. type = forms.ChoiceField(
  1757. choices=PortTypeChoices.CHOICES
  1758. )
  1759. class Meta:
  1760. model = RearPortTemplate
  1761. fields = [
  1762. 'device_type', 'name', 'type', 'positions', 'label', 'description',
  1763. ]
  1764. class DeviceBayTemplateImportForm(ComponentTemplateImportForm):
  1765. class Meta:
  1766. model = DeviceBayTemplate
  1767. fields = [
  1768. 'device_type', 'name', 'label', 'description',
  1769. ]
  1770. #
  1771. # Device roles
  1772. #
  1773. class DeviceRoleForm(BootstrapMixin, CustomFieldModelForm):
  1774. slug = SlugField()
  1775. class Meta:
  1776. model = DeviceRole
  1777. fields = [
  1778. 'name', 'slug', 'color', 'vm_role', 'description',
  1779. ]
  1780. class DeviceRoleCSVForm(CustomFieldModelCSVForm):
  1781. slug = SlugField()
  1782. class Meta:
  1783. model = DeviceRole
  1784. fields = ('name', 'slug', 'color', 'vm_role', 'description')
  1785. help_texts = {
  1786. 'color': mark_safe('RGB color in hexadecimal (e.g. <code>00ff00</code>)'),
  1787. }
  1788. class DeviceRoleBulkEditForm(BootstrapMixin, CustomFieldModelBulkEditForm):
  1789. pk = forms.ModelMultipleChoiceField(
  1790. queryset=DeviceRole.objects.all(),
  1791. widget=forms.MultipleHiddenInput
  1792. )
  1793. color = ColorField(
  1794. required=False
  1795. )
  1796. vm_role = forms.NullBooleanField(
  1797. required=False,
  1798. widget=BulkEditNullBooleanSelect,
  1799. label='VM role'
  1800. )
  1801. description = forms.CharField(
  1802. max_length=200,
  1803. required=False
  1804. )
  1805. class Meta:
  1806. nullable_fields = ['color', 'description']
  1807. #
  1808. # Platforms
  1809. #
  1810. class PlatformForm(BootstrapMixin, CustomFieldModelForm):
  1811. manufacturer = DynamicModelChoiceField(
  1812. queryset=Manufacturer.objects.all(),
  1813. required=False
  1814. )
  1815. slug = SlugField(
  1816. max_length=64
  1817. )
  1818. class Meta:
  1819. model = Platform
  1820. fields = [
  1821. 'name', 'slug', 'manufacturer', 'napalm_driver', 'napalm_args', 'description',
  1822. ]
  1823. widgets = {
  1824. 'napalm_args': SmallTextarea(),
  1825. }
  1826. class PlatformCSVForm(CustomFieldModelCSVForm):
  1827. slug = SlugField()
  1828. manufacturer = CSVModelChoiceField(
  1829. queryset=Manufacturer.objects.all(),
  1830. required=False,
  1831. to_field_name='name',
  1832. help_text='Limit platform assignments to this manufacturer'
  1833. )
  1834. class Meta:
  1835. model = Platform
  1836. fields = ('name', 'slug', 'manufacturer', 'napalm_driver', 'napalm_args', 'description')
  1837. class PlatformBulkEditForm(BootstrapMixin, CustomFieldModelBulkEditForm):
  1838. pk = forms.ModelMultipleChoiceField(
  1839. queryset=Platform.objects.all(),
  1840. widget=forms.MultipleHiddenInput
  1841. )
  1842. manufacturer = DynamicModelChoiceField(
  1843. queryset=Manufacturer.objects.all(),
  1844. required=False
  1845. )
  1846. napalm_driver = forms.CharField(
  1847. max_length=50,
  1848. required=False
  1849. )
  1850. # TODO: Bulk edit support for napalm_args
  1851. description = forms.CharField(
  1852. max_length=200,
  1853. required=False
  1854. )
  1855. class Meta:
  1856. nullable_fields = ['manufacturer', 'napalm_driver', 'description']
  1857. class PlatformFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  1858. model = Platform
  1859. q = forms.CharField(
  1860. required=False,
  1861. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  1862. label=_('Search')
  1863. )
  1864. manufacturer_id = DynamicModelMultipleChoiceField(
  1865. queryset=Manufacturer.objects.all(),
  1866. required=False,
  1867. label=_('Manufacturer'),
  1868. fetch_trigger='open'
  1869. )
  1870. #
  1871. # Devices
  1872. #
  1873. class DeviceForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
  1874. region = DynamicModelChoiceField(
  1875. queryset=Region.objects.all(),
  1876. required=False,
  1877. initial_params={
  1878. 'sites': '$site'
  1879. }
  1880. )
  1881. site_group = DynamicModelChoiceField(
  1882. queryset=SiteGroup.objects.all(),
  1883. required=False,
  1884. initial_params={
  1885. 'sites': '$site'
  1886. }
  1887. )
  1888. site = DynamicModelChoiceField(
  1889. queryset=Site.objects.all(),
  1890. query_params={
  1891. 'region_id': '$region',
  1892. 'group_id': '$site_group',
  1893. }
  1894. )
  1895. location = DynamicModelChoiceField(
  1896. queryset=Location.objects.all(),
  1897. required=False,
  1898. query_params={
  1899. 'site_id': '$site'
  1900. },
  1901. initial_params={
  1902. 'racks': '$rack'
  1903. }
  1904. )
  1905. rack = DynamicModelChoiceField(
  1906. queryset=Rack.objects.all(),
  1907. required=False,
  1908. query_params={
  1909. 'site_id': '$site',
  1910. 'location_id': '$location',
  1911. }
  1912. )
  1913. position = forms.IntegerField(
  1914. required=False,
  1915. help_text="The lowest-numbered unit occupied by the device",
  1916. widget=APISelect(
  1917. api_url='/api/dcim/racks/{{rack}}/elevation/',
  1918. attrs={
  1919. 'disabled-indicator': 'device',
  1920. 'data-query-param-face': "[\"$face\"]",
  1921. # The UI will not sort this element's options.
  1922. 'pre-sorted': ''
  1923. }
  1924. )
  1925. )
  1926. manufacturer = DynamicModelChoiceField(
  1927. queryset=Manufacturer.objects.all(),
  1928. required=False,
  1929. initial_params={
  1930. 'device_types': '$device_type'
  1931. }
  1932. )
  1933. device_type = DynamicModelChoiceField(
  1934. queryset=DeviceType.objects.all(),
  1935. query_params={
  1936. 'manufacturer_id': '$manufacturer'
  1937. }
  1938. )
  1939. device_role = DynamicModelChoiceField(
  1940. queryset=DeviceRole.objects.all()
  1941. )
  1942. platform = DynamicModelChoiceField(
  1943. queryset=Platform.objects.all(),
  1944. required=False,
  1945. query_params={
  1946. 'manufacturer_id': ['$manufacturer', 'null']
  1947. }
  1948. )
  1949. cluster_group = DynamicModelChoiceField(
  1950. queryset=ClusterGroup.objects.all(),
  1951. required=False,
  1952. null_option='None',
  1953. initial_params={
  1954. 'clusters': '$cluster'
  1955. }
  1956. )
  1957. cluster = DynamicModelChoiceField(
  1958. queryset=Cluster.objects.all(),
  1959. required=False,
  1960. query_params={
  1961. 'group_id': '$cluster_group'
  1962. }
  1963. )
  1964. comments = CommentField()
  1965. local_context_data = JSONField(
  1966. required=False,
  1967. label=''
  1968. )
  1969. tags = DynamicModelMultipleChoiceField(
  1970. queryset=Tag.objects.all(),
  1971. required=False
  1972. )
  1973. class Meta:
  1974. model = Device
  1975. fields = [
  1976. 'name', 'device_role', 'device_type', 'serial', 'asset_tag', 'region', 'site_group', 'site', 'rack',
  1977. 'location', 'position', 'face', 'status', 'platform', 'primary_ip4', 'primary_ip6', 'cluster_group',
  1978. 'cluster', 'tenant_group', 'tenant', 'comments', 'tags', 'local_context_data'
  1979. ]
  1980. help_texts = {
  1981. 'device_role': "The function this device serves",
  1982. 'serial': "Chassis serial number",
  1983. 'local_context_data': "Local config context data overwrites all source contexts in the final rendered "
  1984. "config context",
  1985. }
  1986. widgets = {
  1987. 'face': StaticSelect(),
  1988. 'status': StaticSelect(),
  1989. 'primary_ip4': StaticSelect(),
  1990. 'primary_ip6': StaticSelect(),
  1991. }
  1992. def __init__(self, *args, **kwargs):
  1993. super().__init__(*args, **kwargs)
  1994. if self.instance.pk:
  1995. # Compile list of choices for primary IPv4 and IPv6 addresses
  1996. for family in [4, 6]:
  1997. ip_choices = [(None, '---------')]
  1998. # Gather PKs of all interfaces belonging to this Device or a peer VirtualChassis member
  1999. interface_ids = self.instance.vc_interfaces(if_master=False).values_list('pk', flat=True)
  2000. # Collect interface IPs
  2001. interface_ips = IPAddress.objects.filter(
  2002. address__family=family,
  2003. assigned_object_type=ContentType.objects.get_for_model(Interface),
  2004. assigned_object_id__in=interface_ids
  2005. ).prefetch_related('assigned_object')
  2006. if interface_ips:
  2007. ip_list = [(ip.id, f'{ip.address} ({ip.assigned_object})') for ip in interface_ips]
  2008. ip_choices.append(('Interface IPs', ip_list))
  2009. # Collect NAT IPs
  2010. nat_ips = IPAddress.objects.prefetch_related('nat_inside').filter(
  2011. address__family=family,
  2012. nat_inside__assigned_object_type=ContentType.objects.get_for_model(Interface),
  2013. nat_inside__assigned_object_id__in=interface_ids
  2014. ).prefetch_related('assigned_object')
  2015. if nat_ips:
  2016. ip_list = [(ip.id, f'{ip.address} (NAT)') for ip in nat_ips]
  2017. ip_choices.append(('NAT IPs', ip_list))
  2018. self.fields['primary_ip{}'.format(family)].choices = ip_choices
  2019. # If editing an existing device, exclude it from the list of occupied rack units. This ensures that a device
  2020. # can be flipped from one face to another.
  2021. self.fields['position'].widget.add_query_param('exclude', self.instance.pk)
  2022. # Limit platform by manufacturer
  2023. self.fields['platform'].queryset = Platform.objects.filter(
  2024. Q(manufacturer__isnull=True) | Q(manufacturer=self.instance.device_type.manufacturer)
  2025. )
  2026. # Disable rack assignment if this is a child device installed in a parent device
  2027. if self.instance.device_type.is_child_device and hasattr(self.instance, 'parent_bay'):
  2028. self.fields['site'].disabled = True
  2029. self.fields['rack'].disabled = True
  2030. self.initial['site'] = self.instance.parent_bay.device.site_id
  2031. self.initial['rack'] = self.instance.parent_bay.device.rack_id
  2032. else:
  2033. # An object that doesn't exist yet can't have any IPs assigned to it
  2034. self.fields['primary_ip4'].choices = []
  2035. self.fields['primary_ip4'].widget.attrs['readonly'] = True
  2036. self.fields['primary_ip6'].choices = []
  2037. self.fields['primary_ip6'].widget.attrs['readonly'] = True
  2038. # Rack position
  2039. position = self.data.get('position') or self.initial.get('position')
  2040. if position:
  2041. self.fields['position'].widget.choices = [(position, f'U{position}')]
  2042. class BaseDeviceCSVForm(CustomFieldModelCSVForm):
  2043. device_role = CSVModelChoiceField(
  2044. queryset=DeviceRole.objects.all(),
  2045. to_field_name='name',
  2046. help_text='Assigned role'
  2047. )
  2048. tenant = CSVModelChoiceField(
  2049. queryset=Tenant.objects.all(),
  2050. required=False,
  2051. to_field_name='name',
  2052. help_text='Assigned tenant'
  2053. )
  2054. manufacturer = CSVModelChoiceField(
  2055. queryset=Manufacturer.objects.all(),
  2056. to_field_name='name',
  2057. help_text='Device type manufacturer'
  2058. )
  2059. device_type = CSVModelChoiceField(
  2060. queryset=DeviceType.objects.all(),
  2061. to_field_name='model',
  2062. help_text='Device type model'
  2063. )
  2064. platform = CSVModelChoiceField(
  2065. queryset=Platform.objects.all(),
  2066. required=False,
  2067. to_field_name='name',
  2068. help_text='Assigned platform'
  2069. )
  2070. status = CSVChoiceField(
  2071. choices=DeviceStatusChoices,
  2072. help_text='Operational status'
  2073. )
  2074. virtual_chassis = CSVModelChoiceField(
  2075. queryset=VirtualChassis.objects.all(),
  2076. to_field_name='name',
  2077. required=False,
  2078. help_text='Virtual chassis'
  2079. )
  2080. cluster = CSVModelChoiceField(
  2081. queryset=Cluster.objects.all(),
  2082. to_field_name='name',
  2083. required=False,
  2084. help_text='Virtualization cluster'
  2085. )
  2086. class Meta:
  2087. fields = []
  2088. model = Device
  2089. help_texts = {
  2090. 'vc_position': 'Virtual chassis position',
  2091. 'vc_priority': 'Virtual chassis priority',
  2092. }
  2093. def __init__(self, data=None, *args, **kwargs):
  2094. super().__init__(data, *args, **kwargs)
  2095. if data:
  2096. # Limit device type queryset by manufacturer
  2097. params = {f"manufacturer__{self.fields['manufacturer'].to_field_name}": data.get('manufacturer')}
  2098. self.fields['device_type'].queryset = self.fields['device_type'].queryset.filter(**params)
  2099. class DeviceCSVForm(BaseDeviceCSVForm):
  2100. site = CSVModelChoiceField(
  2101. queryset=Site.objects.all(),
  2102. to_field_name='name',
  2103. help_text='Assigned site'
  2104. )
  2105. location = CSVModelChoiceField(
  2106. queryset=Location.objects.all(),
  2107. to_field_name='name',
  2108. required=False,
  2109. help_text="Assigned location (if any)"
  2110. )
  2111. rack = CSVModelChoiceField(
  2112. queryset=Rack.objects.all(),
  2113. to_field_name='name',
  2114. required=False,
  2115. help_text="Assigned rack (if any)"
  2116. )
  2117. face = CSVChoiceField(
  2118. choices=DeviceFaceChoices,
  2119. required=False,
  2120. help_text='Mounted rack face'
  2121. )
  2122. class Meta(BaseDeviceCSVForm.Meta):
  2123. fields = [
  2124. 'name', 'device_role', 'tenant', 'manufacturer', 'device_type', 'platform', 'serial', 'asset_tag', 'status',
  2125. 'site', 'location', 'rack', 'position', 'face', 'virtual_chassis', 'vc_position', 'vc_priority', 'cluster',
  2126. 'comments',
  2127. ]
  2128. def __init__(self, data=None, *args, **kwargs):
  2129. super().__init__(data, *args, **kwargs)
  2130. if data:
  2131. # Limit location queryset by assigned site
  2132. params = {f"site__{self.fields['site'].to_field_name}": data.get('site')}
  2133. self.fields['location'].queryset = self.fields['location'].queryset.filter(**params)
  2134. # Limit rack queryset by assigned site and group
  2135. params = {
  2136. f"site__{self.fields['site'].to_field_name}": data.get('site'),
  2137. f"location__{self.fields['location'].to_field_name}": data.get('location'),
  2138. }
  2139. self.fields['rack'].queryset = self.fields['rack'].queryset.filter(**params)
  2140. class ChildDeviceCSVForm(BaseDeviceCSVForm):
  2141. parent = CSVModelChoiceField(
  2142. queryset=Device.objects.all(),
  2143. to_field_name='name',
  2144. help_text='Parent device'
  2145. )
  2146. device_bay = CSVModelChoiceField(
  2147. queryset=DeviceBay.objects.all(),
  2148. to_field_name='name',
  2149. help_text='Device bay in which this device is installed'
  2150. )
  2151. class Meta(BaseDeviceCSVForm.Meta):
  2152. fields = [
  2153. 'name', 'device_role', 'tenant', 'manufacturer', 'device_type', 'platform', 'serial', 'asset_tag', 'status',
  2154. 'parent', 'device_bay', 'virtual_chassis', 'vc_position', 'vc_priority', 'cluster', 'comments',
  2155. ]
  2156. def __init__(self, data=None, *args, **kwargs):
  2157. super().__init__(data, *args, **kwargs)
  2158. if data:
  2159. # Limit device bay queryset by parent device
  2160. params = {f"device__{self.fields['parent'].to_field_name}": data.get('parent')}
  2161. self.fields['device_bay'].queryset = self.fields['device_bay'].queryset.filter(**params)
  2162. def clean(self):
  2163. super().clean()
  2164. # Set parent_bay reverse relationship
  2165. device_bay = self.cleaned_data.get('device_bay')
  2166. if device_bay:
  2167. self.instance.parent_bay = device_bay
  2168. # Inherit site and rack from parent device
  2169. parent = self.cleaned_data.get('parent')
  2170. if parent:
  2171. self.instance.site = parent.site
  2172. self.instance.rack = parent.rack
  2173. class DeviceBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  2174. pk = forms.ModelMultipleChoiceField(
  2175. queryset=Device.objects.all(),
  2176. widget=forms.MultipleHiddenInput()
  2177. )
  2178. manufacturer = DynamicModelChoiceField(
  2179. queryset=Manufacturer.objects.all(),
  2180. required=False
  2181. )
  2182. device_type = DynamicModelChoiceField(
  2183. queryset=DeviceType.objects.all(),
  2184. required=False,
  2185. query_params={
  2186. 'manufacturer_id': '$manufacturer'
  2187. }
  2188. )
  2189. device_role = DynamicModelChoiceField(
  2190. queryset=DeviceRole.objects.all(),
  2191. required=False
  2192. )
  2193. site = DynamicModelChoiceField(
  2194. queryset=Site.objects.all(),
  2195. required=False
  2196. )
  2197. location = DynamicModelChoiceField(
  2198. queryset=Location.objects.all(),
  2199. required=False,
  2200. query_params={
  2201. 'site_id': '$site'
  2202. }
  2203. )
  2204. tenant = DynamicModelChoiceField(
  2205. queryset=Tenant.objects.all(),
  2206. required=False
  2207. )
  2208. platform = DynamicModelChoiceField(
  2209. queryset=Platform.objects.all(),
  2210. required=False
  2211. )
  2212. status = forms.ChoiceField(
  2213. choices=add_blank_choice(DeviceStatusChoices),
  2214. required=False,
  2215. widget=StaticSelect()
  2216. )
  2217. serial = forms.CharField(
  2218. max_length=50,
  2219. required=False,
  2220. label='Serial Number'
  2221. )
  2222. class Meta:
  2223. nullable_fields = [
  2224. 'tenant', 'platform', 'serial',
  2225. ]
  2226. class DeviceFilterForm(BootstrapMixin, LocalConfigContextFilterForm, TenancyFilterForm, CustomFieldModelFilterForm):
  2227. model = Device
  2228. field_order = [
  2229. 'q', 'region_id', 'site_group_id', 'site_id', 'location_id', 'rack_id', 'status', 'role_id', 'tenant_group_id',
  2230. 'tenant_id', 'manufacturer_id', 'device_type_id', 'asset_tag', 'mac_address', 'has_primary_ip',
  2231. ]
  2232. field_groups = [
  2233. ['q', 'tag'],
  2234. ['region_id', 'site_group_id', 'site_id', 'location_id', 'rack_id'],
  2235. ['status', 'role_id', 'asset_tag', 'mac_address'],
  2236. ['manufacturer_id', 'device_type_id', 'platform_id'],
  2237. ['tenant_group_id', 'tenant_id'],
  2238. [
  2239. 'has_primary_ip', 'virtual_chassis_member', 'console_ports', 'console_server_ports', 'power_ports',
  2240. 'power_outlets', 'interfaces', 'pass_through_ports', 'local_context_data',
  2241. ],
  2242. ]
  2243. q = forms.CharField(
  2244. required=False,
  2245. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  2246. label=_('Search')
  2247. )
  2248. region_id = DynamicModelMultipleChoiceField(
  2249. queryset=Region.objects.all(),
  2250. required=False,
  2251. label=_('Region'),
  2252. fetch_trigger='open'
  2253. )
  2254. site_group_id = DynamicModelMultipleChoiceField(
  2255. queryset=SiteGroup.objects.all(),
  2256. required=False,
  2257. label=_('Site group'),
  2258. fetch_trigger='open'
  2259. )
  2260. site_id = DynamicModelMultipleChoiceField(
  2261. queryset=Site.objects.all(),
  2262. required=False,
  2263. query_params={
  2264. 'region_id': '$region_id',
  2265. 'group_id': '$site_group_id',
  2266. },
  2267. label=_('Site'),
  2268. fetch_trigger='open'
  2269. )
  2270. location_id = DynamicModelMultipleChoiceField(
  2271. queryset=Location.objects.all(),
  2272. required=False,
  2273. null_option='None',
  2274. query_params={
  2275. 'site_id': '$site_id'
  2276. },
  2277. label=_('Location'),
  2278. fetch_trigger='open'
  2279. )
  2280. rack_id = DynamicModelMultipleChoiceField(
  2281. queryset=Rack.objects.all(),
  2282. required=False,
  2283. null_option='None',
  2284. query_params={
  2285. 'site_id': '$site_id',
  2286. 'location_id': '$location_id',
  2287. },
  2288. label=_('Rack'),
  2289. fetch_trigger='open'
  2290. )
  2291. role_id = DynamicModelMultipleChoiceField(
  2292. queryset=DeviceRole.objects.all(),
  2293. required=False,
  2294. label=_('Role'),
  2295. fetch_trigger='open'
  2296. )
  2297. manufacturer_id = DynamicModelMultipleChoiceField(
  2298. queryset=Manufacturer.objects.all(),
  2299. required=False,
  2300. label=_('Manufacturer'),
  2301. fetch_trigger='open'
  2302. )
  2303. device_type_id = DynamicModelMultipleChoiceField(
  2304. queryset=DeviceType.objects.all(),
  2305. required=False,
  2306. query_params={
  2307. 'manufacturer_id': '$manufacturer_id'
  2308. },
  2309. label=_('Model'),
  2310. fetch_trigger='open'
  2311. )
  2312. platform_id = DynamicModelMultipleChoiceField(
  2313. queryset=Platform.objects.all(),
  2314. required=False,
  2315. null_option='None',
  2316. label=_('Platform'),
  2317. fetch_trigger='open'
  2318. )
  2319. status = forms.MultipleChoiceField(
  2320. choices=DeviceStatusChoices,
  2321. required=False,
  2322. widget=StaticSelectMultiple()
  2323. )
  2324. asset_tag = forms.CharField(
  2325. required=False
  2326. )
  2327. mac_address = forms.CharField(
  2328. required=False,
  2329. label='MAC address'
  2330. )
  2331. has_primary_ip = forms.NullBooleanField(
  2332. required=False,
  2333. label='Has a primary IP',
  2334. widget=StaticSelect(
  2335. choices=BOOLEAN_WITH_BLANK_CHOICES
  2336. )
  2337. )
  2338. virtual_chassis_member = forms.NullBooleanField(
  2339. required=False,
  2340. label='Virtual chassis member',
  2341. widget=StaticSelect(
  2342. choices=BOOLEAN_WITH_BLANK_CHOICES
  2343. )
  2344. )
  2345. console_ports = forms.NullBooleanField(
  2346. required=False,
  2347. label='Has console ports',
  2348. widget=StaticSelect(
  2349. choices=BOOLEAN_WITH_BLANK_CHOICES
  2350. )
  2351. )
  2352. console_server_ports = forms.NullBooleanField(
  2353. required=False,
  2354. label='Has console server ports',
  2355. widget=StaticSelect(
  2356. choices=BOOLEAN_WITH_BLANK_CHOICES
  2357. )
  2358. )
  2359. power_ports = forms.NullBooleanField(
  2360. required=False,
  2361. label='Has power ports',
  2362. widget=StaticSelect(
  2363. choices=BOOLEAN_WITH_BLANK_CHOICES
  2364. )
  2365. )
  2366. power_outlets = forms.NullBooleanField(
  2367. required=False,
  2368. label='Has power outlets',
  2369. widget=StaticSelect(
  2370. choices=BOOLEAN_WITH_BLANK_CHOICES
  2371. )
  2372. )
  2373. interfaces = forms.NullBooleanField(
  2374. required=False,
  2375. label='Has interfaces',
  2376. widget=StaticSelect(
  2377. choices=BOOLEAN_WITH_BLANK_CHOICES
  2378. )
  2379. )
  2380. pass_through_ports = forms.NullBooleanField(
  2381. required=False,
  2382. label='Has pass-through ports',
  2383. widget=StaticSelect(
  2384. choices=BOOLEAN_WITH_BLANK_CHOICES
  2385. )
  2386. )
  2387. tag = TagFilterField(model)
  2388. #
  2389. # Device components
  2390. #
  2391. class ComponentCreateForm(BootstrapMixin, CustomFieldsMixin, ComponentForm):
  2392. """
  2393. Base form for the creation of device components (models subclassed from ComponentModel).
  2394. """
  2395. device = DynamicModelChoiceField(
  2396. queryset=Device.objects.all()
  2397. )
  2398. description = forms.CharField(
  2399. max_length=200,
  2400. required=False
  2401. )
  2402. tags = DynamicModelMultipleChoiceField(
  2403. queryset=Tag.objects.all(),
  2404. required=False
  2405. )
  2406. class DeviceBulkAddComponentForm(BootstrapMixin, CustomFieldsMixin, ComponentForm):
  2407. pk = forms.ModelMultipleChoiceField(
  2408. queryset=Device.objects.all(),
  2409. widget=forms.MultipleHiddenInput()
  2410. )
  2411. description = forms.CharField(
  2412. max_length=100,
  2413. required=False
  2414. )
  2415. tags = DynamicModelMultipleChoiceField(
  2416. queryset=Tag.objects.all(),
  2417. required=False
  2418. )
  2419. #
  2420. # Console ports
  2421. #
  2422. class ConsolePortFilterForm(DeviceComponentFilterForm):
  2423. model = ConsolePort
  2424. field_groups = [
  2425. ['q', 'tag'],
  2426. ['name', 'label', 'type', 'speed'],
  2427. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  2428. ]
  2429. type = forms.MultipleChoiceField(
  2430. choices=ConsolePortTypeChoices,
  2431. required=False,
  2432. widget=StaticSelectMultiple()
  2433. )
  2434. speed = forms.MultipleChoiceField(
  2435. choices=ConsolePortSpeedChoices,
  2436. required=False,
  2437. widget=StaticSelectMultiple()
  2438. )
  2439. tag = TagFilterField(model)
  2440. class ConsolePortForm(BootstrapMixin, CustomFieldModelForm):
  2441. tags = DynamicModelMultipleChoiceField(
  2442. queryset=Tag.objects.all(),
  2443. required=False
  2444. )
  2445. class Meta:
  2446. model = ConsolePort
  2447. fields = [
  2448. 'device', 'name', 'label', 'type', 'speed', 'mark_connected', 'description', 'tags',
  2449. ]
  2450. widgets = {
  2451. 'device': forms.HiddenInput(),
  2452. }
  2453. class ConsolePortCreateForm(ComponentCreateForm):
  2454. model = ConsolePort
  2455. type = forms.ChoiceField(
  2456. choices=add_blank_choice(ConsolePortTypeChoices),
  2457. required=False,
  2458. widget=StaticSelect()
  2459. )
  2460. speed = forms.ChoiceField(
  2461. choices=add_blank_choice(ConsolePortSpeedChoices),
  2462. required=False,
  2463. widget=StaticSelect()
  2464. )
  2465. field_order = ('device', 'name_pattern', 'label_pattern', 'type', 'speed', 'mark_connected', 'description', 'tags')
  2466. class ConsolePortBulkCreateForm(
  2467. form_from_model(ConsolePort, ['type', 'speed', 'mark_connected']),
  2468. DeviceBulkAddComponentForm
  2469. ):
  2470. model = ConsolePort
  2471. field_order = ('name_pattern', 'label_pattern', 'type', 'mark_connected', 'description', 'tags')
  2472. class ConsolePortBulkEditForm(
  2473. form_from_model(ConsolePort, ['label', 'type', 'speed', 'mark_connected', 'description']),
  2474. BootstrapMixin,
  2475. AddRemoveTagsForm,
  2476. CustomFieldModelBulkEditForm
  2477. ):
  2478. pk = forms.ModelMultipleChoiceField(
  2479. queryset=ConsolePort.objects.all(),
  2480. widget=forms.MultipleHiddenInput()
  2481. )
  2482. mark_connected = forms.NullBooleanField(
  2483. required=False,
  2484. widget=BulkEditNullBooleanSelect
  2485. )
  2486. class Meta:
  2487. nullable_fields = ['label', 'description']
  2488. class ConsolePortCSVForm(CustomFieldModelCSVForm):
  2489. device = CSVModelChoiceField(
  2490. queryset=Device.objects.all(),
  2491. to_field_name='name'
  2492. )
  2493. type = CSVChoiceField(
  2494. choices=ConsolePortTypeChoices,
  2495. required=False,
  2496. help_text='Port type'
  2497. )
  2498. speed = CSVTypedChoiceField(
  2499. choices=ConsolePortSpeedChoices,
  2500. coerce=int,
  2501. empty_value=None,
  2502. required=False,
  2503. help_text='Port speed in bps'
  2504. )
  2505. class Meta:
  2506. model = ConsolePort
  2507. fields = ('device', 'name', 'label', 'type', 'speed', 'mark_connected', 'description')
  2508. #
  2509. # Console server ports
  2510. #
  2511. class ConsoleServerPortFilterForm(DeviceComponentFilterForm):
  2512. model = ConsoleServerPort
  2513. field_groups = [
  2514. ['q', 'tag'],
  2515. ['name', 'label', 'type', 'speed'],
  2516. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  2517. ]
  2518. type = forms.MultipleChoiceField(
  2519. choices=ConsolePortTypeChoices,
  2520. required=False,
  2521. widget=StaticSelectMultiple()
  2522. )
  2523. speed = forms.MultipleChoiceField(
  2524. choices=ConsolePortSpeedChoices,
  2525. required=False,
  2526. widget=StaticSelectMultiple()
  2527. )
  2528. tag = TagFilterField(model)
  2529. class ConsoleServerPortForm(BootstrapMixin, CustomFieldModelForm):
  2530. tags = DynamicModelMultipleChoiceField(
  2531. queryset=Tag.objects.all(),
  2532. required=False
  2533. )
  2534. class Meta:
  2535. model = ConsoleServerPort
  2536. fields = [
  2537. 'device', 'name', 'label', 'type', 'speed', 'mark_connected', 'description', 'tags',
  2538. ]
  2539. widgets = {
  2540. 'device': forms.HiddenInput(),
  2541. }
  2542. class ConsoleServerPortCreateForm(ComponentCreateForm):
  2543. model = ConsoleServerPort
  2544. type = forms.ChoiceField(
  2545. choices=add_blank_choice(ConsolePortTypeChoices),
  2546. required=False,
  2547. widget=StaticSelect()
  2548. )
  2549. speed = forms.ChoiceField(
  2550. choices=add_blank_choice(ConsolePortSpeedChoices),
  2551. required=False,
  2552. widget=StaticSelect()
  2553. )
  2554. field_order = ('device', 'name_pattern', 'label_pattern', 'type', 'speed', 'mark_connected', 'description', 'tags')
  2555. class ConsoleServerPortBulkCreateForm(
  2556. form_from_model(ConsoleServerPort, ['type', 'speed', 'mark_connected']),
  2557. DeviceBulkAddComponentForm
  2558. ):
  2559. model = ConsoleServerPort
  2560. field_order = ('name_pattern', 'label_pattern', 'type', 'speed', 'description', 'tags')
  2561. class ConsoleServerPortBulkEditForm(
  2562. form_from_model(ConsoleServerPort, ['label', 'type', 'speed', 'mark_connected', 'description']),
  2563. BootstrapMixin,
  2564. AddRemoveTagsForm,
  2565. CustomFieldModelBulkEditForm
  2566. ):
  2567. pk = forms.ModelMultipleChoiceField(
  2568. queryset=ConsoleServerPort.objects.all(),
  2569. widget=forms.MultipleHiddenInput()
  2570. )
  2571. mark_connected = forms.NullBooleanField(
  2572. required=False,
  2573. widget=BulkEditNullBooleanSelect
  2574. )
  2575. class Meta:
  2576. nullable_fields = ['label', 'description']
  2577. class ConsoleServerPortCSVForm(CustomFieldModelCSVForm):
  2578. device = CSVModelChoiceField(
  2579. queryset=Device.objects.all(),
  2580. to_field_name='name'
  2581. )
  2582. type = CSVChoiceField(
  2583. choices=ConsolePortTypeChoices,
  2584. required=False,
  2585. help_text='Port type'
  2586. )
  2587. speed = CSVTypedChoiceField(
  2588. choices=ConsolePortSpeedChoices,
  2589. coerce=int,
  2590. empty_value=None,
  2591. required=False,
  2592. help_text='Port speed in bps'
  2593. )
  2594. class Meta:
  2595. model = ConsoleServerPort
  2596. fields = ('device', 'name', 'label', 'type', 'speed', 'mark_connected', 'description')
  2597. #
  2598. # Power ports
  2599. #
  2600. class PowerPortFilterForm(DeviceComponentFilterForm):
  2601. model = PowerPort
  2602. field_groups = [
  2603. ['q', 'tag'],
  2604. ['name', 'label', 'type'],
  2605. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  2606. ]
  2607. type = forms.MultipleChoiceField(
  2608. choices=PowerPortTypeChoices,
  2609. required=False,
  2610. widget=StaticSelectMultiple()
  2611. )
  2612. tag = TagFilterField(model)
  2613. class PowerPortForm(BootstrapMixin, CustomFieldModelForm):
  2614. tags = DynamicModelMultipleChoiceField(
  2615. queryset=Tag.objects.all(),
  2616. required=False
  2617. )
  2618. class Meta:
  2619. model = PowerPort
  2620. fields = [
  2621. 'device', 'name', 'label', 'type', 'maximum_draw', 'allocated_draw', 'mark_connected', 'description',
  2622. 'tags',
  2623. ]
  2624. widgets = {
  2625. 'device': forms.HiddenInput(),
  2626. }
  2627. class PowerPortCreateForm(ComponentCreateForm):
  2628. model = PowerPort
  2629. type = forms.ChoiceField(
  2630. choices=add_blank_choice(PowerPortTypeChoices),
  2631. required=False,
  2632. widget=StaticSelect()
  2633. )
  2634. maximum_draw = forms.IntegerField(
  2635. min_value=1,
  2636. required=False,
  2637. help_text="Maximum draw in watts"
  2638. )
  2639. allocated_draw = forms.IntegerField(
  2640. min_value=1,
  2641. required=False,
  2642. help_text="Allocated draw in watts"
  2643. )
  2644. field_order = (
  2645. 'device', 'name_pattern', 'label_pattern', 'type', 'maximum_draw', 'allocated_draw', 'mark_connected',
  2646. 'description', 'tags',
  2647. )
  2648. class PowerPortBulkCreateForm(
  2649. form_from_model(PowerPort, ['type', 'maximum_draw', 'allocated_draw', 'mark_connected']),
  2650. DeviceBulkAddComponentForm
  2651. ):
  2652. model = PowerPort
  2653. field_order = ('name_pattern', 'label_pattern', 'type', 'maximum_draw', 'allocated_draw', 'description', 'tags')
  2654. class PowerPortBulkEditForm(
  2655. form_from_model(PowerPort, ['label', 'type', 'maximum_draw', 'allocated_draw', 'mark_connected', 'description']),
  2656. BootstrapMixin,
  2657. AddRemoveTagsForm,
  2658. CustomFieldModelBulkEditForm
  2659. ):
  2660. pk = forms.ModelMultipleChoiceField(
  2661. queryset=PowerPort.objects.all(),
  2662. widget=forms.MultipleHiddenInput()
  2663. )
  2664. mark_connected = forms.NullBooleanField(
  2665. required=False,
  2666. widget=BulkEditNullBooleanSelect
  2667. )
  2668. class Meta:
  2669. nullable_fields = ['label', 'description']
  2670. class PowerPortCSVForm(CustomFieldModelCSVForm):
  2671. device = CSVModelChoiceField(
  2672. queryset=Device.objects.all(),
  2673. to_field_name='name'
  2674. )
  2675. type = CSVChoiceField(
  2676. choices=PowerPortTypeChoices,
  2677. required=False,
  2678. help_text='Port type'
  2679. )
  2680. class Meta:
  2681. model = PowerPort
  2682. fields = (
  2683. 'device', 'name', 'label', 'type', 'mark_connected', 'maximum_draw', 'allocated_draw', 'description',
  2684. )
  2685. #
  2686. # Power outlets
  2687. #
  2688. class PowerOutletFilterForm(DeviceComponentFilterForm):
  2689. model = PowerOutlet
  2690. field_groups = [
  2691. ['q', 'tag'],
  2692. ['name', 'label', 'type'],
  2693. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  2694. ]
  2695. type = forms.MultipleChoiceField(
  2696. choices=PowerOutletTypeChoices,
  2697. required=False,
  2698. widget=StaticSelectMultiple()
  2699. )
  2700. tag = TagFilterField(model)
  2701. class PowerOutletForm(BootstrapMixin, CustomFieldModelForm):
  2702. power_port = forms.ModelChoiceField(
  2703. queryset=PowerPort.objects.all(),
  2704. required=False
  2705. )
  2706. tags = DynamicModelMultipleChoiceField(
  2707. queryset=Tag.objects.all(),
  2708. required=False
  2709. )
  2710. class Meta:
  2711. model = PowerOutlet
  2712. fields = [
  2713. 'device', 'name', 'label', 'type', 'power_port', 'feed_leg', 'mark_connected', 'description', 'tags',
  2714. ]
  2715. widgets = {
  2716. 'device': forms.HiddenInput(),
  2717. }
  2718. def __init__(self, *args, **kwargs):
  2719. super().__init__(*args, **kwargs)
  2720. # Limit power_port choices to the local device
  2721. if hasattr(self.instance, 'device'):
  2722. self.fields['power_port'].queryset = PowerPort.objects.filter(
  2723. device=self.instance.device
  2724. )
  2725. class PowerOutletCreateForm(ComponentCreateForm):
  2726. model = PowerOutlet
  2727. type = forms.ChoiceField(
  2728. choices=add_blank_choice(PowerOutletTypeChoices),
  2729. required=False,
  2730. widget=StaticSelect()
  2731. )
  2732. power_port = forms.ModelChoiceField(
  2733. queryset=PowerPort.objects.all(),
  2734. required=False
  2735. )
  2736. feed_leg = forms.ChoiceField(
  2737. choices=add_blank_choice(PowerOutletFeedLegChoices),
  2738. required=False
  2739. )
  2740. field_order = (
  2741. 'device', 'name_pattern', 'label_pattern', 'type', 'power_port', 'feed_leg', 'mark_connected', 'description',
  2742. 'tags',
  2743. )
  2744. def __init__(self, *args, **kwargs):
  2745. super().__init__(*args, **kwargs)
  2746. # Limit power_port queryset to PowerPorts which belong to the parent Device
  2747. device = Device.objects.get(
  2748. pk=self.initial.get('device') or self.data.get('device')
  2749. )
  2750. self.fields['power_port'].queryset = PowerPort.objects.filter(device=device)
  2751. class PowerOutletBulkCreateForm(
  2752. form_from_model(PowerOutlet, ['type', 'feed_leg', 'mark_connected']),
  2753. DeviceBulkAddComponentForm
  2754. ):
  2755. model = PowerOutlet
  2756. field_order = ('name_pattern', 'label_pattern', 'type', 'feed_leg', 'description', 'tags')
  2757. class PowerOutletBulkEditForm(
  2758. form_from_model(PowerOutlet, ['label', 'type', 'feed_leg', 'power_port', 'mark_connected', 'description']),
  2759. BootstrapMixin,
  2760. AddRemoveTagsForm,
  2761. CustomFieldModelBulkEditForm
  2762. ):
  2763. pk = forms.ModelMultipleChoiceField(
  2764. queryset=PowerOutlet.objects.all(),
  2765. widget=forms.MultipleHiddenInput()
  2766. )
  2767. device = forms.ModelChoiceField(
  2768. queryset=Device.objects.all(),
  2769. required=False,
  2770. disabled=True,
  2771. widget=forms.HiddenInput()
  2772. )
  2773. mark_connected = forms.NullBooleanField(
  2774. required=False,
  2775. widget=BulkEditNullBooleanSelect
  2776. )
  2777. class Meta:
  2778. nullable_fields = ['label', 'type', 'feed_leg', 'power_port', 'description']
  2779. def __init__(self, *args, **kwargs):
  2780. super().__init__(*args, **kwargs)
  2781. # Limit power_port queryset to PowerPorts which belong to the parent Device
  2782. if 'device' in self.initial:
  2783. device = Device.objects.filter(pk=self.initial['device']).first()
  2784. self.fields['power_port'].queryset = PowerPort.objects.filter(device=device)
  2785. else:
  2786. self.fields['power_port'].choices = ()
  2787. self.fields['power_port'].widget.attrs['disabled'] = True
  2788. class PowerOutletCSVForm(CustomFieldModelCSVForm):
  2789. device = CSVModelChoiceField(
  2790. queryset=Device.objects.all(),
  2791. to_field_name='name'
  2792. )
  2793. type = CSVChoiceField(
  2794. choices=PowerOutletTypeChoices,
  2795. required=False,
  2796. help_text='Outlet type'
  2797. )
  2798. power_port = CSVModelChoiceField(
  2799. queryset=PowerPort.objects.all(),
  2800. required=False,
  2801. to_field_name='name',
  2802. help_text='Local power port which feeds this outlet'
  2803. )
  2804. feed_leg = CSVChoiceField(
  2805. choices=PowerOutletFeedLegChoices,
  2806. required=False,
  2807. help_text='Electrical phase (for three-phase circuits)'
  2808. )
  2809. class Meta:
  2810. model = PowerOutlet
  2811. fields = ('device', 'name', 'label', 'type', 'mark_connected', 'power_port', 'feed_leg', 'description')
  2812. def __init__(self, *args, **kwargs):
  2813. super().__init__(*args, **kwargs)
  2814. # Limit PowerPort choices to those belonging to this device (or VC master)
  2815. if self.is_bound:
  2816. try:
  2817. device = self.fields['device'].to_python(self.data['device'])
  2818. except forms.ValidationError:
  2819. device = None
  2820. else:
  2821. try:
  2822. device = self.instance.device
  2823. except Device.DoesNotExist:
  2824. device = None
  2825. if device:
  2826. self.fields['power_port'].queryset = PowerPort.objects.filter(
  2827. device__in=[device, device.get_vc_master()]
  2828. )
  2829. else:
  2830. self.fields['power_port'].queryset = PowerPort.objects.none()
  2831. #
  2832. # Interfaces
  2833. #
  2834. class InterfaceFilterForm(DeviceComponentFilterForm):
  2835. model = Interface
  2836. field_groups = [
  2837. ['q', 'tag'],
  2838. ['name', 'label', 'type', 'enabled', 'mgmt_only', 'mac_address'],
  2839. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  2840. ]
  2841. type = forms.MultipleChoiceField(
  2842. choices=InterfaceTypeChoices,
  2843. required=False,
  2844. widget=StaticSelectMultiple()
  2845. )
  2846. enabled = forms.NullBooleanField(
  2847. required=False,
  2848. widget=StaticSelect(
  2849. choices=BOOLEAN_WITH_BLANK_CHOICES
  2850. )
  2851. )
  2852. mgmt_only = forms.NullBooleanField(
  2853. required=False,
  2854. widget=StaticSelect(
  2855. choices=BOOLEAN_WITH_BLANK_CHOICES
  2856. )
  2857. )
  2858. mac_address = forms.CharField(
  2859. required=False,
  2860. label='MAC address'
  2861. )
  2862. tag = TagFilterField(model)
  2863. class InterfaceForm(BootstrapMixin, InterfaceCommonForm, CustomFieldModelForm):
  2864. parent = DynamicModelChoiceField(
  2865. queryset=Interface.objects.all(),
  2866. required=False,
  2867. label='Parent interface'
  2868. )
  2869. lag = DynamicModelChoiceField(
  2870. queryset=Interface.objects.all(),
  2871. required=False,
  2872. label='LAG interface',
  2873. query_params={
  2874. 'type': 'lag',
  2875. }
  2876. )
  2877. vlan_group = DynamicModelChoiceField(
  2878. queryset=VLANGroup.objects.all(),
  2879. required=False,
  2880. label='VLAN group'
  2881. )
  2882. untagged_vlan = DynamicModelChoiceField(
  2883. queryset=VLAN.objects.all(),
  2884. required=False,
  2885. label='Untagged VLAN',
  2886. query_params={
  2887. 'group_id': '$vlan_group',
  2888. }
  2889. )
  2890. tagged_vlans = DynamicModelMultipleChoiceField(
  2891. queryset=VLAN.objects.all(),
  2892. required=False,
  2893. label='Tagged VLANs',
  2894. query_params={
  2895. 'group_id': '$vlan_group',
  2896. }
  2897. )
  2898. tags = DynamicModelMultipleChoiceField(
  2899. queryset=Tag.objects.all(),
  2900. required=False
  2901. )
  2902. class Meta:
  2903. model = Interface
  2904. fields = [
  2905. 'device', 'name', 'label', 'type', 'enabled', 'parent', 'lag', 'mac_address', 'mtu', 'mgmt_only',
  2906. 'mark_connected', 'description', 'mode', 'untagged_vlan', 'tagged_vlans', 'tags',
  2907. ]
  2908. widgets = {
  2909. 'device': forms.HiddenInput(),
  2910. 'type': StaticSelect(),
  2911. 'mode': StaticSelect(),
  2912. }
  2913. labels = {
  2914. 'mode': '802.1Q Mode',
  2915. }
  2916. help_texts = {
  2917. 'mode': INTERFACE_MODE_HELP_TEXT,
  2918. }
  2919. def __init__(self, *args, **kwargs):
  2920. super().__init__(*args, **kwargs)
  2921. device = Device.objects.get(pk=self.data['device']) if self.is_bound else self.instance.device
  2922. # Restrict parent/LAG interface assignment by device/VC
  2923. self.fields['parent'].widget.add_query_param('device_id', device.pk)
  2924. if device.virtual_chassis and device.virtual_chassis.master:
  2925. # Get available LAG interfaces by VirtualChassis master
  2926. self.fields['lag'].widget.add_query_param('device_id', device.virtual_chassis.master.pk)
  2927. else:
  2928. self.fields['lag'].widget.add_query_param('device_id', device.pk)
  2929. # Limit VLAN choices by device
  2930. self.fields['untagged_vlan'].widget.add_query_param('available_on_device', device.pk)
  2931. self.fields['tagged_vlans'].widget.add_query_param('available_on_device', device.pk)
  2932. class InterfaceCreateForm(ComponentCreateForm, InterfaceCommonForm):
  2933. model = Interface
  2934. type = forms.ChoiceField(
  2935. choices=InterfaceTypeChoices,
  2936. widget=StaticSelect(),
  2937. )
  2938. enabled = forms.BooleanField(
  2939. required=False,
  2940. initial=True
  2941. )
  2942. parent = DynamicModelChoiceField(
  2943. queryset=Interface.objects.all(),
  2944. required=False,
  2945. query_params={
  2946. 'device_id': '$device',
  2947. }
  2948. )
  2949. lag = DynamicModelChoiceField(
  2950. queryset=Interface.objects.all(),
  2951. required=False,
  2952. query_params={
  2953. 'device_id': '$device',
  2954. 'type': 'lag',
  2955. }
  2956. )
  2957. mac_address = forms.CharField(
  2958. required=False,
  2959. label='MAC Address'
  2960. )
  2961. mgmt_only = forms.BooleanField(
  2962. required=False,
  2963. label='Management only',
  2964. help_text='This interface is used only for out-of-band management'
  2965. )
  2966. mode = forms.ChoiceField(
  2967. choices=add_blank_choice(InterfaceModeChoices),
  2968. required=False,
  2969. widget=StaticSelect(),
  2970. )
  2971. untagged_vlan = DynamicModelChoiceField(
  2972. queryset=VLAN.objects.all(),
  2973. required=False
  2974. )
  2975. tagged_vlans = DynamicModelMultipleChoiceField(
  2976. queryset=VLAN.objects.all(),
  2977. required=False
  2978. )
  2979. field_order = (
  2980. 'device', 'name_pattern', 'label_pattern', 'type', 'enabled', 'parent', 'lag', 'mtu', 'mac_address',
  2981. 'description', 'mgmt_only', 'mark_connected', 'mode', 'untagged_vlan', 'tagged_vlans', 'tags'
  2982. )
  2983. def __init__(self, *args, **kwargs):
  2984. super().__init__(*args, **kwargs)
  2985. # Limit VLAN choices by device
  2986. device_id = self.initial.get('device') or self.data.get('device')
  2987. self.fields['untagged_vlan'].widget.add_query_param('available_on_device', device_id)
  2988. self.fields['tagged_vlans'].widget.add_query_param('available_on_device', device_id)
  2989. class InterfaceBulkCreateForm(
  2990. form_from_model(Interface, ['type', 'enabled', 'mtu', 'mgmt_only', 'mark_connected']),
  2991. DeviceBulkAddComponentForm
  2992. ):
  2993. model = Interface
  2994. field_order = (
  2995. 'name_pattern', 'label_pattern', 'type', 'enabled', 'mtu', 'mgmt_only', 'mark_connected', 'description', 'tags',
  2996. )
  2997. class InterfaceBulkEditForm(
  2998. form_from_model(Interface, [
  2999. 'label', 'type', 'parent', 'lag', 'mac_address', 'mtu', 'mgmt_only', 'mark_connected', 'description', 'mode',
  3000. ]),
  3001. BootstrapMixin,
  3002. AddRemoveTagsForm,
  3003. CustomFieldModelBulkEditForm
  3004. ):
  3005. pk = forms.ModelMultipleChoiceField(
  3006. queryset=Interface.objects.all(),
  3007. widget=forms.MultipleHiddenInput()
  3008. )
  3009. device = forms.ModelChoiceField(
  3010. queryset=Device.objects.all(),
  3011. required=False,
  3012. disabled=True,
  3013. widget=forms.HiddenInput()
  3014. )
  3015. enabled = forms.NullBooleanField(
  3016. required=False,
  3017. widget=BulkEditNullBooleanSelect
  3018. )
  3019. parent = DynamicModelChoiceField(
  3020. queryset=Interface.objects.all(),
  3021. required=False
  3022. )
  3023. lag = DynamicModelChoiceField(
  3024. queryset=Interface.objects.all(),
  3025. required=False,
  3026. query_params={
  3027. 'type': 'lag',
  3028. }
  3029. )
  3030. mgmt_only = forms.NullBooleanField(
  3031. required=False,
  3032. widget=BulkEditNullBooleanSelect,
  3033. label='Management only'
  3034. )
  3035. mark_connected = forms.NullBooleanField(
  3036. required=False,
  3037. widget=BulkEditNullBooleanSelect
  3038. )
  3039. untagged_vlan = DynamicModelChoiceField(
  3040. queryset=VLAN.objects.all(),
  3041. required=False
  3042. )
  3043. tagged_vlans = DynamicModelMultipleChoiceField(
  3044. queryset=VLAN.objects.all(),
  3045. required=False
  3046. )
  3047. class Meta:
  3048. nullable_fields = [
  3049. 'label', 'parent', 'lag', 'mac_address', 'mtu', 'description', 'mode', 'untagged_vlan', 'tagged_vlans'
  3050. ]
  3051. def __init__(self, *args, **kwargs):
  3052. super().__init__(*args, **kwargs)
  3053. if 'device' in self.initial:
  3054. device = Device.objects.filter(pk=self.initial['device']).first()
  3055. # Restrict parent/LAG interface assignment by device
  3056. self.fields['parent'].widget.add_query_param('device_id', device.pk)
  3057. self.fields['lag'].widget.add_query_param('device_id', device.pk)
  3058. # Limit VLAN choices by device
  3059. self.fields['untagged_vlan'].widget.add_query_param('available_on_device', device.pk)
  3060. self.fields['tagged_vlans'].widget.add_query_param('available_on_device', device.pk)
  3061. else:
  3062. # See #4523
  3063. if 'pk' in self.initial:
  3064. site = None
  3065. interfaces = Interface.objects.filter(pk__in=self.initial['pk']).prefetch_related('device__site')
  3066. # Check interface sites. First interface should set site, further interfaces will either continue the
  3067. # loop or reset back to no site and break the loop.
  3068. for interface in interfaces:
  3069. if site is None:
  3070. site = interface.device.site
  3071. elif interface.device.site is not site:
  3072. site = None
  3073. break
  3074. if site is not None:
  3075. self.fields['untagged_vlan'].widget.add_query_param('site_id', site.pk)
  3076. self.fields['tagged_vlans'].widget.add_query_param('site_id', site.pk)
  3077. self.fields['parent'].choices = ()
  3078. self.fields['parent'].widget.attrs['disabled'] = True
  3079. self.fields['lag'].choices = ()
  3080. self.fields['lag'].widget.attrs['disabled'] = True
  3081. def clean(self):
  3082. super().clean()
  3083. # Untagged interfaces cannot be assigned tagged VLANs
  3084. if self.cleaned_data['mode'] == InterfaceModeChoices.MODE_ACCESS and self.cleaned_data['tagged_vlans']:
  3085. raise forms.ValidationError({
  3086. 'mode': "An access interface cannot have tagged VLANs assigned."
  3087. })
  3088. # Remove all tagged VLAN assignments from "tagged all" interfaces
  3089. elif self.cleaned_data['mode'] == InterfaceModeChoices.MODE_TAGGED_ALL:
  3090. self.cleaned_data['tagged_vlans'] = []
  3091. class InterfaceCSVForm(CustomFieldModelCSVForm):
  3092. device = CSVModelChoiceField(
  3093. queryset=Device.objects.all(),
  3094. to_field_name='name'
  3095. )
  3096. parent = CSVModelChoiceField(
  3097. queryset=Interface.objects.all(),
  3098. required=False,
  3099. to_field_name='name',
  3100. help_text='Parent interface'
  3101. )
  3102. lag = CSVModelChoiceField(
  3103. queryset=Interface.objects.all(),
  3104. required=False,
  3105. to_field_name='name',
  3106. help_text='Parent LAG interface'
  3107. )
  3108. type = CSVChoiceField(
  3109. choices=InterfaceTypeChoices,
  3110. help_text='Physical medium'
  3111. )
  3112. mode = CSVChoiceField(
  3113. choices=InterfaceModeChoices,
  3114. required=False,
  3115. help_text='IEEE 802.1Q operational mode (for L2 interfaces)'
  3116. )
  3117. class Meta:
  3118. model = Interface
  3119. fields = (
  3120. 'device', 'name', 'label', 'parent', 'lag', 'type', 'enabled', 'mark_connected', 'mac_address', 'mtu',
  3121. 'mgmt_only', 'description', 'mode',
  3122. )
  3123. def __init__(self, *args, **kwargs):
  3124. super().__init__(*args, **kwargs)
  3125. # Limit LAG choices to interfaces belonging to this device (or virtual chassis)
  3126. device = None
  3127. if self.is_bound and 'device' in self.data:
  3128. try:
  3129. device = self.fields['device'].to_python(self.data['device'])
  3130. except forms.ValidationError:
  3131. pass
  3132. if device and device.virtual_chassis:
  3133. self.fields['lag'].queryset = Interface.objects.filter(
  3134. Q(device=device) | Q(device__virtual_chassis=device.virtual_chassis),
  3135. type=InterfaceTypeChoices.TYPE_LAG
  3136. )
  3137. self.fields['parent'].queryset = Interface.objects.filter(
  3138. Q(device=device) | Q(device__virtual_chassis=device.virtual_chassis)
  3139. )
  3140. elif device:
  3141. self.fields['lag'].queryset = Interface.objects.filter(
  3142. device=device,
  3143. type=InterfaceTypeChoices.TYPE_LAG
  3144. )
  3145. self.fields['parent'].queryset = Interface.objects.filter(device=device)
  3146. else:
  3147. self.fields['lag'].queryset = Interface.objects.none()
  3148. self.fields['parent'].queryset = Interface.objects.none()
  3149. def clean_enabled(self):
  3150. # Make sure enabled is True when it's not included in the uploaded data
  3151. if 'enabled' not in self.data:
  3152. return True
  3153. else:
  3154. return self.cleaned_data['enabled']
  3155. #
  3156. # Front pass-through ports
  3157. #
  3158. class FrontPortFilterForm(DeviceComponentFilterForm):
  3159. field_groups = [
  3160. ['q', 'tag'],
  3161. ['name', 'label', 'type', 'color'],
  3162. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  3163. ]
  3164. model = FrontPort
  3165. type = forms.MultipleChoiceField(
  3166. choices=PortTypeChoices,
  3167. required=False,
  3168. widget=StaticSelectMultiple()
  3169. )
  3170. color = ColorField(
  3171. required=False
  3172. )
  3173. tag = TagFilterField(model)
  3174. class FrontPortForm(BootstrapMixin, CustomFieldModelForm):
  3175. tags = DynamicModelMultipleChoiceField(
  3176. queryset=Tag.objects.all(),
  3177. required=False
  3178. )
  3179. class Meta:
  3180. model = FrontPort
  3181. fields = [
  3182. 'device', 'name', 'label', 'type', 'color', 'rear_port', 'rear_port_position', 'mark_connected',
  3183. 'description', 'tags',
  3184. ]
  3185. widgets = {
  3186. 'device': forms.HiddenInput(),
  3187. 'type': StaticSelect(),
  3188. 'rear_port': StaticSelect(),
  3189. }
  3190. def __init__(self, *args, **kwargs):
  3191. super().__init__(*args, **kwargs)
  3192. # Limit RearPort choices to the local device
  3193. if hasattr(self.instance, 'device'):
  3194. self.fields['rear_port'].queryset = self.fields['rear_port'].queryset.filter(
  3195. device=self.instance.device
  3196. )
  3197. # TODO: Merge with FrontPortTemplateCreateForm to remove duplicate logic
  3198. class FrontPortCreateForm(ComponentCreateForm):
  3199. model = FrontPort
  3200. type = forms.ChoiceField(
  3201. choices=PortTypeChoices,
  3202. widget=StaticSelect(),
  3203. )
  3204. color = ColorField(
  3205. required=False
  3206. )
  3207. rear_port_set = forms.MultipleChoiceField(
  3208. choices=[],
  3209. label='Rear ports',
  3210. help_text='Select one rear port assignment for each front port being created.',
  3211. )
  3212. field_order = (
  3213. 'device', 'name_pattern', 'label_pattern', 'type', 'color', 'rear_port_set', 'mark_connected', 'description',
  3214. 'tags',
  3215. )
  3216. def __init__(self, *args, **kwargs):
  3217. super().__init__(*args, **kwargs)
  3218. device = Device.objects.get(
  3219. pk=self.initial.get('device') or self.data.get('device')
  3220. )
  3221. # Determine which rear port positions are occupied. These will be excluded from the list of available
  3222. # mappings.
  3223. occupied_port_positions = [
  3224. (front_port.rear_port_id, front_port.rear_port_position)
  3225. for front_port in device.frontports.all()
  3226. ]
  3227. # Populate rear port choices
  3228. choices = []
  3229. rear_ports = RearPort.objects.filter(device=device)
  3230. for rear_port in rear_ports:
  3231. for i in range(1, rear_port.positions + 1):
  3232. if (rear_port.pk, i) not in occupied_port_positions:
  3233. choices.append(
  3234. ('{}:{}'.format(rear_port.pk, i), '{}:{}'.format(rear_port.name, i))
  3235. )
  3236. self.fields['rear_port_set'].choices = choices
  3237. def clean(self):
  3238. super().clean()
  3239. # Validate that the number of ports being created equals the number of selected (rear port, position) tuples
  3240. front_port_count = len(self.cleaned_data['name_pattern'])
  3241. rear_port_count = len(self.cleaned_data['rear_port_set'])
  3242. if front_port_count != rear_port_count:
  3243. raise forms.ValidationError({
  3244. 'rear_port_set': 'The provided name pattern will create {} ports, however {} rear port assignments '
  3245. 'were selected. These counts must match.'.format(front_port_count, rear_port_count)
  3246. })
  3247. def get_iterative_data(self, iteration):
  3248. # Assign rear port and position from selected set
  3249. rear_port, position = self.cleaned_data['rear_port_set'][iteration].split(':')
  3250. return {
  3251. 'rear_port': int(rear_port),
  3252. 'rear_port_position': int(position),
  3253. }
  3254. # class FrontPortBulkCreateForm(
  3255. # form_from_model(FrontPort, ['label', 'type', 'description', 'tags']),
  3256. # DeviceBulkAddComponentForm
  3257. # ):
  3258. # pass
  3259. class FrontPortBulkEditForm(
  3260. form_from_model(FrontPort, ['label', 'type', 'color', 'mark_connected', 'description']),
  3261. BootstrapMixin,
  3262. AddRemoveTagsForm,
  3263. CustomFieldModelBulkEditForm
  3264. ):
  3265. pk = forms.ModelMultipleChoiceField(
  3266. queryset=FrontPort.objects.all(),
  3267. widget=forms.MultipleHiddenInput()
  3268. )
  3269. class Meta:
  3270. nullable_fields = ['label', 'description']
  3271. class FrontPortCSVForm(CustomFieldModelCSVForm):
  3272. device = CSVModelChoiceField(
  3273. queryset=Device.objects.all(),
  3274. to_field_name='name'
  3275. )
  3276. rear_port = CSVModelChoiceField(
  3277. queryset=RearPort.objects.all(),
  3278. to_field_name='name',
  3279. help_text='Corresponding rear port'
  3280. )
  3281. type = CSVChoiceField(
  3282. choices=PortTypeChoices,
  3283. help_text='Physical medium classification'
  3284. )
  3285. class Meta:
  3286. model = FrontPort
  3287. fields = (
  3288. 'device', 'name', 'label', 'type', 'color', 'mark_connected', 'rear_port', 'rear_port_position',
  3289. 'description',
  3290. )
  3291. help_texts = {
  3292. 'rear_port_position': 'Mapped position on corresponding rear port',
  3293. }
  3294. def __init__(self, *args, **kwargs):
  3295. super().__init__(*args, **kwargs)
  3296. # Limit RearPort choices to those belonging to this device (or VC master)
  3297. if self.is_bound:
  3298. try:
  3299. device = self.fields['device'].to_python(self.data['device'])
  3300. except forms.ValidationError:
  3301. device = None
  3302. else:
  3303. try:
  3304. device = self.instance.device
  3305. except Device.DoesNotExist:
  3306. device = None
  3307. if device:
  3308. self.fields['rear_port'].queryset = RearPort.objects.filter(
  3309. device__in=[device, device.get_vc_master()]
  3310. )
  3311. else:
  3312. self.fields['rear_port'].queryset = RearPort.objects.none()
  3313. #
  3314. # Rear pass-through ports
  3315. #
  3316. class RearPortFilterForm(DeviceComponentFilterForm):
  3317. model = RearPort
  3318. field_groups = [
  3319. ['q', 'tag'],
  3320. ['name', 'label', 'type', 'color'],
  3321. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  3322. ]
  3323. type = forms.MultipleChoiceField(
  3324. choices=PortTypeChoices,
  3325. required=False,
  3326. widget=StaticSelectMultiple()
  3327. )
  3328. color = ColorField(
  3329. required=False
  3330. )
  3331. tag = TagFilterField(model)
  3332. class RearPortForm(BootstrapMixin, CustomFieldModelForm):
  3333. tags = DynamicModelMultipleChoiceField(
  3334. queryset=Tag.objects.all(),
  3335. required=False
  3336. )
  3337. class Meta:
  3338. model = RearPort
  3339. fields = [
  3340. 'device', 'name', 'label', 'type', 'color', 'positions', 'mark_connected', 'description', 'tags',
  3341. ]
  3342. widgets = {
  3343. 'device': forms.HiddenInput(),
  3344. 'type': StaticSelect(),
  3345. }
  3346. class RearPortCreateForm(ComponentCreateForm):
  3347. model = RearPort
  3348. type = forms.ChoiceField(
  3349. choices=PortTypeChoices,
  3350. widget=StaticSelect(),
  3351. )
  3352. color = ColorField(
  3353. required=False
  3354. )
  3355. positions = forms.IntegerField(
  3356. min_value=REARPORT_POSITIONS_MIN,
  3357. max_value=REARPORT_POSITIONS_MAX,
  3358. initial=1,
  3359. help_text='The number of front ports which may be mapped to each rear port'
  3360. )
  3361. field_order = (
  3362. 'device', 'name_pattern', 'label_pattern', 'type', 'color', 'positions', 'mark_connected', 'description',
  3363. 'tags',
  3364. )
  3365. class RearPortBulkCreateForm(
  3366. form_from_model(RearPort, ['type', 'color', 'positions', 'mark_connected']),
  3367. DeviceBulkAddComponentForm
  3368. ):
  3369. model = RearPort
  3370. field_order = ('name_pattern', 'label_pattern', 'type', 'positions', 'mark_connected', 'description', 'tags')
  3371. class RearPortBulkEditForm(
  3372. form_from_model(RearPort, ['label', 'type', 'color', 'mark_connected', 'description']),
  3373. BootstrapMixin,
  3374. AddRemoveTagsForm,
  3375. CustomFieldModelBulkEditForm
  3376. ):
  3377. pk = forms.ModelMultipleChoiceField(
  3378. queryset=RearPort.objects.all(),
  3379. widget=forms.MultipleHiddenInput()
  3380. )
  3381. class Meta:
  3382. nullable_fields = ['label', 'description']
  3383. class RearPortCSVForm(CustomFieldModelCSVForm):
  3384. device = CSVModelChoiceField(
  3385. queryset=Device.objects.all(),
  3386. to_field_name='name'
  3387. )
  3388. type = CSVChoiceField(
  3389. help_text='Physical medium classification',
  3390. choices=PortTypeChoices,
  3391. )
  3392. class Meta:
  3393. model = RearPort
  3394. fields = ('device', 'name', 'label', 'type', 'color', 'mark_connected', 'positions', 'description')
  3395. help_texts = {
  3396. 'positions': 'Number of front ports which may be mapped'
  3397. }
  3398. #
  3399. # Device bays
  3400. #
  3401. class DeviceBayFilterForm(DeviceComponentFilterForm):
  3402. model = DeviceBay
  3403. field_groups = [
  3404. ['q', 'tag'],
  3405. ['name', 'label'],
  3406. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  3407. ]
  3408. tag = TagFilterField(model)
  3409. class DeviceBayForm(BootstrapMixin, CustomFieldModelForm):
  3410. tags = DynamicModelMultipleChoiceField(
  3411. queryset=Tag.objects.all(),
  3412. required=False
  3413. )
  3414. class Meta:
  3415. model = DeviceBay
  3416. fields = [
  3417. 'device', 'name', 'label', 'description', 'tags',
  3418. ]
  3419. widgets = {
  3420. 'device': forms.HiddenInput(),
  3421. }
  3422. class DeviceBayCreateForm(ComponentCreateForm):
  3423. model = DeviceBay
  3424. field_order = ('device', 'name_pattern', 'label_pattern', 'description', 'tags')
  3425. class PopulateDeviceBayForm(BootstrapMixin, forms.Form):
  3426. installed_device = forms.ModelChoiceField(
  3427. queryset=Device.objects.all(),
  3428. label='Child Device',
  3429. help_text="Child devices must first be created and assigned to the site/rack of the parent device.",
  3430. widget=StaticSelect(),
  3431. )
  3432. def __init__(self, device_bay, *args, **kwargs):
  3433. super().__init__(*args, **kwargs)
  3434. self.fields['installed_device'].queryset = Device.objects.filter(
  3435. site=device_bay.device.site,
  3436. rack=device_bay.device.rack,
  3437. parent_bay__isnull=True,
  3438. device_type__u_height=0,
  3439. device_type__subdevice_role=SubdeviceRoleChoices.ROLE_CHILD
  3440. ).exclude(pk=device_bay.device.pk)
  3441. class DeviceBayBulkCreateForm(DeviceBulkAddComponentForm):
  3442. model = DeviceBay
  3443. field_order = ('name_pattern', 'label_pattern', 'description', 'tags')
  3444. class DeviceBayBulkEditForm(
  3445. form_from_model(DeviceBay, ['label', 'description']),
  3446. BootstrapMixin,
  3447. AddRemoveTagsForm,
  3448. CustomFieldModelBulkEditForm
  3449. ):
  3450. pk = forms.ModelMultipleChoiceField(
  3451. queryset=DeviceBay.objects.all(),
  3452. widget=forms.MultipleHiddenInput()
  3453. )
  3454. class Meta:
  3455. nullable_fields = ['label', 'description']
  3456. class DeviceBayCSVForm(CustomFieldModelCSVForm):
  3457. device = CSVModelChoiceField(
  3458. queryset=Device.objects.all(),
  3459. to_field_name='name'
  3460. )
  3461. installed_device = CSVModelChoiceField(
  3462. queryset=Device.objects.all(),
  3463. required=False,
  3464. to_field_name='name',
  3465. help_text='Child device installed within this bay',
  3466. error_messages={
  3467. 'invalid_choice': 'Child device not found.',
  3468. }
  3469. )
  3470. class Meta:
  3471. model = DeviceBay
  3472. fields = ('device', 'name', 'label', 'installed_device', 'description')
  3473. def __init__(self, *args, **kwargs):
  3474. super().__init__(*args, **kwargs)
  3475. # Limit installed device choices to devices of the correct type and location
  3476. if self.is_bound:
  3477. try:
  3478. device = self.fields['device'].to_python(self.data['device'])
  3479. except forms.ValidationError:
  3480. device = None
  3481. else:
  3482. try:
  3483. device = self.instance.device
  3484. except Device.DoesNotExist:
  3485. device = None
  3486. if device:
  3487. self.fields['installed_device'].queryset = Device.objects.filter(
  3488. site=device.site,
  3489. rack=device.rack,
  3490. parent_bay__isnull=True,
  3491. device_type__u_height=0,
  3492. device_type__subdevice_role=SubdeviceRoleChoices.ROLE_CHILD
  3493. ).exclude(pk=device.pk)
  3494. else:
  3495. self.fields['installed_device'].queryset = Interface.objects.none()
  3496. #
  3497. # Inventory items
  3498. #
  3499. class InventoryItemForm(BootstrapMixin, CustomFieldModelForm):
  3500. device = DynamicModelChoiceField(
  3501. queryset=Device.objects.all()
  3502. )
  3503. parent = DynamicModelChoiceField(
  3504. queryset=InventoryItem.objects.all(),
  3505. required=False,
  3506. query_params={
  3507. 'device_id': '$device'
  3508. }
  3509. )
  3510. manufacturer = DynamicModelChoiceField(
  3511. queryset=Manufacturer.objects.all(),
  3512. required=False
  3513. )
  3514. tags = DynamicModelMultipleChoiceField(
  3515. queryset=Tag.objects.all(),
  3516. required=False
  3517. )
  3518. class Meta:
  3519. model = InventoryItem
  3520. fields = [
  3521. 'device', 'parent', 'name', 'label', 'manufacturer', 'part_id', 'serial', 'asset_tag', 'description',
  3522. 'tags',
  3523. ]
  3524. class InventoryItemCreateForm(ComponentCreateForm):
  3525. model = InventoryItem
  3526. manufacturer = DynamicModelChoiceField(
  3527. queryset=Manufacturer.objects.all(),
  3528. required=False
  3529. )
  3530. parent = DynamicModelChoiceField(
  3531. queryset=InventoryItem.objects.all(),
  3532. required=False,
  3533. query_params={
  3534. 'device_id': '$device'
  3535. }
  3536. )
  3537. part_id = forms.CharField(
  3538. max_length=50,
  3539. required=False,
  3540. label='Part ID'
  3541. )
  3542. serial = forms.CharField(
  3543. max_length=50,
  3544. required=False,
  3545. )
  3546. asset_tag = forms.CharField(
  3547. max_length=50,
  3548. required=False,
  3549. )
  3550. field_order = (
  3551. 'device', 'parent', 'name_pattern', 'label_pattern', 'manufacturer', 'part_id', 'serial', 'asset_tag',
  3552. 'description', 'tags',
  3553. )
  3554. class InventoryItemCSVForm(CustomFieldModelCSVForm):
  3555. device = CSVModelChoiceField(
  3556. queryset=Device.objects.all(),
  3557. to_field_name='name'
  3558. )
  3559. manufacturer = CSVModelChoiceField(
  3560. queryset=Manufacturer.objects.all(),
  3561. to_field_name='name',
  3562. required=False
  3563. )
  3564. parent = CSVModelChoiceField(
  3565. queryset=Device.objects.all(),
  3566. to_field_name='name',
  3567. required=False,
  3568. help_text='Parent inventory item'
  3569. )
  3570. class Meta:
  3571. model = InventoryItem
  3572. fields = (
  3573. 'device', 'name', 'label', 'manufacturer', 'part_id', 'serial', 'asset_tag', 'discovered', 'description',
  3574. )
  3575. def __init__(self, *args, **kwargs):
  3576. super().__init__(*args, **kwargs)
  3577. # Limit parent choices to inventory items belonging to this device
  3578. device = None
  3579. if self.is_bound and 'device' in self.data:
  3580. try:
  3581. device = self.fields['device'].to_python(self.data['device'])
  3582. except forms.ValidationError:
  3583. pass
  3584. if device:
  3585. self.fields['parent'].queryset = InventoryItem.objects.filter(device=device)
  3586. else:
  3587. self.fields['parent'].queryset = InventoryItem.objects.none()
  3588. class InventoryItemBulkCreateForm(
  3589. form_from_model(InventoryItem, ['manufacturer', 'part_id', 'serial', 'asset_tag', 'discovered']),
  3590. DeviceBulkAddComponentForm
  3591. ):
  3592. model = InventoryItem
  3593. field_order = (
  3594. 'name_pattern', 'label_pattern', 'manufacturer', 'part_id', 'serial', 'asset_tag', 'discovered', 'description',
  3595. 'tags',
  3596. )
  3597. class InventoryItemBulkEditForm(
  3598. form_from_model(InventoryItem, ['label', 'manufacturer', 'part_id', 'description']),
  3599. BootstrapMixin,
  3600. AddRemoveTagsForm,
  3601. CustomFieldModelBulkEditForm
  3602. ):
  3603. pk = forms.ModelMultipleChoiceField(
  3604. queryset=InventoryItem.objects.all(),
  3605. widget=forms.MultipleHiddenInput()
  3606. )
  3607. manufacturer = DynamicModelChoiceField(
  3608. queryset=Manufacturer.objects.all(),
  3609. required=False
  3610. )
  3611. class Meta:
  3612. nullable_fields = ['label', 'manufacturer', 'part_id', 'description']
  3613. class InventoryItemFilterForm(DeviceComponentFilterForm):
  3614. model = InventoryItem
  3615. field_groups = [
  3616. ['q', 'tag'],
  3617. ['name', 'label', 'manufacturer_id', 'serial', 'asset_tag', 'discovered'],
  3618. ['region_id', 'site_group_id', 'site_id', 'location_id', 'device_id'],
  3619. ]
  3620. manufacturer_id = DynamicModelMultipleChoiceField(
  3621. queryset=Manufacturer.objects.all(),
  3622. required=False,
  3623. label=_('Manufacturer'),
  3624. fetch_trigger='open'
  3625. )
  3626. serial = forms.CharField(
  3627. required=False
  3628. )
  3629. asset_tag = forms.CharField(
  3630. required=False
  3631. )
  3632. discovered = forms.NullBooleanField(
  3633. required=False,
  3634. widget=StaticSelect(
  3635. choices=BOOLEAN_WITH_BLANK_CHOICES
  3636. )
  3637. )
  3638. tag = TagFilterField(model)
  3639. #
  3640. # Cables
  3641. #
  3642. class ConnectCableToDeviceForm(BootstrapMixin, CustomFieldModelForm):
  3643. """
  3644. Base form for connecting a Cable to a Device component
  3645. """
  3646. termination_b_region = DynamicModelChoiceField(
  3647. queryset=Region.objects.all(),
  3648. label='Region',
  3649. required=False
  3650. )
  3651. termination_b_site_group = DynamicModelChoiceField(
  3652. queryset=SiteGroup.objects.all(),
  3653. label='Site group',
  3654. required=False
  3655. )
  3656. termination_b_site = DynamicModelChoiceField(
  3657. queryset=Site.objects.all(),
  3658. label='Site',
  3659. required=False,
  3660. query_params={
  3661. 'region_id': '$termination_b_region',
  3662. 'group_id': '$termination_b_site_group',
  3663. }
  3664. )
  3665. termination_b_location = DynamicModelChoiceField(
  3666. queryset=Location.objects.all(),
  3667. label='Location',
  3668. required=False,
  3669. null_option='None',
  3670. query_params={
  3671. 'site_id': '$termination_b_site'
  3672. }
  3673. )
  3674. termination_b_rack = DynamicModelChoiceField(
  3675. queryset=Rack.objects.all(),
  3676. label='Rack',
  3677. required=False,
  3678. null_option='None',
  3679. query_params={
  3680. 'site_id': '$termination_b_site',
  3681. 'location_id': '$termination_b_location',
  3682. }
  3683. )
  3684. termination_b_device = DynamicModelChoiceField(
  3685. queryset=Device.objects.all(),
  3686. label='Device',
  3687. required=False,
  3688. query_params={
  3689. 'site_id': '$termination_b_site',
  3690. 'location_id': '$termination_b_location',
  3691. 'rack_id': '$termination_b_rack',
  3692. }
  3693. )
  3694. tags = DynamicModelMultipleChoiceField(
  3695. queryset=Tag.objects.all(),
  3696. required=False
  3697. )
  3698. class Meta:
  3699. model = Cable
  3700. fields = [
  3701. 'termination_b_region', 'termination_b_site', 'termination_b_rack', 'termination_b_device',
  3702. 'termination_b_id', 'type', 'status', 'label', 'color', 'length', 'length_unit', 'tags',
  3703. ]
  3704. widgets = {
  3705. 'status': StaticSelect,
  3706. 'type': StaticSelect,
  3707. 'length_unit': StaticSelect,
  3708. }
  3709. def clean_termination_b_id(self):
  3710. # Return the PK rather than the object
  3711. return getattr(self.cleaned_data['termination_b_id'], 'pk', None)
  3712. class ConnectCableToConsolePortForm(ConnectCableToDeviceForm):
  3713. termination_b_id = DynamicModelChoiceField(
  3714. queryset=ConsolePort.objects.all(),
  3715. label='Name',
  3716. disabled_indicator='_occupied',
  3717. query_params={
  3718. 'device_id': '$termination_b_device'
  3719. }
  3720. )
  3721. class ConnectCableToConsoleServerPortForm(ConnectCableToDeviceForm):
  3722. termination_b_id = DynamicModelChoiceField(
  3723. queryset=ConsoleServerPort.objects.all(),
  3724. label='Name',
  3725. disabled_indicator='_occupied',
  3726. query_params={
  3727. 'device_id': '$termination_b_device'
  3728. }
  3729. )
  3730. class ConnectCableToPowerPortForm(ConnectCableToDeviceForm):
  3731. termination_b_id = DynamicModelChoiceField(
  3732. queryset=PowerPort.objects.all(),
  3733. label='Name',
  3734. disabled_indicator='_occupied',
  3735. query_params={
  3736. 'device_id': '$termination_b_device'
  3737. }
  3738. )
  3739. class ConnectCableToPowerOutletForm(ConnectCableToDeviceForm):
  3740. termination_b_id = DynamicModelChoiceField(
  3741. queryset=PowerOutlet.objects.all(),
  3742. label='Name',
  3743. disabled_indicator='_occupied',
  3744. query_params={
  3745. 'device_id': '$termination_b_device'
  3746. }
  3747. )
  3748. class ConnectCableToInterfaceForm(ConnectCableToDeviceForm):
  3749. termination_b_id = DynamicModelChoiceField(
  3750. queryset=Interface.objects.all(),
  3751. label='Name',
  3752. disabled_indicator='_occupied',
  3753. query_params={
  3754. 'device_id': '$termination_b_device',
  3755. 'kind': 'physical',
  3756. }
  3757. )
  3758. class ConnectCableToFrontPortForm(ConnectCableToDeviceForm):
  3759. termination_b_id = DynamicModelChoiceField(
  3760. queryset=FrontPort.objects.all(),
  3761. label='Name',
  3762. disabled_indicator='_occupied',
  3763. query_params={
  3764. 'device_id': '$termination_b_device'
  3765. }
  3766. )
  3767. class ConnectCableToRearPortForm(ConnectCableToDeviceForm):
  3768. termination_b_id = DynamicModelChoiceField(
  3769. queryset=RearPort.objects.all(),
  3770. label='Name',
  3771. disabled_indicator='_occupied',
  3772. query_params={
  3773. 'device_id': '$termination_b_device'
  3774. }
  3775. )
  3776. class ConnectCableToCircuitTerminationForm(BootstrapMixin, CustomFieldModelForm):
  3777. termination_b_provider = DynamicModelChoiceField(
  3778. queryset=Provider.objects.all(),
  3779. label='Provider',
  3780. required=False
  3781. )
  3782. termination_b_region = DynamicModelChoiceField(
  3783. queryset=Region.objects.all(),
  3784. label='Region',
  3785. required=False
  3786. )
  3787. termination_b_site_group = DynamicModelChoiceField(
  3788. queryset=SiteGroup.objects.all(),
  3789. label='Site group',
  3790. required=False
  3791. )
  3792. termination_b_site = DynamicModelChoiceField(
  3793. queryset=Site.objects.all(),
  3794. label='Site',
  3795. required=False,
  3796. query_params={
  3797. 'region_id': '$termination_b_region',
  3798. 'group_id': '$termination_b_site_group',
  3799. }
  3800. )
  3801. termination_b_circuit = DynamicModelChoiceField(
  3802. queryset=Circuit.objects.all(),
  3803. label='Circuit',
  3804. query_params={
  3805. 'provider_id': '$termination_b_provider',
  3806. 'site_id': '$termination_b_site',
  3807. }
  3808. )
  3809. termination_b_id = DynamicModelChoiceField(
  3810. queryset=CircuitTermination.objects.all(),
  3811. label='Side',
  3812. disabled_indicator='_occupied',
  3813. query_params={
  3814. 'circuit_id': '$termination_b_circuit'
  3815. }
  3816. )
  3817. tags = DynamicModelMultipleChoiceField(
  3818. queryset=Tag.objects.all(),
  3819. required=False
  3820. )
  3821. class Meta:
  3822. model = Cable
  3823. fields = [
  3824. 'termination_b_provider', 'termination_b_region', 'termination_b_site', 'termination_b_circuit',
  3825. 'termination_b_id', 'type', 'status', 'label', 'color', 'length', 'length_unit', 'tags',
  3826. ]
  3827. def clean_termination_b_id(self):
  3828. # Return the PK rather than the object
  3829. return getattr(self.cleaned_data['termination_b_id'], 'pk', None)
  3830. class ConnectCableToPowerFeedForm(BootstrapMixin, CustomFieldModelForm):
  3831. termination_b_region = DynamicModelChoiceField(
  3832. queryset=Region.objects.all(),
  3833. label='Region',
  3834. required=False
  3835. )
  3836. termination_b_site_group = DynamicModelChoiceField(
  3837. queryset=SiteGroup.objects.all(),
  3838. label='Site group',
  3839. required=False
  3840. )
  3841. termination_b_site = DynamicModelChoiceField(
  3842. queryset=Site.objects.all(),
  3843. label='Site',
  3844. required=False,
  3845. query_params={
  3846. 'region_id': '$termination_b_region',
  3847. 'group_id': '$termination_b_site_group',
  3848. }
  3849. )
  3850. termination_b_location = DynamicModelChoiceField(
  3851. queryset=Location.objects.all(),
  3852. label='Location',
  3853. required=False,
  3854. query_params={
  3855. 'site_id': '$termination_b_site'
  3856. }
  3857. )
  3858. termination_b_powerpanel = DynamicModelChoiceField(
  3859. queryset=PowerPanel.objects.all(),
  3860. label='Power Panel',
  3861. required=False,
  3862. query_params={
  3863. 'site_id': '$termination_b_site',
  3864. 'location_id': '$termination_b_location',
  3865. }
  3866. )
  3867. termination_b_id = DynamicModelChoiceField(
  3868. queryset=PowerFeed.objects.all(),
  3869. label='Name',
  3870. disabled_indicator='_occupied',
  3871. query_params={
  3872. 'power_panel_id': '$termination_b_powerpanel'
  3873. }
  3874. )
  3875. tags = DynamicModelMultipleChoiceField(
  3876. queryset=Tag.objects.all(),
  3877. required=False
  3878. )
  3879. class Meta:
  3880. model = Cable
  3881. fields = [
  3882. 'termination_b_location', 'termination_b_powerpanel', 'termination_b_id', 'type', 'status', 'label',
  3883. 'color', 'length', 'length_unit', 'tags',
  3884. ]
  3885. def clean_termination_b_id(self):
  3886. # Return the PK rather than the object
  3887. return getattr(self.cleaned_data['termination_b_id'], 'pk', None)
  3888. class CableForm(BootstrapMixin, CustomFieldModelForm):
  3889. tags = DynamicModelMultipleChoiceField(
  3890. queryset=Tag.objects.all(),
  3891. required=False
  3892. )
  3893. class Meta:
  3894. model = Cable
  3895. fields = [
  3896. 'type', 'status', 'label', 'color', 'length', 'length_unit', 'tags',
  3897. ]
  3898. widgets = {
  3899. 'status': StaticSelect,
  3900. 'type': StaticSelect,
  3901. 'length_unit': StaticSelect,
  3902. }
  3903. error_messages = {
  3904. 'length': {
  3905. 'max_value': 'Maximum length is 32767 (any unit)'
  3906. }
  3907. }
  3908. class CableCSVForm(CustomFieldModelCSVForm):
  3909. # Termination A
  3910. side_a_device = CSVModelChoiceField(
  3911. queryset=Device.objects.all(),
  3912. to_field_name='name',
  3913. help_text='Side A device'
  3914. )
  3915. side_a_type = CSVContentTypeField(
  3916. queryset=ContentType.objects.all(),
  3917. limit_choices_to=CABLE_TERMINATION_MODELS,
  3918. help_text='Side A type'
  3919. )
  3920. side_a_name = forms.CharField(
  3921. help_text='Side A component name'
  3922. )
  3923. # Termination B
  3924. side_b_device = CSVModelChoiceField(
  3925. queryset=Device.objects.all(),
  3926. to_field_name='name',
  3927. help_text='Side B device'
  3928. )
  3929. side_b_type = CSVContentTypeField(
  3930. queryset=ContentType.objects.all(),
  3931. limit_choices_to=CABLE_TERMINATION_MODELS,
  3932. help_text='Side B type'
  3933. )
  3934. side_b_name = forms.CharField(
  3935. help_text='Side B component name'
  3936. )
  3937. # Cable attributes
  3938. status = CSVChoiceField(
  3939. choices=CableStatusChoices,
  3940. required=False,
  3941. help_text='Connection status'
  3942. )
  3943. type = CSVChoiceField(
  3944. choices=CableTypeChoices,
  3945. required=False,
  3946. help_text='Physical medium classification'
  3947. )
  3948. length_unit = CSVChoiceField(
  3949. choices=CableLengthUnitChoices,
  3950. required=False,
  3951. help_text='Length unit'
  3952. )
  3953. class Meta:
  3954. model = Cable
  3955. fields = [
  3956. 'side_a_device', 'side_a_type', 'side_a_name', 'side_b_device', 'side_b_type', 'side_b_name', 'type',
  3957. 'status', 'label', 'color', 'length', 'length_unit',
  3958. ]
  3959. help_texts = {
  3960. 'color': mark_safe('RGB color in hexadecimal (e.g. <code>00ff00</code>)'),
  3961. }
  3962. def _clean_side(self, side):
  3963. """
  3964. Derive a Cable's A/B termination objects.
  3965. :param side: 'a' or 'b'
  3966. """
  3967. assert side in 'ab', f"Invalid side designation: {side}"
  3968. device = self.cleaned_data.get(f'side_{side}_device')
  3969. content_type = self.cleaned_data.get(f'side_{side}_type')
  3970. name = self.cleaned_data.get(f'side_{side}_name')
  3971. if not device or not content_type or not name:
  3972. return None
  3973. model = content_type.model_class()
  3974. try:
  3975. termination_object = model.objects.get(device=device, name=name)
  3976. if termination_object.cable is not None:
  3977. raise forms.ValidationError(f"Side {side.upper()}: {device} {termination_object} is already connected")
  3978. except ObjectDoesNotExist:
  3979. raise forms.ValidationError(f"{side.upper()} side termination not found: {device} {name}")
  3980. setattr(self.instance, f'termination_{side}', termination_object)
  3981. return termination_object
  3982. def clean_side_a_name(self):
  3983. return self._clean_side('a')
  3984. def clean_side_b_name(self):
  3985. return self._clean_side('b')
  3986. def clean_length_unit(self):
  3987. # Avoid trying to save as NULL
  3988. length_unit = self.cleaned_data.get('length_unit', None)
  3989. return length_unit if length_unit is not None else ''
  3990. class CableBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  3991. pk = forms.ModelMultipleChoiceField(
  3992. queryset=Cable.objects.all(),
  3993. widget=forms.MultipleHiddenInput
  3994. )
  3995. type = forms.ChoiceField(
  3996. choices=add_blank_choice(CableTypeChoices),
  3997. required=False,
  3998. initial='',
  3999. widget=StaticSelect()
  4000. )
  4001. status = forms.ChoiceField(
  4002. choices=add_blank_choice(CableStatusChoices),
  4003. required=False,
  4004. widget=StaticSelect(),
  4005. initial=''
  4006. )
  4007. label = forms.CharField(
  4008. max_length=100,
  4009. required=False
  4010. )
  4011. color = ColorField(
  4012. required=False
  4013. )
  4014. length = forms.DecimalField(
  4015. min_value=0,
  4016. required=False
  4017. )
  4018. length_unit = forms.ChoiceField(
  4019. choices=add_blank_choice(CableLengthUnitChoices),
  4020. required=False,
  4021. initial='',
  4022. widget=StaticSelect()
  4023. )
  4024. class Meta:
  4025. nullable_fields = [
  4026. 'type', 'status', 'label', 'color', 'length',
  4027. ]
  4028. def clean(self):
  4029. super().clean()
  4030. # Validate length/unit
  4031. length = self.cleaned_data.get('length')
  4032. length_unit = self.cleaned_data.get('length_unit')
  4033. if length and not length_unit:
  4034. raise forms.ValidationError({
  4035. 'length_unit': "Must specify a unit when setting length"
  4036. })
  4037. class CableFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  4038. model = Cable
  4039. field_groups = [
  4040. ['q', 'tag'],
  4041. ['site_id', 'rack_id', 'device_id'],
  4042. ['type', 'status', 'color'],
  4043. ['tenant_id'],
  4044. ]
  4045. q = forms.CharField(
  4046. required=False,
  4047. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  4048. label=_('Search')
  4049. )
  4050. region_id = DynamicModelMultipleChoiceField(
  4051. queryset=Region.objects.all(),
  4052. required=False,
  4053. label=_('Region'),
  4054. fetch_trigger='open'
  4055. )
  4056. site_id = DynamicModelMultipleChoiceField(
  4057. queryset=Site.objects.all(),
  4058. required=False,
  4059. query_params={
  4060. 'region_id': '$region_id'
  4061. },
  4062. label=_('Site'),
  4063. fetch_trigger='open'
  4064. )
  4065. tenant_id = DynamicModelMultipleChoiceField(
  4066. queryset=Tenant.objects.all(),
  4067. required=False,
  4068. label=_('Tenant'),
  4069. fetch_trigger='open'
  4070. )
  4071. rack_id = DynamicModelMultipleChoiceField(
  4072. queryset=Rack.objects.all(),
  4073. required=False,
  4074. label=_('Rack'),
  4075. null_option='None',
  4076. query_params={
  4077. 'site_id': '$site_id'
  4078. },
  4079. fetch_trigger='open'
  4080. )
  4081. type = forms.MultipleChoiceField(
  4082. choices=add_blank_choice(CableTypeChoices),
  4083. required=False,
  4084. widget=StaticSelect()
  4085. )
  4086. status = forms.ChoiceField(
  4087. required=False,
  4088. choices=add_blank_choice(CableStatusChoices),
  4089. widget=StaticSelect()
  4090. )
  4091. color = ColorField(
  4092. required=False
  4093. )
  4094. device_id = DynamicModelMultipleChoiceField(
  4095. queryset=Device.objects.all(),
  4096. required=False,
  4097. query_params={
  4098. 'site_id': '$site_id',
  4099. 'tenant_id': '$tenant_id',
  4100. 'rack_id': '$rack_id',
  4101. },
  4102. label=_('Device'),
  4103. fetch_trigger='open'
  4104. )
  4105. tag = TagFilterField(model)
  4106. #
  4107. # Connections
  4108. #
  4109. class ConsoleConnectionFilterForm(BootstrapMixin, forms.Form):
  4110. region_id = DynamicModelMultipleChoiceField(
  4111. queryset=Region.objects.all(),
  4112. required=False,
  4113. label=_('Region'),
  4114. fetch_trigger='open'
  4115. )
  4116. site_id = DynamicModelMultipleChoiceField(
  4117. queryset=Site.objects.all(),
  4118. required=False,
  4119. query_params={
  4120. 'region_id': '$region_id'
  4121. },
  4122. label=_('Site'),
  4123. fetch_trigger='open'
  4124. )
  4125. device_id = DynamicModelMultipleChoiceField(
  4126. queryset=Device.objects.all(),
  4127. required=False,
  4128. query_params={
  4129. 'site_id': '$site_id'
  4130. },
  4131. label=_('Device'),
  4132. fetch_trigger='open'
  4133. )
  4134. class PowerConnectionFilterForm(BootstrapMixin, forms.Form):
  4135. region_id = DynamicModelMultipleChoiceField(
  4136. queryset=Region.objects.all(),
  4137. required=False,
  4138. label=_('Region'),
  4139. fetch_trigger='open'
  4140. )
  4141. site_id = DynamicModelMultipleChoiceField(
  4142. queryset=Site.objects.all(),
  4143. required=False,
  4144. query_params={
  4145. 'region_id': '$region_id'
  4146. },
  4147. label=_('Site'),
  4148. fetch_trigger='open'
  4149. )
  4150. device_id = DynamicModelMultipleChoiceField(
  4151. queryset=Device.objects.all(),
  4152. required=False,
  4153. query_params={
  4154. 'site_id': '$site_id'
  4155. },
  4156. label=_('Device'),
  4157. fetch_trigger='open'
  4158. )
  4159. class InterfaceConnectionFilterForm(BootstrapMixin, forms.Form):
  4160. region_id = DynamicModelMultipleChoiceField(
  4161. queryset=Region.objects.all(),
  4162. required=False,
  4163. label=_('Region'),
  4164. fetch_trigger='open'
  4165. )
  4166. site_id = DynamicModelMultipleChoiceField(
  4167. queryset=Site.objects.all(),
  4168. required=False,
  4169. query_params={
  4170. 'region_id': '$region_id'
  4171. },
  4172. label=_('Site'),
  4173. fetch_trigger='open'
  4174. )
  4175. device_id = DynamicModelMultipleChoiceField(
  4176. queryset=Device.objects.all(),
  4177. required=False,
  4178. query_params={
  4179. 'site_id': '$site_id'
  4180. },
  4181. label=_('Device'),
  4182. fetch_trigger='open'
  4183. )
  4184. #
  4185. # Virtual chassis
  4186. #
  4187. class DeviceSelectionForm(forms.Form):
  4188. pk = forms.ModelMultipleChoiceField(
  4189. queryset=Device.objects.all(),
  4190. widget=forms.MultipleHiddenInput()
  4191. )
  4192. class VirtualChassisCreateForm(BootstrapMixin, CustomFieldModelForm):
  4193. region = DynamicModelChoiceField(
  4194. queryset=Region.objects.all(),
  4195. required=False,
  4196. initial_params={
  4197. 'sites': '$site'
  4198. }
  4199. )
  4200. site_group = DynamicModelChoiceField(
  4201. queryset=SiteGroup.objects.all(),
  4202. required=False,
  4203. initial_params={
  4204. 'sites': '$site'
  4205. }
  4206. )
  4207. site = DynamicModelChoiceField(
  4208. queryset=Site.objects.all(),
  4209. required=False,
  4210. query_params={
  4211. 'region_id': '$region',
  4212. 'group_id': '$site_group',
  4213. }
  4214. )
  4215. rack = DynamicModelChoiceField(
  4216. queryset=Rack.objects.all(),
  4217. required=False,
  4218. null_option='None',
  4219. query_params={
  4220. 'site_id': '$site'
  4221. }
  4222. )
  4223. members = DynamicModelMultipleChoiceField(
  4224. queryset=Device.objects.all(),
  4225. required=False,
  4226. query_params={
  4227. 'site_id': '$site',
  4228. 'rack_id': '$rack',
  4229. }
  4230. )
  4231. initial_position = forms.IntegerField(
  4232. initial=1,
  4233. required=False,
  4234. help_text='Position of the first member device. Increases by one for each additional member.'
  4235. )
  4236. tags = DynamicModelMultipleChoiceField(
  4237. queryset=Tag.objects.all(),
  4238. required=False
  4239. )
  4240. class Meta:
  4241. model = VirtualChassis
  4242. fields = [
  4243. 'name', 'domain', 'region', 'site_group', 'site', 'rack', 'members', 'initial_position', 'tags',
  4244. ]
  4245. def save(self, *args, **kwargs):
  4246. instance = super().save(*args, **kwargs)
  4247. # Assign VC members
  4248. if instance.pk:
  4249. initial_position = self.cleaned_data.get('initial_position') or 1
  4250. for i, member in enumerate(self.cleaned_data['members'], start=initial_position):
  4251. member.virtual_chassis = instance
  4252. member.vc_position = i
  4253. member.save()
  4254. return instance
  4255. class VirtualChassisForm(BootstrapMixin, CustomFieldModelForm):
  4256. master = forms.ModelChoiceField(
  4257. queryset=Device.objects.all(),
  4258. required=False,
  4259. )
  4260. tags = DynamicModelMultipleChoiceField(
  4261. queryset=Tag.objects.all(),
  4262. required=False
  4263. )
  4264. class Meta:
  4265. model = VirtualChassis
  4266. fields = [
  4267. 'name', 'domain', 'master', 'tags',
  4268. ]
  4269. widgets = {
  4270. 'master': SelectWithPK(),
  4271. }
  4272. def __init__(self, *args, **kwargs):
  4273. super().__init__(*args, **kwargs)
  4274. self.fields['master'].queryset = Device.objects.filter(virtual_chassis=self.instance)
  4275. class BaseVCMemberFormSet(forms.BaseModelFormSet):
  4276. def clean(self):
  4277. super().clean()
  4278. # Check for duplicate VC position values
  4279. vc_position_list = []
  4280. for form in self.forms:
  4281. vc_position = form.cleaned_data.get('vc_position')
  4282. if vc_position:
  4283. if vc_position in vc_position_list:
  4284. error_msg = 'A virtual chassis member already exists in position {}.'.format(vc_position)
  4285. form.add_error('vc_position', error_msg)
  4286. vc_position_list.append(vc_position)
  4287. class DeviceVCMembershipForm(forms.ModelForm):
  4288. class Meta:
  4289. model = Device
  4290. fields = [
  4291. 'vc_position', 'vc_priority',
  4292. ]
  4293. labels = {
  4294. 'vc_position': 'Position',
  4295. 'vc_priority': 'Priority',
  4296. }
  4297. def __init__(self, validate_vc_position=False, *args, **kwargs):
  4298. super().__init__(*args, **kwargs)
  4299. # Require VC position (only required when the Device is a VirtualChassis member)
  4300. self.fields['vc_position'].required = True
  4301. # Add bootstrap classes to form elements.
  4302. self.fields['vc_position'].widget.attrs = {'class': 'form-control'}
  4303. self.fields['vc_priority'].widget.attrs = {'class': 'form-control'}
  4304. # Validation of vc_position is optional. This is only required when adding a new member to an existing
  4305. # VirtualChassis. Otherwise, vc_position validation is handled by BaseVCMemberFormSet.
  4306. self.validate_vc_position = validate_vc_position
  4307. def clean_vc_position(self):
  4308. vc_position = self.cleaned_data['vc_position']
  4309. if self.validate_vc_position:
  4310. conflicting_members = Device.objects.filter(
  4311. virtual_chassis=self.instance.virtual_chassis,
  4312. vc_position=vc_position
  4313. )
  4314. if conflicting_members.exists():
  4315. raise forms.ValidationError(
  4316. 'A virtual chassis member already exists in position {}.'.format(vc_position)
  4317. )
  4318. return vc_position
  4319. class VCMemberSelectForm(BootstrapMixin, forms.Form):
  4320. region = DynamicModelChoiceField(
  4321. queryset=Region.objects.all(),
  4322. required=False,
  4323. initial_params={
  4324. 'sites': '$site'
  4325. }
  4326. )
  4327. site_group = DynamicModelChoiceField(
  4328. queryset=SiteGroup.objects.all(),
  4329. required=False,
  4330. initial_params={
  4331. 'sites': '$site'
  4332. }
  4333. )
  4334. site = DynamicModelChoiceField(
  4335. queryset=Site.objects.all(),
  4336. required=False,
  4337. query_params={
  4338. 'region_id': '$region',
  4339. 'group_id': '$site_group',
  4340. }
  4341. )
  4342. rack = DynamicModelChoiceField(
  4343. queryset=Rack.objects.all(),
  4344. required=False,
  4345. null_option='None',
  4346. query_params={
  4347. 'site_id': '$site'
  4348. }
  4349. )
  4350. device = DynamicModelChoiceField(
  4351. queryset=Device.objects.all(),
  4352. query_params={
  4353. 'site_id': '$site',
  4354. 'rack_id': '$rack',
  4355. 'virtual_chassis_id': 'null',
  4356. }
  4357. )
  4358. def clean_device(self):
  4359. device = self.cleaned_data['device']
  4360. if device.virtual_chassis is not None:
  4361. raise forms.ValidationError(
  4362. f"Device {device} is already assigned to a virtual chassis."
  4363. )
  4364. return device
  4365. class VirtualChassisBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  4366. pk = forms.ModelMultipleChoiceField(
  4367. queryset=VirtualChassis.objects.all(),
  4368. widget=forms.MultipleHiddenInput()
  4369. )
  4370. domain = forms.CharField(
  4371. max_length=30,
  4372. required=False
  4373. )
  4374. class Meta:
  4375. nullable_fields = ['domain']
  4376. class VirtualChassisCSVForm(CustomFieldModelCSVForm):
  4377. master = CSVModelChoiceField(
  4378. queryset=Device.objects.all(),
  4379. to_field_name='name',
  4380. required=False,
  4381. help_text='Master device'
  4382. )
  4383. class Meta:
  4384. model = VirtualChassis
  4385. fields = ('name', 'domain', 'master')
  4386. class VirtualChassisFilterForm(BootstrapMixin, TenancyFilterForm, CustomFieldModelFilterForm):
  4387. model = VirtualChassis
  4388. field_order = ['q', 'region_id', 'site_group_id', 'site_id', 'tenant_group_id', 'tenant_id']
  4389. field_groups = [
  4390. ['q', 'tag'],
  4391. ['region_id', 'site_group_id', 'site_id'],
  4392. ['tenant_group_id', 'tenant_id'],
  4393. ]
  4394. q = forms.CharField(
  4395. required=False,
  4396. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  4397. label=_('Search')
  4398. )
  4399. region_id = DynamicModelMultipleChoiceField(
  4400. queryset=Region.objects.all(),
  4401. required=False,
  4402. label=_('Region'),
  4403. fetch_trigger='open'
  4404. )
  4405. site_group_id = DynamicModelMultipleChoiceField(
  4406. queryset=SiteGroup.objects.all(),
  4407. required=False,
  4408. label=_('Site group'),
  4409. fetch_trigger='open'
  4410. )
  4411. site_id = DynamicModelMultipleChoiceField(
  4412. queryset=Site.objects.all(),
  4413. required=False,
  4414. query_params={
  4415. 'region_id': '$region_id',
  4416. 'group_id': '$site_group_id',
  4417. },
  4418. label=_('Site'),
  4419. fetch_trigger='open'
  4420. )
  4421. tag = TagFilterField(model)
  4422. #
  4423. # Power panels
  4424. #
  4425. class PowerPanelForm(BootstrapMixin, CustomFieldModelForm):
  4426. region = DynamicModelChoiceField(
  4427. queryset=Region.objects.all(),
  4428. required=False,
  4429. initial_params={
  4430. 'sites': '$site'
  4431. }
  4432. )
  4433. site_group = DynamicModelChoiceField(
  4434. queryset=SiteGroup.objects.all(),
  4435. required=False,
  4436. initial_params={
  4437. 'sites': '$site'
  4438. }
  4439. )
  4440. site = DynamicModelChoiceField(
  4441. queryset=Site.objects.all(),
  4442. query_params={
  4443. 'region_id': '$region',
  4444. 'group_id': '$site_group',
  4445. }
  4446. )
  4447. location = DynamicModelChoiceField(
  4448. queryset=Location.objects.all(),
  4449. required=False,
  4450. query_params={
  4451. 'site_id': '$site'
  4452. }
  4453. )
  4454. tags = DynamicModelMultipleChoiceField(
  4455. queryset=Tag.objects.all(),
  4456. required=False
  4457. )
  4458. class Meta:
  4459. model = PowerPanel
  4460. fields = [
  4461. 'region', 'site_group', 'site', 'location', 'name', 'tags',
  4462. ]
  4463. fieldsets = (
  4464. ('Power Panel', ('region', 'site_group', 'site', 'location', 'name', 'tags')),
  4465. )
  4466. class PowerPanelCSVForm(CustomFieldModelCSVForm):
  4467. site = CSVModelChoiceField(
  4468. queryset=Site.objects.all(),
  4469. to_field_name='name',
  4470. help_text='Name of parent site'
  4471. )
  4472. location = CSVModelChoiceField(
  4473. queryset=Location.objects.all(),
  4474. required=False,
  4475. to_field_name='name'
  4476. )
  4477. class Meta:
  4478. model = PowerPanel
  4479. fields = ('site', 'location', 'name')
  4480. def __init__(self, data=None, *args, **kwargs):
  4481. super().__init__(data, *args, **kwargs)
  4482. if data:
  4483. # Limit group queryset by assigned site
  4484. params = {f"site__{self.fields['site'].to_field_name}": data.get('site')}
  4485. self.fields['location'].queryset = self.fields['location'].queryset.filter(**params)
  4486. class PowerPanelBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  4487. pk = forms.ModelMultipleChoiceField(
  4488. queryset=PowerPanel.objects.all(),
  4489. widget=forms.MultipleHiddenInput
  4490. )
  4491. region = DynamicModelChoiceField(
  4492. queryset=Region.objects.all(),
  4493. required=False,
  4494. initial_params={
  4495. 'sites': '$site'
  4496. }
  4497. )
  4498. site_group = DynamicModelChoiceField(
  4499. queryset=SiteGroup.objects.all(),
  4500. required=False,
  4501. initial_params={
  4502. 'sites': '$site'
  4503. }
  4504. )
  4505. site = DynamicModelChoiceField(
  4506. queryset=Site.objects.all(),
  4507. required=False,
  4508. query_params={
  4509. 'region_id': '$region',
  4510. 'group_id': '$site_group',
  4511. }
  4512. )
  4513. location = DynamicModelChoiceField(
  4514. queryset=Location.objects.all(),
  4515. required=False,
  4516. query_params={
  4517. 'site_id': '$site'
  4518. }
  4519. )
  4520. class Meta:
  4521. nullable_fields = ['location']
  4522. class PowerPanelFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  4523. model = PowerPanel
  4524. field_groups = (
  4525. ('q', 'tag'),
  4526. ('region_id', 'site_group_id', 'site_id', 'location_id')
  4527. )
  4528. q = forms.CharField(
  4529. required=False,
  4530. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  4531. label=_('Search')
  4532. )
  4533. region_id = DynamicModelMultipleChoiceField(
  4534. queryset=Region.objects.all(),
  4535. required=False,
  4536. label=_('Region'),
  4537. fetch_trigger='open'
  4538. )
  4539. site_group_id = DynamicModelMultipleChoiceField(
  4540. queryset=SiteGroup.objects.all(),
  4541. required=False,
  4542. label=_('Site group'),
  4543. fetch_trigger='open'
  4544. )
  4545. site_id = DynamicModelMultipleChoiceField(
  4546. queryset=Site.objects.all(),
  4547. required=False,
  4548. query_params={
  4549. 'region_id': '$region_id',
  4550. 'group_id': '$site_group_id',
  4551. },
  4552. label=_('Site'),
  4553. fetch_trigger='open'
  4554. )
  4555. location_id = DynamicModelMultipleChoiceField(
  4556. queryset=Location.objects.all(),
  4557. required=False,
  4558. null_option='None',
  4559. query_params={
  4560. 'site_id': '$site_id'
  4561. },
  4562. label=_('Location'),
  4563. fetch_trigger='open'
  4564. )
  4565. tag = TagFilterField(model)
  4566. #
  4567. # Power feeds
  4568. #
  4569. class PowerFeedForm(BootstrapMixin, CustomFieldModelForm):
  4570. region = DynamicModelChoiceField(
  4571. queryset=Region.objects.all(),
  4572. required=False,
  4573. initial_params={
  4574. 'sites__powerpanel': '$power_panel'
  4575. }
  4576. )
  4577. site_group = DynamicModelChoiceField(
  4578. queryset=SiteGroup.objects.all(),
  4579. required=False,
  4580. initial_params={
  4581. 'sites': '$site'
  4582. }
  4583. )
  4584. site = DynamicModelChoiceField(
  4585. queryset=Site.objects.all(),
  4586. required=False,
  4587. initial_params={
  4588. 'powerpanel': '$power_panel'
  4589. },
  4590. query_params={
  4591. 'region_id': '$region',
  4592. 'group_id': '$site_group',
  4593. }
  4594. )
  4595. power_panel = DynamicModelChoiceField(
  4596. queryset=PowerPanel.objects.all(),
  4597. query_params={
  4598. 'site_id': '$site'
  4599. }
  4600. )
  4601. rack = DynamicModelChoiceField(
  4602. queryset=Rack.objects.all(),
  4603. required=False,
  4604. query_params={
  4605. 'site_id': '$site'
  4606. }
  4607. )
  4608. comments = CommentField()
  4609. tags = DynamicModelMultipleChoiceField(
  4610. queryset=Tag.objects.all(),
  4611. required=False
  4612. )
  4613. class Meta:
  4614. model = PowerFeed
  4615. fields = [
  4616. 'region', 'site_group', 'site', 'power_panel', 'rack', 'name', 'status', 'type', 'mark_connected', 'supply',
  4617. 'phase', 'voltage', 'amperage', 'max_utilization', 'comments', 'tags',
  4618. ]
  4619. fieldsets = (
  4620. ('Power Panel', ('region', 'site', 'power_panel')),
  4621. ('Power Feed', ('rack', 'name', 'status', 'type', 'mark_connected', 'tags')),
  4622. ('Characteristics', ('supply', 'voltage', 'amperage', 'phase', 'max_utilization')),
  4623. )
  4624. widgets = {
  4625. 'status': StaticSelect(),
  4626. 'type': StaticSelect(),
  4627. 'supply': StaticSelect(),
  4628. 'phase': StaticSelect(),
  4629. }
  4630. class PowerFeedCSVForm(CustomFieldModelCSVForm):
  4631. site = CSVModelChoiceField(
  4632. queryset=Site.objects.all(),
  4633. to_field_name='name',
  4634. help_text='Assigned site'
  4635. )
  4636. power_panel = CSVModelChoiceField(
  4637. queryset=PowerPanel.objects.all(),
  4638. to_field_name='name',
  4639. help_text='Upstream power panel'
  4640. )
  4641. location = CSVModelChoiceField(
  4642. queryset=Location.objects.all(),
  4643. to_field_name='name',
  4644. required=False,
  4645. help_text="Rack's location (if any)"
  4646. )
  4647. rack = CSVModelChoiceField(
  4648. queryset=Rack.objects.all(),
  4649. to_field_name='name',
  4650. required=False,
  4651. help_text='Rack'
  4652. )
  4653. status = CSVChoiceField(
  4654. choices=PowerFeedStatusChoices,
  4655. required=False,
  4656. help_text='Operational status'
  4657. )
  4658. type = CSVChoiceField(
  4659. choices=PowerFeedTypeChoices,
  4660. required=False,
  4661. help_text='Primary or redundant'
  4662. )
  4663. supply = CSVChoiceField(
  4664. choices=PowerFeedSupplyChoices,
  4665. required=False,
  4666. help_text='Supply type (AC/DC)'
  4667. )
  4668. phase = CSVChoiceField(
  4669. choices=PowerFeedPhaseChoices,
  4670. required=False,
  4671. help_text='Single or three-phase'
  4672. )
  4673. class Meta:
  4674. model = PowerFeed
  4675. fields = (
  4676. 'site', 'power_panel', 'location', 'rack', 'name', 'status', 'type', 'mark_connected', 'supply', 'phase',
  4677. 'voltage', 'amperage', 'max_utilization', 'comments',
  4678. )
  4679. def __init__(self, data=None, *args, **kwargs):
  4680. super().__init__(data, *args, **kwargs)
  4681. if data:
  4682. # Limit power_panel queryset by site
  4683. params = {f"site__{self.fields['site'].to_field_name}": data.get('site')}
  4684. self.fields['power_panel'].queryset = self.fields['power_panel'].queryset.filter(**params)
  4685. # Limit location queryset by site
  4686. params = {f"site__{self.fields['site'].to_field_name}": data.get('site')}
  4687. self.fields['location'].queryset = self.fields['location'].queryset.filter(**params)
  4688. # Limit rack queryset by site and group
  4689. params = {
  4690. f"site__{self.fields['site'].to_field_name}": data.get('site'),
  4691. f"location__{self.fields['location'].to_field_name}": data.get('location'),
  4692. }
  4693. self.fields['rack'].queryset = self.fields['rack'].queryset.filter(**params)
  4694. class PowerFeedBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldModelBulkEditForm):
  4695. pk = forms.ModelMultipleChoiceField(
  4696. queryset=PowerFeed.objects.all(),
  4697. widget=forms.MultipleHiddenInput
  4698. )
  4699. power_panel = DynamicModelChoiceField(
  4700. queryset=PowerPanel.objects.all(),
  4701. required=False
  4702. )
  4703. rack = DynamicModelChoiceField(
  4704. queryset=Rack.objects.all(),
  4705. required=False,
  4706. )
  4707. status = forms.ChoiceField(
  4708. choices=add_blank_choice(PowerFeedStatusChoices),
  4709. required=False,
  4710. initial='',
  4711. widget=StaticSelect()
  4712. )
  4713. type = forms.ChoiceField(
  4714. choices=add_blank_choice(PowerFeedTypeChoices),
  4715. required=False,
  4716. initial='',
  4717. widget=StaticSelect()
  4718. )
  4719. supply = forms.ChoiceField(
  4720. choices=add_blank_choice(PowerFeedSupplyChoices),
  4721. required=False,
  4722. initial='',
  4723. widget=StaticSelect()
  4724. )
  4725. phase = forms.ChoiceField(
  4726. choices=add_blank_choice(PowerFeedPhaseChoices),
  4727. required=False,
  4728. initial='',
  4729. widget=StaticSelect()
  4730. )
  4731. voltage = forms.IntegerField(
  4732. required=False
  4733. )
  4734. amperage = forms.IntegerField(
  4735. required=False
  4736. )
  4737. max_utilization = forms.IntegerField(
  4738. required=False
  4739. )
  4740. mark_connected = forms.NullBooleanField(
  4741. required=False,
  4742. widget=BulkEditNullBooleanSelect
  4743. )
  4744. comments = CommentField(
  4745. widget=SmallTextarea,
  4746. label='Comments'
  4747. )
  4748. class Meta:
  4749. nullable_fields = [
  4750. 'location', 'comments',
  4751. ]
  4752. class PowerFeedFilterForm(BootstrapMixin, CustomFieldModelFilterForm):
  4753. model = PowerFeed
  4754. field_groups = [
  4755. ['q', 'tag'],
  4756. ['region_id', 'site_group_id', 'site_id'],
  4757. ['power_panel_id', 'rack_id'],
  4758. ['status', 'type', 'supply', 'phase', 'voltage', 'amperage', 'max_utilization'],
  4759. ]
  4760. q = forms.CharField(
  4761. required=False,
  4762. widget=forms.TextInput(attrs={'placeholder': _('All Fields')}),
  4763. label=_('Search')
  4764. )
  4765. region_id = DynamicModelMultipleChoiceField(
  4766. queryset=Region.objects.all(),
  4767. required=False,
  4768. label=_('Region'),
  4769. fetch_trigger='open'
  4770. )
  4771. site_group_id = DynamicModelMultipleChoiceField(
  4772. queryset=SiteGroup.objects.all(),
  4773. required=False,
  4774. label=_('Site group'),
  4775. fetch_trigger='open'
  4776. )
  4777. site_id = DynamicModelMultipleChoiceField(
  4778. queryset=Site.objects.all(),
  4779. required=False,
  4780. query_params={
  4781. 'region_id': '$region_id'
  4782. },
  4783. label=_('Site'),
  4784. fetch_trigger='open'
  4785. )
  4786. power_panel_id = DynamicModelMultipleChoiceField(
  4787. queryset=PowerPanel.objects.all(),
  4788. required=False,
  4789. null_option='None',
  4790. query_params={
  4791. 'site_id': '$site_id'
  4792. },
  4793. label=_('Power panel'),
  4794. fetch_trigger='open'
  4795. )
  4796. rack_id = DynamicModelMultipleChoiceField(
  4797. queryset=Rack.objects.all(),
  4798. required=False,
  4799. null_option='None',
  4800. query_params={
  4801. 'site_id': '$site_id'
  4802. },
  4803. label=_('Rack'),
  4804. fetch_trigger='open'
  4805. )
  4806. status = forms.MultipleChoiceField(
  4807. choices=PowerFeedStatusChoices,
  4808. required=False,
  4809. widget=StaticSelectMultiple()
  4810. )
  4811. type = forms.ChoiceField(
  4812. choices=add_blank_choice(PowerFeedTypeChoices),
  4813. required=False,
  4814. widget=StaticSelect()
  4815. )
  4816. supply = forms.ChoiceField(
  4817. choices=add_blank_choice(PowerFeedSupplyChoices),
  4818. required=False,
  4819. widget=StaticSelect()
  4820. )
  4821. phase = forms.ChoiceField(
  4822. choices=add_blank_choice(PowerFeedPhaseChoices),
  4823. required=False,
  4824. widget=StaticSelect()
  4825. )
  4826. voltage = forms.IntegerField(
  4827. required=False
  4828. )
  4829. amperage = forms.IntegerField(
  4830. required=False
  4831. )
  4832. max_utilization = forms.IntegerField(
  4833. required=False
  4834. )
  4835. tag = TagFilterField(model)