models.py 97 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190
  1. from collections import OrderedDict
  2. from itertools import count, groupby
  3. from django.conf import settings
  4. from django.contrib.auth.models import User
  5. from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
  6. from django.contrib.contenttypes.models import ContentType
  7. from django.contrib.postgres.fields import ArrayField, JSONField
  8. from django.core.exceptions import ObjectDoesNotExist, ValidationError
  9. from django.core.validators import MaxValueValidator, MinValueValidator
  10. from django.db import models
  11. from django.db.models import Count, F, ProtectedError, Q, Sum
  12. from django.urls import reverse
  13. from mptt.models import MPTTModel, TreeForeignKey
  14. from taggit.managers import TaggableManager
  15. from timezone_field import TimeZoneField
  16. from extras.models import ConfigContextModel, CustomFieldModel, ObjectChange, TaggedItem
  17. from utilities.fields import ColorField
  18. from utilities.managers import NaturalOrderingManager
  19. from utilities.models import ChangeLoggedModel
  20. from utilities.utils import serialize_object, to_meters
  21. from .constants import *
  22. from .exceptions import LoopDetected
  23. from .fields import ASNField, MACAddressField
  24. from .managers import InterfaceManager
  25. class ComponentTemplateModel(models.Model):
  26. class Meta:
  27. abstract = True
  28. def instantiate(self, device):
  29. """
  30. Instantiate a new component on the specified Device.
  31. """
  32. raise NotImplementedError()
  33. def to_objectchange(self, action):
  34. return ObjectChange(
  35. changed_object=self,
  36. object_repr=str(self),
  37. action=action,
  38. related_object=self.device_type,
  39. object_data=serialize_object(self)
  40. )
  41. class ComponentModel(models.Model):
  42. description = models.CharField(
  43. max_length=100,
  44. blank=True
  45. )
  46. class Meta:
  47. abstract = True
  48. def to_objectchange(self, action):
  49. # Annotate the parent Device/VM
  50. try:
  51. parent = getattr(self, 'device', None) or getattr(self, 'virtual_machine', None)
  52. except ObjectDoesNotExist:
  53. # The parent device/VM has already been deleted
  54. parent = None
  55. return ObjectChange(
  56. changed_object=self,
  57. object_repr=str(self),
  58. action=action,
  59. related_object=parent,
  60. object_data=serialize_object(self)
  61. )
  62. @property
  63. def parent(self):
  64. return getattr(self, 'device', None)
  65. class CableTermination(models.Model):
  66. cable = models.ForeignKey(
  67. to='dcim.Cable',
  68. on_delete=models.SET_NULL,
  69. related_name='+',
  70. blank=True,
  71. null=True
  72. )
  73. # Generic relations to Cable. These ensure that an attached Cable is deleted if the terminated object is deleted.
  74. _cabled_as_a = GenericRelation(
  75. to='dcim.Cable',
  76. content_type_field='termination_a_type',
  77. object_id_field='termination_a_id'
  78. )
  79. _cabled_as_b = GenericRelation(
  80. to='dcim.Cable',
  81. content_type_field='termination_b_type',
  82. object_id_field='termination_b_id'
  83. )
  84. is_path_endpoint = True
  85. class Meta:
  86. abstract = True
  87. def trace(self, position=1, follow_circuits=False, cable_history=None):
  88. """
  89. Return a list representing a complete cable path, with each individual segment represented as a three-tuple:
  90. [
  91. (termination A, cable, termination B),
  92. (termination C, cable, termination D),
  93. (termination E, cable, termination F)
  94. ]
  95. """
  96. def get_peer_port(termination, position=1, follow_circuits=False):
  97. from circuits.models import CircuitTermination
  98. # Map a front port to its corresponding rear port
  99. if isinstance(termination, FrontPort):
  100. return termination.rear_port, termination.rear_port_position
  101. # Map a rear port/position to its corresponding front port
  102. elif isinstance(termination, RearPort):
  103. if position not in range(1, termination.positions + 1):
  104. raise Exception("Invalid position for {} ({} positions): {})".format(
  105. termination, termination.positions, position
  106. ))
  107. try:
  108. peer_port = FrontPort.objects.get(
  109. rear_port=termination,
  110. rear_port_position=position,
  111. )
  112. return peer_port, 1
  113. except ObjectDoesNotExist:
  114. return None, None
  115. # Follow a circuit to its other termination
  116. elif isinstance(termination, CircuitTermination) and follow_circuits:
  117. peer_termination = termination.get_peer_termination()
  118. if peer_termination is None:
  119. return None, None
  120. return peer_termination, position
  121. # Termination is not a pass-through port
  122. else:
  123. return None, None
  124. if not self.cable:
  125. return [(self, None, None)]
  126. # Record cable history to detect loops
  127. if cable_history is None:
  128. cable_history = []
  129. elif self.cable in cable_history:
  130. raise LoopDetected()
  131. cable_history.append(self.cable)
  132. far_end = self.cable.termination_b if self.cable.termination_a == self else self.cable.termination_a
  133. path = [(self, self.cable, far_end)]
  134. peer_port, position = get_peer_port(far_end, position, follow_circuits)
  135. if peer_port is None:
  136. return path
  137. try:
  138. next_segment = peer_port.trace(position, follow_circuits, cable_history)
  139. except LoopDetected:
  140. return path
  141. if next_segment is None:
  142. return path + [(peer_port, None, None)]
  143. return path + next_segment
  144. def get_cable_peer(self):
  145. if self.cable is None:
  146. return None
  147. if self._cabled_as_a.exists():
  148. return self.cable.termination_b
  149. if self._cabled_as_b.exists():
  150. return self.cable.termination_a
  151. #
  152. # Regions
  153. #
  154. class Region(MPTTModel, ChangeLoggedModel):
  155. """
  156. Sites can be grouped within geographic Regions.
  157. """
  158. parent = TreeForeignKey(
  159. to='self',
  160. on_delete=models.CASCADE,
  161. related_name='children',
  162. blank=True,
  163. null=True,
  164. db_index=True
  165. )
  166. name = models.CharField(
  167. max_length=50,
  168. unique=True
  169. )
  170. slug = models.SlugField(
  171. unique=True
  172. )
  173. csv_headers = ['name', 'slug', 'parent']
  174. class MPTTMeta:
  175. order_insertion_by = ['name']
  176. def __str__(self):
  177. return self.name
  178. def get_absolute_url(self):
  179. return "{}?region={}".format(reverse('dcim:site_list'), self.slug)
  180. def to_csv(self):
  181. return (
  182. self.name,
  183. self.slug,
  184. self.parent.name if self.parent else None,
  185. )
  186. def get_site_count(self):
  187. return Site.objects.filter(
  188. Q(region=self) |
  189. Q(region__in=self.get_descendants())
  190. ).count()
  191. #
  192. # Sites
  193. #
  194. class Site(ChangeLoggedModel, CustomFieldModel):
  195. """
  196. A Site represents a geographic location within a network; typically a building or campus. The optional facility
  197. field can be used to include an external designation, such as a data center name (e.g. Equinix SV6).
  198. """
  199. name = models.CharField(
  200. max_length=50,
  201. unique=True
  202. )
  203. slug = models.SlugField(
  204. unique=True
  205. )
  206. status = models.PositiveSmallIntegerField(
  207. choices=SITE_STATUS_CHOICES,
  208. default=SITE_STATUS_ACTIVE
  209. )
  210. region = models.ForeignKey(
  211. to='dcim.Region',
  212. on_delete=models.SET_NULL,
  213. related_name='sites',
  214. blank=True,
  215. null=True
  216. )
  217. tenant = models.ForeignKey(
  218. to='tenancy.Tenant',
  219. on_delete=models.PROTECT,
  220. related_name='sites',
  221. blank=True,
  222. null=True
  223. )
  224. facility = models.CharField(
  225. max_length=50,
  226. blank=True
  227. )
  228. asn = ASNField(
  229. blank=True,
  230. null=True,
  231. verbose_name='ASN'
  232. )
  233. time_zone = TimeZoneField(
  234. blank=True
  235. )
  236. description = models.CharField(
  237. max_length=100,
  238. blank=True
  239. )
  240. physical_address = models.CharField(
  241. max_length=200,
  242. blank=True
  243. )
  244. shipping_address = models.CharField(
  245. max_length=200,
  246. blank=True
  247. )
  248. latitude = models.DecimalField(
  249. max_digits=8,
  250. decimal_places=6,
  251. blank=True,
  252. null=True
  253. )
  254. longitude = models.DecimalField(
  255. max_digits=9,
  256. decimal_places=6,
  257. blank=True,
  258. null=True
  259. )
  260. contact_name = models.CharField(
  261. max_length=50,
  262. blank=True
  263. )
  264. contact_phone = models.CharField(
  265. max_length=20,
  266. blank=True
  267. )
  268. contact_email = models.EmailField(
  269. blank=True,
  270. verbose_name='Contact E-mail'
  271. )
  272. comments = models.TextField(
  273. blank=True
  274. )
  275. custom_field_values = GenericRelation(
  276. to='extras.CustomFieldValue',
  277. content_type_field='obj_type',
  278. object_id_field='obj_id'
  279. )
  280. images = GenericRelation(
  281. to='extras.ImageAttachment'
  282. )
  283. objects = NaturalOrderingManager()
  284. tags = TaggableManager(through=TaggedItem)
  285. csv_headers = [
  286. 'name', 'slug', 'status', 'region', 'tenant', 'facility', 'asn', 'time_zone', 'description', 'physical_address',
  287. 'shipping_address', 'latitude', 'longitude', 'contact_name', 'contact_phone', 'contact_email', 'comments',
  288. ]
  289. class Meta:
  290. ordering = ['name']
  291. def __str__(self):
  292. return self.name
  293. def get_absolute_url(self):
  294. return reverse('dcim:site', args=[self.slug])
  295. def to_csv(self):
  296. return (
  297. self.name,
  298. self.slug,
  299. self.get_status_display(),
  300. self.region.name if self.region else None,
  301. self.tenant.name if self.tenant else None,
  302. self.facility,
  303. self.asn,
  304. self.time_zone,
  305. self.description,
  306. self.physical_address,
  307. self.shipping_address,
  308. self.latitude,
  309. self.longitude,
  310. self.contact_name,
  311. self.contact_phone,
  312. self.contact_email,
  313. self.comments,
  314. )
  315. def get_status_class(self):
  316. return STATUS_CLASSES[self.status]
  317. #
  318. # Racks
  319. #
  320. class RackGroup(ChangeLoggedModel):
  321. """
  322. Racks can be grouped as subsets within a Site. The scope of a group will depend on how Sites are defined. For
  323. example, if a Site spans a corporate campus, a RackGroup might be defined to represent each building within that
  324. campus. If a Site instead represents a single building, a RackGroup might represent a single room or floor.
  325. """
  326. name = models.CharField(
  327. max_length=50
  328. )
  329. slug = models.SlugField()
  330. site = models.ForeignKey(
  331. to='dcim.Site',
  332. on_delete=models.CASCADE,
  333. related_name='rack_groups'
  334. )
  335. csv_headers = ['site', 'name', 'slug']
  336. class Meta:
  337. ordering = ['site', 'name']
  338. unique_together = [
  339. ['site', 'name'],
  340. ['site', 'slug'],
  341. ]
  342. def __str__(self):
  343. return self.name
  344. def get_absolute_url(self):
  345. return "{}?group_id={}".format(reverse('dcim:rack_list'), self.pk)
  346. def to_csv(self):
  347. return (
  348. self.site,
  349. self.name,
  350. self.slug,
  351. )
  352. class RackRole(ChangeLoggedModel):
  353. """
  354. Racks can be organized by functional role, similar to Devices.
  355. """
  356. name = models.CharField(
  357. max_length=50,
  358. unique=True
  359. )
  360. slug = models.SlugField(
  361. unique=True
  362. )
  363. color = ColorField()
  364. csv_headers = ['name', 'slug', 'color']
  365. class Meta:
  366. ordering = ['name']
  367. def __str__(self):
  368. return self.name
  369. def get_absolute_url(self):
  370. return "{}?role={}".format(reverse('dcim:rack_list'), self.slug)
  371. def to_csv(self):
  372. return (
  373. self.name,
  374. self.slug,
  375. self.color,
  376. )
  377. class Rack(ChangeLoggedModel, CustomFieldModel):
  378. """
  379. Devices are housed within Racks. Each rack has a defined height measured in rack units, and a front and rear face.
  380. Each Rack is assigned to a Site and (optionally) a RackGroup.
  381. """
  382. name = models.CharField(
  383. max_length=50
  384. )
  385. facility_id = models.CharField(
  386. max_length=50,
  387. blank=True,
  388. null=True,
  389. verbose_name='Facility ID'
  390. )
  391. site = models.ForeignKey(
  392. to='dcim.Site',
  393. on_delete=models.PROTECT,
  394. related_name='racks'
  395. )
  396. group = models.ForeignKey(
  397. to='dcim.RackGroup',
  398. on_delete=models.SET_NULL,
  399. related_name='racks',
  400. blank=True,
  401. null=True
  402. )
  403. tenant = models.ForeignKey(
  404. to='tenancy.Tenant',
  405. on_delete=models.PROTECT,
  406. related_name='racks',
  407. blank=True,
  408. null=True
  409. )
  410. status = models.PositiveSmallIntegerField(
  411. choices=RACK_STATUS_CHOICES,
  412. default=RACK_STATUS_ACTIVE
  413. )
  414. role = models.ForeignKey(
  415. to='dcim.RackRole',
  416. on_delete=models.PROTECT,
  417. related_name='racks',
  418. blank=True,
  419. null=True
  420. )
  421. serial = models.CharField(
  422. max_length=50,
  423. blank=True,
  424. verbose_name='Serial number'
  425. )
  426. asset_tag = models.CharField(
  427. max_length=50,
  428. blank=True,
  429. null=True,
  430. unique=True,
  431. verbose_name='Asset tag',
  432. help_text='A unique tag used to identify this rack'
  433. )
  434. type = models.PositiveSmallIntegerField(
  435. choices=RACK_TYPE_CHOICES,
  436. blank=True,
  437. null=True,
  438. verbose_name='Type'
  439. )
  440. width = models.PositiveSmallIntegerField(
  441. choices=RACK_WIDTH_CHOICES,
  442. default=RACK_WIDTH_19IN,
  443. verbose_name='Width',
  444. help_text='Rail-to-rail width'
  445. )
  446. u_height = models.PositiveSmallIntegerField(
  447. default=42,
  448. verbose_name='Height (U)',
  449. validators=[MinValueValidator(1), MaxValueValidator(100)]
  450. )
  451. desc_units = models.BooleanField(
  452. default=False,
  453. verbose_name='Descending units',
  454. help_text='Units are numbered top-to-bottom'
  455. )
  456. outer_width = models.PositiveSmallIntegerField(
  457. blank=True,
  458. null=True
  459. )
  460. outer_depth = models.PositiveSmallIntegerField(
  461. blank=True,
  462. null=True
  463. )
  464. outer_unit = models.PositiveSmallIntegerField(
  465. choices=RACK_DIMENSION_UNIT_CHOICES,
  466. blank=True,
  467. null=True
  468. )
  469. comments = models.TextField(
  470. blank=True
  471. )
  472. custom_field_values = GenericRelation(
  473. to='extras.CustomFieldValue',
  474. content_type_field='obj_type',
  475. object_id_field='obj_id'
  476. )
  477. images = GenericRelation(
  478. to='extras.ImageAttachment'
  479. )
  480. objects = NaturalOrderingManager()
  481. tags = TaggableManager(through=TaggedItem)
  482. csv_headers = [
  483. 'site', 'group_name', 'name', 'facility_id', 'tenant', 'status', 'role', 'type', 'serial', 'asset_tag', 'width',
  484. 'u_height', 'desc_units', 'outer_width', 'outer_depth', 'outer_unit', 'comments',
  485. ]
  486. class Meta:
  487. ordering = ['site', 'group', 'name']
  488. unique_together = [
  489. ['group', 'name'],
  490. ['group', 'facility_id'],
  491. ]
  492. def __str__(self):
  493. return self.display_name or super().__str__()
  494. def get_absolute_url(self):
  495. return reverse('dcim:rack', args=[self.pk])
  496. def clean(self):
  497. # Validate outer dimensions and unit
  498. if (self.outer_width is not None or self.outer_depth is not None) and self.outer_unit is None:
  499. raise ValidationError("Must specify a unit when setting an outer width/depth")
  500. elif self.outer_width is None and self.outer_depth is None:
  501. self.outer_unit = None
  502. if self.pk:
  503. # Validate that Rack is tall enough to house the installed Devices
  504. top_device = Device.objects.filter(rack=self).exclude(position__isnull=True).order_by('-position').first()
  505. if top_device:
  506. min_height = top_device.position + top_device.device_type.u_height - 1
  507. if self.u_height < min_height:
  508. raise ValidationError({
  509. 'u_height': "Rack must be at least {}U tall to house currently installed devices.".format(
  510. min_height
  511. )
  512. })
  513. # Validate that Rack was assigned a group of its same site, if applicable
  514. if self.group:
  515. if self.group.site != self.site:
  516. raise ValidationError({
  517. 'group': "Rack group must be from the same site, {}.".format(self.site)
  518. })
  519. def save(self, *args, **kwargs):
  520. # Record the original site assignment for this rack.
  521. _site_id = None
  522. if self.pk:
  523. _site_id = Rack.objects.get(pk=self.pk).site_id
  524. super().save(*args, **kwargs)
  525. # Update racked devices if the assigned Site has been changed.
  526. if _site_id is not None and self.site_id != _site_id:
  527. devices = Device.objects.filter(rack=self)
  528. for device in devices:
  529. device.site = self.site
  530. device.save()
  531. def to_csv(self):
  532. return (
  533. self.site.name,
  534. self.group.name if self.group else None,
  535. self.name,
  536. self.facility_id,
  537. self.tenant.name if self.tenant else None,
  538. self.get_status_display(),
  539. self.role.name if self.role else None,
  540. self.get_type_display() if self.type else None,
  541. self.serial,
  542. self.asset_tag,
  543. self.width,
  544. self.u_height,
  545. self.desc_units,
  546. self.outer_width,
  547. self.outer_depth,
  548. self.outer_unit,
  549. self.comments,
  550. )
  551. @property
  552. def units(self):
  553. if self.desc_units:
  554. return range(1, self.u_height + 1)
  555. else:
  556. return reversed(range(1, self.u_height + 1))
  557. @property
  558. def display_name(self):
  559. if self.facility_id:
  560. return "{} ({})".format(self.name, self.facility_id)
  561. elif self.name:
  562. return self.name
  563. return ""
  564. def get_status_class(self):
  565. return STATUS_CLASSES[self.status]
  566. def get_rack_units(self, face=RACK_FACE_FRONT, exclude=None, remove_redundant=False):
  567. """
  568. Return a list of rack units as dictionaries. Example: {'device': None, 'face': 0, 'id': 48, 'name': 'U48'}
  569. Each key 'device' is either a Device or None. By default, multi-U devices are repeated for each U they occupy.
  570. :param face: Rack face (front or rear)
  571. :param exclude: PK of a Device to exclude (optional); helpful when relocating a Device within a Rack
  572. :param remove_redundant: If True, rack units occupied by a device already listed will be omitted
  573. """
  574. elevation = OrderedDict()
  575. for u in self.units:
  576. elevation[u] = {'id': u, 'name': 'U{}'.format(u), 'face': face, 'device': None}
  577. # Add devices to rack units list
  578. if self.pk:
  579. for device in Device.objects.prefetch_related('device_type__manufacturer', 'device_role')\
  580. .annotate(devicebay_count=Count('device_bays'))\
  581. .exclude(pk=exclude)\
  582. .filter(rack=self, position__gt=0)\
  583. .filter(Q(face=face) | Q(device_type__is_full_depth=True)):
  584. if remove_redundant:
  585. elevation[device.position]['device'] = device
  586. for u in range(device.position + 1, device.position + device.device_type.u_height):
  587. elevation.pop(u, None)
  588. else:
  589. for u in range(device.position, device.position + device.device_type.u_height):
  590. elevation[u]['device'] = device
  591. return [u for u in elevation.values()]
  592. def get_front_elevation(self):
  593. return self.get_rack_units(face=RACK_FACE_FRONT, remove_redundant=True)
  594. def get_rear_elevation(self):
  595. return self.get_rack_units(face=RACK_FACE_REAR, remove_redundant=True)
  596. def get_available_units(self, u_height=1, rack_face=None, exclude=list()):
  597. """
  598. Return a list of units within the rack available to accommodate a device of a given U height (default 1).
  599. Optionally exclude one or more devices when calculating empty units (needed when moving a device from one
  600. position to another within a rack).
  601. :param u_height: Minimum number of contiguous free units required
  602. :param rack_face: The face of the rack (front or rear) required; 'None' if device is full depth
  603. :param exclude: List of devices IDs to exclude (useful when moving a device within a rack)
  604. """
  605. # Gather all devices which consume U space within the rack
  606. devices = self.devices.prefetch_related('device_type').filter(position__gte=1).exclude(pk__in=exclude)
  607. # Initialize the rack unit skeleton
  608. units = list(range(1, self.u_height + 1))
  609. # Remove units consumed by installed devices
  610. for d in devices:
  611. if rack_face is None or d.face == rack_face or d.device_type.is_full_depth:
  612. for u in range(d.position, d.position + d.device_type.u_height):
  613. try:
  614. units.remove(u)
  615. except ValueError:
  616. # Found overlapping devices in the rack!
  617. pass
  618. # Remove units without enough space above them to accommodate a device of the specified height
  619. available_units = []
  620. for u in units:
  621. if set(range(u, u + u_height)).issubset(units):
  622. available_units.append(u)
  623. return list(reversed(available_units))
  624. def get_reserved_units(self):
  625. """
  626. Return a dictionary mapping all reserved units within the rack to their reservation.
  627. """
  628. reserved_units = {}
  629. for r in self.reservations.all():
  630. for u in r.units:
  631. reserved_units[u] = r
  632. return reserved_units
  633. def get_0u_devices(self):
  634. return self.devices.filter(position=0)
  635. def get_utilization(self):
  636. """
  637. Determine the utilization rate of the rack and return it as a percentage. Occupied and reserved units both count
  638. as utilized.
  639. """
  640. # Determine unoccupied units
  641. available_units = self.get_available_units()
  642. # Remove reserved units
  643. for u in self.get_reserved_units():
  644. if u in available_units:
  645. available_units.remove(u)
  646. occupied_unit_count = self.u_height - len(available_units)
  647. percentage = int(float(occupied_unit_count) / self.u_height * 100)
  648. return percentage
  649. def get_power_utilization(self):
  650. """
  651. Determine the utilization rate of power in the rack and return it as a percentage.
  652. """
  653. power_stats = PowerFeed.objects.filter(
  654. rack=self
  655. ).annotate(
  656. allocated_draw_total=Sum('connected_endpoint__poweroutlets__connected_endpoint__allocated_draw'),
  657. ).values(
  658. 'allocated_draw_total',
  659. 'available_power'
  660. )
  661. if power_stats:
  662. allocated_draw_total = sum(x['allocated_draw_total'] for x in power_stats)
  663. available_power_total = sum(x['available_power'] for x in power_stats)
  664. return int(allocated_draw_total / available_power_total * 100) or 0
  665. return 0
  666. class RackReservation(ChangeLoggedModel):
  667. """
  668. One or more reserved units within a Rack.
  669. """
  670. rack = models.ForeignKey(
  671. to='dcim.Rack',
  672. on_delete=models.CASCADE,
  673. related_name='reservations'
  674. )
  675. units = ArrayField(
  676. base_field=models.PositiveSmallIntegerField()
  677. )
  678. tenant = models.ForeignKey(
  679. to='tenancy.Tenant',
  680. on_delete=models.PROTECT,
  681. related_name='rackreservations',
  682. blank=True,
  683. null=True
  684. )
  685. user = models.ForeignKey(
  686. to=User,
  687. on_delete=models.PROTECT
  688. )
  689. description = models.CharField(
  690. max_length=100
  691. )
  692. class Meta:
  693. ordering = ['created']
  694. def __str__(self):
  695. return "Reservation for rack {}".format(self.rack)
  696. def clean(self):
  697. if self.units:
  698. # Validate that all specified units exist in the Rack.
  699. invalid_units = [u for u in self.units if u not in self.rack.units]
  700. if invalid_units:
  701. raise ValidationError({
  702. 'units': "Invalid unit(s) for {}U rack: {}".format(
  703. self.rack.u_height,
  704. ', '.join([str(u) for u in invalid_units]),
  705. ),
  706. })
  707. # Check that none of the units has already been reserved for this Rack.
  708. reserved_units = []
  709. for resv in self.rack.reservations.exclude(pk=self.pk):
  710. reserved_units += resv.units
  711. conflicting_units = [u for u in self.units if u in reserved_units]
  712. if conflicting_units:
  713. raise ValidationError({
  714. 'units': 'The following units have already been reserved: {}'.format(
  715. ', '.join([str(u) for u in conflicting_units]),
  716. )
  717. })
  718. @property
  719. def unit_list(self):
  720. """
  721. Express the assigned units as a string of summarized ranges. For example:
  722. [0, 1, 2, 10, 14, 15, 16] => "0-2, 10, 14-16"
  723. """
  724. group = (list(x) for _, x in groupby(sorted(self.units), lambda x, c=count(): next(c) - x))
  725. return ', '.join('-'.join(map(str, (g[0], g[-1])[:len(g)])) for g in group)
  726. #
  727. # Device Types
  728. #
  729. class Manufacturer(ChangeLoggedModel):
  730. """
  731. A Manufacturer represents a company which produces hardware devices; for example, Juniper or Dell.
  732. """
  733. name = models.CharField(
  734. max_length=50,
  735. unique=True
  736. )
  737. slug = models.SlugField(
  738. unique=True
  739. )
  740. csv_headers = ['name', 'slug']
  741. class Meta:
  742. ordering = ['name']
  743. def __str__(self):
  744. return self.name
  745. def get_absolute_url(self):
  746. return "{}?manufacturer={}".format(reverse('dcim:devicetype_list'), self.slug)
  747. def to_csv(self):
  748. return (
  749. self.name,
  750. self.slug,
  751. )
  752. class DeviceType(ChangeLoggedModel, CustomFieldModel):
  753. """
  754. A DeviceType represents a particular make (Manufacturer) and model of device. It specifies rack height and depth, as
  755. well as high-level functional role(s).
  756. Each DeviceType can have an arbitrary number of component templates assigned to it, which define console, power, and
  757. interface objects. For example, a Juniper EX4300-48T DeviceType would have:
  758. * 1 ConsolePortTemplate
  759. * 2 PowerPortTemplates
  760. * 48 InterfaceTemplates
  761. When a new Device of this type is created, the appropriate console, power, and interface objects (as defined by the
  762. DeviceType) are automatically created as well.
  763. """
  764. manufacturer = models.ForeignKey(
  765. to='dcim.Manufacturer',
  766. on_delete=models.PROTECT,
  767. related_name='device_types'
  768. )
  769. model = models.CharField(
  770. max_length=50
  771. )
  772. slug = models.SlugField()
  773. part_number = models.CharField(
  774. max_length=50,
  775. blank=True,
  776. help_text='Discrete part number (optional)'
  777. )
  778. u_height = models.PositiveSmallIntegerField(
  779. default=1,
  780. verbose_name='Height (U)'
  781. )
  782. is_full_depth = models.BooleanField(
  783. default=True,
  784. verbose_name='Is full depth',
  785. help_text='Device consumes both front and rear rack faces'
  786. )
  787. subdevice_role = models.NullBooleanField(
  788. default=None,
  789. verbose_name='Parent/child status',
  790. choices=SUBDEVICE_ROLE_CHOICES,
  791. help_text='Parent devices house child devices in device bays. Select '
  792. '"None" if this device type is neither a parent nor a child.'
  793. )
  794. comments = models.TextField(
  795. blank=True
  796. )
  797. custom_field_values = GenericRelation(
  798. to='extras.CustomFieldValue',
  799. content_type_field='obj_type',
  800. object_id_field='obj_id'
  801. )
  802. tags = TaggableManager(through=TaggedItem)
  803. csv_headers = [
  804. 'manufacturer', 'model', 'slug', 'part_number', 'u_height', 'is_full_depth', 'subdevice_role', 'comments',
  805. ]
  806. class Meta:
  807. ordering = ['manufacturer', 'model']
  808. unique_together = [
  809. ['manufacturer', 'model'],
  810. ['manufacturer', 'slug'],
  811. ]
  812. def __str__(self):
  813. return self.model
  814. def __init__(self, *args, **kwargs):
  815. super().__init__(*args, **kwargs)
  816. # Save a copy of u_height for validation in clean()
  817. self._original_u_height = self.u_height
  818. def get_absolute_url(self):
  819. return reverse('dcim:devicetype', args=[self.pk])
  820. def to_csv(self):
  821. return (
  822. self.manufacturer.name,
  823. self.model,
  824. self.slug,
  825. self.part_number,
  826. self.u_height,
  827. self.is_full_depth,
  828. self.get_subdevice_role_display() if self.subdevice_role else None,
  829. self.comments,
  830. )
  831. def clean(self):
  832. # If editing an existing DeviceType to have a larger u_height, first validate that *all* instances of it have
  833. # room to expand within their racks. This validation will impose a very high performance penalty when there are
  834. # many instances to check, but increasing the u_height of a DeviceType should be a very rare occurrence.
  835. if self.pk is not None and self.u_height > self._original_u_height:
  836. for d in Device.objects.filter(device_type=self, position__isnull=False):
  837. face_required = None if self.is_full_depth else d.face
  838. u_available = d.rack.get_available_units(u_height=self.u_height, rack_face=face_required,
  839. exclude=[d.pk])
  840. if d.position not in u_available:
  841. raise ValidationError({
  842. 'u_height': "Device {} in rack {} does not have sufficient space to accommodate a height of "
  843. "{}U".format(d, d.rack, self.u_height)
  844. })
  845. if self.subdevice_role != SUBDEVICE_ROLE_PARENT and self.device_bay_templates.count():
  846. raise ValidationError({
  847. 'subdevice_role': "Must delete all device bay templates associated with this device before "
  848. "declassifying it as a parent device."
  849. })
  850. if self.u_height and self.subdevice_role == SUBDEVICE_ROLE_CHILD:
  851. raise ValidationError({
  852. 'u_height': "Child device types must be 0U."
  853. })
  854. @property
  855. def display_name(self):
  856. return '{} {}'.format(self.manufacturer.name, self.model)
  857. @property
  858. def is_parent_device(self):
  859. return bool(self.subdevice_role)
  860. @property
  861. def is_child_device(self):
  862. return bool(self.subdevice_role is False)
  863. class ConsolePortTemplate(ComponentTemplateModel):
  864. """
  865. A template for a ConsolePort to be created for a new Device.
  866. """
  867. device_type = models.ForeignKey(
  868. to='dcim.DeviceType',
  869. on_delete=models.CASCADE,
  870. related_name='consoleport_templates'
  871. )
  872. name = models.CharField(
  873. max_length=50
  874. )
  875. objects = NaturalOrderingManager()
  876. class Meta:
  877. ordering = ['device_type', 'name']
  878. unique_together = ['device_type', 'name']
  879. def __str__(self):
  880. return self.name
  881. def instantiate(self, device):
  882. return ConsolePort(
  883. device=device,
  884. name=self.name
  885. )
  886. class ConsoleServerPortTemplate(ComponentTemplateModel):
  887. """
  888. A template for a ConsoleServerPort to be created for a new Device.
  889. """
  890. device_type = models.ForeignKey(
  891. to='dcim.DeviceType',
  892. on_delete=models.CASCADE,
  893. related_name='consoleserverport_templates'
  894. )
  895. name = models.CharField(
  896. max_length=50
  897. )
  898. objects = NaturalOrderingManager()
  899. class Meta:
  900. ordering = ['device_type', 'name']
  901. unique_together = ['device_type', 'name']
  902. def __str__(self):
  903. return self.name
  904. def instantiate(self, device):
  905. return ConsoleServerPort(
  906. device=device,
  907. name=self.name
  908. )
  909. class PowerPortTemplate(ComponentTemplateModel):
  910. """
  911. A template for a PowerPort to be created for a new Device.
  912. """
  913. device_type = models.ForeignKey(
  914. to='dcim.DeviceType',
  915. on_delete=models.CASCADE,
  916. related_name='powerport_templates'
  917. )
  918. name = models.CharField(
  919. max_length=50
  920. )
  921. maximum_draw = models.PositiveSmallIntegerField(
  922. blank=True,
  923. null=True,
  924. validators=[MinValueValidator(1)],
  925. help_text="Maximum current draw (watts)"
  926. )
  927. allocated_draw = models.PositiveSmallIntegerField(
  928. blank=True,
  929. null=True,
  930. validators=[MinValueValidator(1)],
  931. help_text="Allocated current draw (watts)"
  932. )
  933. objects = NaturalOrderingManager()
  934. class Meta:
  935. ordering = ['device_type', 'name']
  936. unique_together = ['device_type', 'name']
  937. def __str__(self):
  938. return self.name
  939. def instantiate(self, device):
  940. return PowerPort(
  941. device=device,
  942. name=self.name,
  943. maximum_draw=self.maximum_draw,
  944. allocated_draw=self.allocated_draw
  945. )
  946. class PowerOutletTemplate(ComponentTemplateModel):
  947. """
  948. A template for a PowerOutlet to be created for a new Device.
  949. """
  950. device_type = models.ForeignKey(
  951. to='dcim.DeviceType',
  952. on_delete=models.CASCADE,
  953. related_name='poweroutlet_templates'
  954. )
  955. name = models.CharField(
  956. max_length=50
  957. )
  958. power_port = models.ForeignKey(
  959. to='dcim.PowerPortTemplate',
  960. on_delete=models.SET_NULL,
  961. blank=True,
  962. null=True,
  963. related_name='poweroutlet_templates'
  964. )
  965. feed_leg = models.PositiveSmallIntegerField(
  966. choices=POWERFEED_LEG_CHOICES,
  967. blank=True,
  968. null=True,
  969. help_text="Phase (for three-phase feeds)"
  970. )
  971. objects = NaturalOrderingManager()
  972. class Meta:
  973. ordering = ['device_type', 'name']
  974. unique_together = ['device_type', 'name']
  975. def __str__(self):
  976. return self.name
  977. def clean(self):
  978. # Validate power port assignment
  979. if self.power_port and self.power_port.device_type != self.device_type:
  980. raise ValidationError(
  981. "Parent power port ({}) must belong to the same device type".format(self.power_port)
  982. )
  983. def instantiate(self, device):
  984. if self.power_port:
  985. power_port = PowerPort.objects.get(device=device, name=self.power_port.name)
  986. else:
  987. power_port = None
  988. return PowerOutlet(
  989. device=device,
  990. name=self.name,
  991. power_port=power_port,
  992. feed_leg=self.feed_leg
  993. )
  994. class InterfaceTemplate(ComponentTemplateModel):
  995. """
  996. A template for a physical data interface on a new Device.
  997. """
  998. device_type = models.ForeignKey(
  999. to='dcim.DeviceType',
  1000. on_delete=models.CASCADE,
  1001. related_name='interface_templates'
  1002. )
  1003. name = models.CharField(
  1004. max_length=64
  1005. )
  1006. type = models.PositiveSmallIntegerField(
  1007. choices=IFACE_TYPE_CHOICES,
  1008. default=IFACE_TYPE_10GE_SFP_PLUS
  1009. )
  1010. mgmt_only = models.BooleanField(
  1011. default=False,
  1012. verbose_name='Management only'
  1013. )
  1014. objects = InterfaceManager()
  1015. class Meta:
  1016. ordering = ['device_type', 'name']
  1017. unique_together = ['device_type', 'name']
  1018. def __str__(self):
  1019. return self.name
  1020. # TODO: Remove in v2.7
  1021. @property
  1022. def form_factor(self):
  1023. """
  1024. Backward-compatibility for form_factor
  1025. """
  1026. return self.type
  1027. # TODO: Remove in v2.7
  1028. @form_factor.setter
  1029. def form_factor(self, value):
  1030. """
  1031. Backward-compatibility for form_factor
  1032. """
  1033. self.type = value
  1034. def instantiate(self, device):
  1035. return Interface(
  1036. device=device,
  1037. name=self.name,
  1038. type=self.type,
  1039. mgmt_only=self.mgmt_only
  1040. )
  1041. class FrontPortTemplate(ComponentTemplateModel):
  1042. """
  1043. Template for a pass-through port on the front of a new Device.
  1044. """
  1045. device_type = models.ForeignKey(
  1046. to='dcim.DeviceType',
  1047. on_delete=models.CASCADE,
  1048. related_name='frontport_templates'
  1049. )
  1050. name = models.CharField(
  1051. max_length=64
  1052. )
  1053. type = models.PositiveSmallIntegerField(
  1054. choices=PORT_TYPE_CHOICES
  1055. )
  1056. rear_port = models.ForeignKey(
  1057. to='dcim.RearPortTemplate',
  1058. on_delete=models.CASCADE,
  1059. related_name='frontport_templates'
  1060. )
  1061. rear_port_position = models.PositiveSmallIntegerField(
  1062. default=1,
  1063. validators=[MinValueValidator(1), MaxValueValidator(64)]
  1064. )
  1065. objects = NaturalOrderingManager()
  1066. class Meta:
  1067. ordering = ['device_type', 'name']
  1068. unique_together = [
  1069. ['device_type', 'name'],
  1070. ['rear_port', 'rear_port_position'],
  1071. ]
  1072. def __str__(self):
  1073. return self.name
  1074. def clean(self):
  1075. # Validate rear port assignment
  1076. if self.rear_port.device_type != self.device_type:
  1077. raise ValidationError(
  1078. "Rear port ({}) must belong to the same device type".format(self.rear_port)
  1079. )
  1080. # Validate rear port position assignment
  1081. if self.rear_port_position > self.rear_port.positions:
  1082. raise ValidationError(
  1083. "Invalid rear port position ({}); rear port {} has only {} positions".format(
  1084. self.rear_port_position, self.rear_port.name, self.rear_port.positions
  1085. )
  1086. )
  1087. def instantiate(self, device):
  1088. if self.rear_port:
  1089. rear_port = RearPort.objects.get(device=device, name=self.rear_port.name)
  1090. else:
  1091. rear_port = None
  1092. return FrontPort(
  1093. device=device,
  1094. name=self.name,
  1095. type=self.type,
  1096. rear_port=rear_port,
  1097. rear_port_position=self.rear_port_position
  1098. )
  1099. class RearPortTemplate(ComponentTemplateModel):
  1100. """
  1101. Template for a pass-through port on the rear of a new Device.
  1102. """
  1103. device_type = models.ForeignKey(
  1104. to='dcim.DeviceType',
  1105. on_delete=models.CASCADE,
  1106. related_name='rearport_templates'
  1107. )
  1108. name = models.CharField(
  1109. max_length=64
  1110. )
  1111. type = models.PositiveSmallIntegerField(
  1112. choices=PORT_TYPE_CHOICES
  1113. )
  1114. positions = models.PositiveSmallIntegerField(
  1115. default=1,
  1116. validators=[MinValueValidator(1), MaxValueValidator(64)]
  1117. )
  1118. objects = NaturalOrderingManager()
  1119. class Meta:
  1120. ordering = ['device_type', 'name']
  1121. unique_together = ['device_type', 'name']
  1122. def __str__(self):
  1123. return self.name
  1124. def instantiate(self, device):
  1125. return RearPort(
  1126. device=device,
  1127. name=self.name,
  1128. type=self.type,
  1129. positions=self.positions
  1130. )
  1131. class DeviceBayTemplate(ComponentTemplateModel):
  1132. """
  1133. A template for a DeviceBay to be created for a new parent Device.
  1134. """
  1135. device_type = models.ForeignKey(
  1136. to='dcim.DeviceType',
  1137. on_delete=models.CASCADE,
  1138. related_name='device_bay_templates'
  1139. )
  1140. name = models.CharField(
  1141. max_length=50
  1142. )
  1143. objects = NaturalOrderingManager()
  1144. class Meta:
  1145. ordering = ['device_type', 'name']
  1146. unique_together = ['device_type', 'name']
  1147. def __str__(self):
  1148. return self.name
  1149. def instantiate(self, device):
  1150. return DeviceBay(
  1151. device=device,
  1152. name=self.name
  1153. )
  1154. #
  1155. # Devices
  1156. #
  1157. class DeviceRole(ChangeLoggedModel):
  1158. """
  1159. Devices are organized by functional role; for example, "Core Switch" or "File Server". Each DeviceRole is assigned a
  1160. color to be used when displaying rack elevations. The vm_role field determines whether the role is applicable to
  1161. virtual machines as well.
  1162. """
  1163. name = models.CharField(
  1164. max_length=50,
  1165. unique=True
  1166. )
  1167. slug = models.SlugField(
  1168. unique=True
  1169. )
  1170. color = ColorField()
  1171. vm_role = models.BooleanField(
  1172. default=True,
  1173. verbose_name='VM Role',
  1174. help_text='Virtual machines may be assigned to this role'
  1175. )
  1176. csv_headers = ['name', 'slug', 'color', 'vm_role']
  1177. class Meta:
  1178. ordering = ['name']
  1179. def __str__(self):
  1180. return self.name
  1181. def to_csv(self):
  1182. return (
  1183. self.name,
  1184. self.slug,
  1185. self.color,
  1186. self.vm_role,
  1187. )
  1188. class Platform(ChangeLoggedModel):
  1189. """
  1190. Platform refers to the software or firmware running on a Device. For example, "Cisco IOS-XR" or "Juniper Junos".
  1191. NetBox uses Platforms to determine how to interact with devices when pulling inventory data or other information by
  1192. specifying a NAPALM driver.
  1193. """
  1194. name = models.CharField(
  1195. max_length=100,
  1196. unique=True
  1197. )
  1198. slug = models.SlugField(
  1199. unique=True,
  1200. max_length=100
  1201. )
  1202. manufacturer = models.ForeignKey(
  1203. to='dcim.Manufacturer',
  1204. on_delete=models.PROTECT,
  1205. related_name='platforms',
  1206. blank=True,
  1207. null=True,
  1208. help_text='Optionally limit this platform to devices of a certain manufacturer'
  1209. )
  1210. napalm_driver = models.CharField(
  1211. max_length=50,
  1212. blank=True,
  1213. verbose_name='NAPALM driver',
  1214. help_text='The name of the NAPALM driver to use when interacting with devices'
  1215. )
  1216. napalm_args = JSONField(
  1217. blank=True,
  1218. null=True,
  1219. verbose_name='NAPALM arguments',
  1220. help_text='Additional arguments to pass when initiating the NAPALM driver (JSON format)'
  1221. )
  1222. csv_headers = ['name', 'slug', 'manufacturer', 'napalm_driver', 'napalm_args']
  1223. class Meta:
  1224. ordering = ['name']
  1225. def __str__(self):
  1226. return self.name
  1227. def get_absolute_url(self):
  1228. return "{}?platform={}".format(reverse('dcim:device_list'), self.slug)
  1229. def to_csv(self):
  1230. return (
  1231. self.name,
  1232. self.slug,
  1233. self.manufacturer.name if self.manufacturer else None,
  1234. self.napalm_driver,
  1235. self.napalm_args,
  1236. )
  1237. class Device(ChangeLoggedModel, ConfigContextModel, CustomFieldModel):
  1238. """
  1239. A Device represents a piece of physical hardware mounted within a Rack. Each Device is assigned a DeviceType,
  1240. DeviceRole, and (optionally) a Platform. Device names are not required, however if one is set it must be unique.
  1241. Each Device must be assigned to a site, and optionally to a rack within that site. Associating a device with a
  1242. particular rack face or unit is optional (for example, vertically mounted PDUs do not consume rack units).
  1243. When a new Device is created, console/power/interface/device bay components are created along with it as dictated
  1244. by the component templates assigned to its DeviceType. Components can also be added, modified, or deleted after the
  1245. creation of a Device.
  1246. """
  1247. device_type = models.ForeignKey(
  1248. to='dcim.DeviceType',
  1249. on_delete=models.PROTECT,
  1250. related_name='instances'
  1251. )
  1252. device_role = models.ForeignKey(
  1253. to='dcim.DeviceRole',
  1254. on_delete=models.PROTECT,
  1255. related_name='devices'
  1256. )
  1257. tenant = models.ForeignKey(
  1258. to='tenancy.Tenant',
  1259. on_delete=models.PROTECT,
  1260. related_name='devices',
  1261. blank=True,
  1262. null=True
  1263. )
  1264. platform = models.ForeignKey(
  1265. to='dcim.Platform',
  1266. on_delete=models.SET_NULL,
  1267. related_name='devices',
  1268. blank=True,
  1269. null=True
  1270. )
  1271. name = models.CharField(
  1272. max_length=64,
  1273. blank=True,
  1274. null=True,
  1275. unique=True
  1276. )
  1277. serial = models.CharField(
  1278. max_length=50,
  1279. blank=True,
  1280. verbose_name='Serial number'
  1281. )
  1282. asset_tag = models.CharField(
  1283. max_length=50,
  1284. blank=True,
  1285. null=True,
  1286. unique=True,
  1287. verbose_name='Asset tag',
  1288. help_text='A unique tag used to identify this device'
  1289. )
  1290. site = models.ForeignKey(
  1291. to='dcim.Site',
  1292. on_delete=models.PROTECT,
  1293. related_name='devices'
  1294. )
  1295. rack = models.ForeignKey(
  1296. to='dcim.Rack',
  1297. on_delete=models.PROTECT,
  1298. related_name='devices',
  1299. blank=True,
  1300. null=True
  1301. )
  1302. position = models.PositiveSmallIntegerField(
  1303. blank=True,
  1304. null=True,
  1305. validators=[MinValueValidator(1)],
  1306. verbose_name='Position (U)',
  1307. help_text='The lowest-numbered unit occupied by the device'
  1308. )
  1309. face = models.PositiveSmallIntegerField(
  1310. blank=True,
  1311. null=True,
  1312. choices=RACK_FACE_CHOICES,
  1313. verbose_name='Rack face'
  1314. )
  1315. status = models.PositiveSmallIntegerField(
  1316. choices=DEVICE_STATUS_CHOICES,
  1317. default=DEVICE_STATUS_ACTIVE,
  1318. verbose_name='Status'
  1319. )
  1320. primary_ip4 = models.OneToOneField(
  1321. to='ipam.IPAddress',
  1322. on_delete=models.SET_NULL,
  1323. related_name='primary_ip4_for',
  1324. blank=True,
  1325. null=True,
  1326. verbose_name='Primary IPv4'
  1327. )
  1328. primary_ip6 = models.OneToOneField(
  1329. to='ipam.IPAddress',
  1330. on_delete=models.SET_NULL,
  1331. related_name='primary_ip6_for',
  1332. blank=True,
  1333. null=True,
  1334. verbose_name='Primary IPv6'
  1335. )
  1336. cluster = models.ForeignKey(
  1337. to='virtualization.Cluster',
  1338. on_delete=models.SET_NULL,
  1339. related_name='devices',
  1340. blank=True,
  1341. null=True
  1342. )
  1343. virtual_chassis = models.ForeignKey(
  1344. to='VirtualChassis',
  1345. on_delete=models.SET_NULL,
  1346. related_name='members',
  1347. blank=True,
  1348. null=True
  1349. )
  1350. vc_position = models.PositiveSmallIntegerField(
  1351. blank=True,
  1352. null=True,
  1353. validators=[MaxValueValidator(255)]
  1354. )
  1355. vc_priority = models.PositiveSmallIntegerField(
  1356. blank=True,
  1357. null=True,
  1358. validators=[MaxValueValidator(255)]
  1359. )
  1360. comments = models.TextField(
  1361. blank=True
  1362. )
  1363. custom_field_values = GenericRelation(
  1364. to='extras.CustomFieldValue',
  1365. content_type_field='obj_type',
  1366. object_id_field='obj_id'
  1367. )
  1368. images = GenericRelation(
  1369. to='extras.ImageAttachment'
  1370. )
  1371. objects = NaturalOrderingManager()
  1372. tags = TaggableManager(through=TaggedItem)
  1373. csv_headers = [
  1374. 'name', 'device_role', 'tenant', 'manufacturer', 'model_name', 'platform', 'serial', 'asset_tag', 'status',
  1375. 'site', 'rack_group', 'rack_name', 'position', 'face', 'comments',
  1376. ]
  1377. class Meta:
  1378. ordering = ['name']
  1379. unique_together = [
  1380. ['rack', 'position', 'face'],
  1381. ['virtual_chassis', 'vc_position'],
  1382. ]
  1383. permissions = (
  1384. ('napalm_read', 'Read-only access to devices via NAPALM'),
  1385. ('napalm_write', 'Read/write access to devices via NAPALM'),
  1386. )
  1387. def __str__(self):
  1388. return self.display_name or super().__str__()
  1389. def get_absolute_url(self):
  1390. return reverse('dcim:device', args=[self.pk])
  1391. def clean(self):
  1392. super().clean()
  1393. # Validate site/rack combination
  1394. if self.rack and self.site != self.rack.site:
  1395. raise ValidationError({
  1396. 'rack': "Rack {} does not belong to site {}.".format(self.rack, self.site),
  1397. })
  1398. if self.rack is None:
  1399. if self.face is not None:
  1400. raise ValidationError({
  1401. 'face': "Cannot select a rack face without assigning a rack.",
  1402. })
  1403. if self.position:
  1404. raise ValidationError({
  1405. 'face': "Cannot select a rack position without assigning a rack.",
  1406. })
  1407. # Validate position/face combination
  1408. if self.position and self.face is None:
  1409. raise ValidationError({
  1410. 'face': "Must specify rack face when defining rack position.",
  1411. })
  1412. # Prevent 0U devices from being assigned to a specific position
  1413. if self.position and self.device_type.u_height == 0:
  1414. raise ValidationError({
  1415. 'position': "A U0 device type ({}) cannot be assigned to a rack position.".format(self.device_type)
  1416. })
  1417. if self.rack:
  1418. try:
  1419. # Child devices cannot be assigned to a rack face/unit
  1420. if self.device_type.is_child_device and self.face is not None:
  1421. raise ValidationError({
  1422. 'face': "Child device types cannot be assigned to a rack face. This is an attribute of the "
  1423. "parent device."
  1424. })
  1425. if self.device_type.is_child_device and self.position:
  1426. raise ValidationError({
  1427. 'position': "Child device types cannot be assigned to a rack position. This is an attribute of "
  1428. "the parent device."
  1429. })
  1430. # Validate rack space
  1431. rack_face = self.face if not self.device_type.is_full_depth else None
  1432. exclude_list = [self.pk] if self.pk else []
  1433. try:
  1434. available_units = self.rack.get_available_units(
  1435. u_height=self.device_type.u_height, rack_face=rack_face, exclude=exclude_list
  1436. )
  1437. if self.position and self.position not in available_units:
  1438. raise ValidationError({
  1439. 'position': "U{} is already occupied or does not have sufficient space to accommodate a(n) "
  1440. "{} ({}U).".format(self.position, self.device_type, self.device_type.u_height)
  1441. })
  1442. except Rack.DoesNotExist:
  1443. pass
  1444. except DeviceType.DoesNotExist:
  1445. pass
  1446. # Validate primary IP addresses
  1447. vc_interfaces = self.vc_interfaces.all()
  1448. if self.primary_ip4:
  1449. if self.primary_ip4.interface in vc_interfaces:
  1450. pass
  1451. elif self.primary_ip4.nat_inside is not None and self.primary_ip4.nat_inside.interface in vc_interfaces:
  1452. pass
  1453. else:
  1454. raise ValidationError({
  1455. 'primary_ip4': "The specified IP address ({}) is not assigned to this device.".format(
  1456. self.primary_ip4),
  1457. })
  1458. if self.primary_ip6:
  1459. if self.primary_ip6.interface in vc_interfaces:
  1460. pass
  1461. elif self.primary_ip6.nat_inside is not None and self.primary_ip6.nat_inside.interface in vc_interfaces:
  1462. pass
  1463. else:
  1464. raise ValidationError({
  1465. 'primary_ip6': "The specified IP address ({}) is not assigned to this device.".format(
  1466. self.primary_ip6),
  1467. })
  1468. # Validate manufacturer/platform
  1469. if hasattr(self, 'device_type') and self.platform:
  1470. if self.platform.manufacturer and self.platform.manufacturer != self.device_type.manufacturer:
  1471. raise ValidationError({
  1472. 'platform': "The assigned platform is limited to {} device types, but this device's type belongs "
  1473. "to {}.".format(self.platform.manufacturer, self.device_type.manufacturer)
  1474. })
  1475. # A Device can only be assigned to a Cluster in the same Site (or no Site)
  1476. if self.cluster and self.cluster.site is not None and self.cluster.site != self.site:
  1477. raise ValidationError({
  1478. 'cluster': "The assigned cluster belongs to a different site ({})".format(self.cluster.site)
  1479. })
  1480. # Validate virtual chassis assignment
  1481. if self.virtual_chassis and self.vc_position is None:
  1482. raise ValidationError({
  1483. 'vc_position': "A device assigned to a virtual chassis must have its position defined."
  1484. })
  1485. def save(self, *args, **kwargs):
  1486. is_new = not bool(self.pk)
  1487. super().save(*args, **kwargs)
  1488. # If this is a new Device, instantiate all of the related components per the DeviceType definition
  1489. if is_new:
  1490. ConsolePort.objects.bulk_create(
  1491. [x.instantiate(self) for x in self.device_type.consoleport_templates.all()]
  1492. )
  1493. ConsoleServerPort.objects.bulk_create(
  1494. [x.instantiate(self) for x in self.device_type.consoleserverport_templates.all()]
  1495. )
  1496. PowerPort.objects.bulk_create(
  1497. [x.instantiate(self) for x in self.device_type.powerport_templates.all()]
  1498. )
  1499. PowerOutlet.objects.bulk_create(
  1500. [x.instantiate(self) for x in self.device_type.poweroutlet_templates.all()]
  1501. )
  1502. Interface.objects.bulk_create(
  1503. [x.instantiate(self) for x in self.device_type.interface_templates.all()]
  1504. )
  1505. RearPort.objects.bulk_create(
  1506. [x.instantiate(self) for x in self.device_type.rearport_templates.all()]
  1507. )
  1508. FrontPort.objects.bulk_create(
  1509. [x.instantiate(self) for x in self.device_type.frontport_templates.all()]
  1510. )
  1511. DeviceBay.objects.bulk_create(
  1512. [x.instantiate(self) for x in self.device_type.device_bay_templates.all()]
  1513. )
  1514. # Update Site and Rack assignment for any child Devices
  1515. devices = Device.objects.filter(parent_bay__device=self)
  1516. for device in devices:
  1517. device.site = self.site
  1518. device.rack = self.rack
  1519. device.save()
  1520. def to_csv(self):
  1521. return (
  1522. self.name or '',
  1523. self.device_role.name,
  1524. self.tenant.name if self.tenant else None,
  1525. self.device_type.manufacturer.name,
  1526. self.device_type.model,
  1527. self.platform.name if self.platform else None,
  1528. self.serial,
  1529. self.asset_tag,
  1530. self.get_status_display(),
  1531. self.site.name,
  1532. self.rack.group.name if self.rack and self.rack.group else None,
  1533. self.rack.name if self.rack else None,
  1534. self.position,
  1535. self.get_face_display(),
  1536. self.comments,
  1537. )
  1538. @property
  1539. def display_name(self):
  1540. if self.name:
  1541. return self.name
  1542. elif self.virtual_chassis and self.virtual_chassis.master.name:
  1543. return "{}:{}".format(self.virtual_chassis.master, self.vc_position)
  1544. elif hasattr(self, 'device_type'):
  1545. return "{}".format(self.device_type)
  1546. return ""
  1547. @property
  1548. def identifier(self):
  1549. """
  1550. Return the device name if set; otherwise return the Device's primary key as {pk}
  1551. """
  1552. if self.name is not None:
  1553. return self.name
  1554. return '{{{}}}'.format(self.pk)
  1555. @property
  1556. def primary_ip(self):
  1557. if settings.PREFER_IPV4 and self.primary_ip4:
  1558. return self.primary_ip4
  1559. elif self.primary_ip6:
  1560. return self.primary_ip6
  1561. elif self.primary_ip4:
  1562. return self.primary_ip4
  1563. else:
  1564. return None
  1565. def get_vc_master(self):
  1566. """
  1567. If this Device is a VirtualChassis member, return the VC master. Otherwise, return None.
  1568. """
  1569. return self.virtual_chassis.master if self.virtual_chassis else None
  1570. @property
  1571. def vc_interfaces(self):
  1572. """
  1573. Return a QuerySet matching all Interfaces assigned to this Device or, if this Device is a VC master, to another
  1574. Device belonging to the same VirtualChassis.
  1575. """
  1576. filter = Q(device=self)
  1577. if self.virtual_chassis and self.virtual_chassis.master == self:
  1578. filter |= Q(device__virtual_chassis=self.virtual_chassis, mgmt_only=False)
  1579. return Interface.objects.filter(filter)
  1580. def get_cables(self, pk_list=False):
  1581. """
  1582. Return a QuerySet or PK list matching all Cables connected to a component of this Device.
  1583. """
  1584. cable_pks = []
  1585. for component_model in [
  1586. ConsolePort, ConsoleServerPort, PowerPort, PowerOutlet, Interface, FrontPort, RearPort
  1587. ]:
  1588. cable_pks += component_model.objects.filter(
  1589. device=self, cable__isnull=False
  1590. ).values_list('cable', flat=True)
  1591. if pk_list:
  1592. return cable_pks
  1593. return Cable.objects.filter(pk__in=cable_pks)
  1594. def get_children(self):
  1595. """
  1596. Return the set of child Devices installed in DeviceBays within this Device.
  1597. """
  1598. return Device.objects.filter(parent_bay__device=self.pk)
  1599. def get_status_class(self):
  1600. return STATUS_CLASSES[self.status]
  1601. #
  1602. # Console ports
  1603. #
  1604. class ConsolePort(CableTermination, ComponentModel):
  1605. """
  1606. A physical console port within a Device. ConsolePorts connect to ConsoleServerPorts.
  1607. """
  1608. device = models.ForeignKey(
  1609. to='dcim.Device',
  1610. on_delete=models.CASCADE,
  1611. related_name='consoleports'
  1612. )
  1613. name = models.CharField(
  1614. max_length=50
  1615. )
  1616. connected_endpoint = models.OneToOneField(
  1617. to='dcim.ConsoleServerPort',
  1618. on_delete=models.SET_NULL,
  1619. related_name='connected_endpoint',
  1620. blank=True,
  1621. null=True
  1622. )
  1623. connection_status = models.NullBooleanField(
  1624. choices=CONNECTION_STATUS_CHOICES,
  1625. blank=True
  1626. )
  1627. objects = NaturalOrderingManager()
  1628. tags = TaggableManager(through=TaggedItem)
  1629. csv_headers = ['device', 'name', 'description']
  1630. class Meta:
  1631. ordering = ['device', 'name']
  1632. unique_together = ['device', 'name']
  1633. def __str__(self):
  1634. return self.name
  1635. def get_absolute_url(self):
  1636. return self.device.get_absolute_url()
  1637. def to_csv(self):
  1638. return (
  1639. self.device.identifier,
  1640. self.name,
  1641. self.description,
  1642. )
  1643. #
  1644. # Console server ports
  1645. #
  1646. class ConsoleServerPort(CableTermination, ComponentModel):
  1647. """
  1648. A physical port within a Device (typically a designated console server) which provides access to ConsolePorts.
  1649. """
  1650. device = models.ForeignKey(
  1651. to='dcim.Device',
  1652. on_delete=models.CASCADE,
  1653. related_name='consoleserverports'
  1654. )
  1655. name = models.CharField(
  1656. max_length=50
  1657. )
  1658. connection_status = models.NullBooleanField(
  1659. choices=CONNECTION_STATUS_CHOICES,
  1660. blank=True
  1661. )
  1662. objects = NaturalOrderingManager()
  1663. tags = TaggableManager(through=TaggedItem)
  1664. csv_headers = ['device', 'name', 'description']
  1665. class Meta:
  1666. unique_together = ['device', 'name']
  1667. def __str__(self):
  1668. return self.name
  1669. def get_absolute_url(self):
  1670. return self.device.get_absolute_url()
  1671. def to_csv(self):
  1672. return (
  1673. self.device.identifier,
  1674. self.name,
  1675. self.description,
  1676. )
  1677. #
  1678. # Power ports
  1679. #
  1680. class PowerPort(CableTermination, ComponentModel):
  1681. """
  1682. A physical power supply (intake) port within a Device. PowerPorts connect to PowerOutlets.
  1683. """
  1684. device = models.ForeignKey(
  1685. to='dcim.Device',
  1686. on_delete=models.CASCADE,
  1687. related_name='powerports'
  1688. )
  1689. name = models.CharField(
  1690. max_length=50
  1691. )
  1692. maximum_draw = models.PositiveSmallIntegerField(
  1693. blank=True,
  1694. null=True,
  1695. validators=[MinValueValidator(1)],
  1696. help_text="Maximum current draw (watts)"
  1697. )
  1698. allocated_draw = models.PositiveSmallIntegerField(
  1699. blank=True,
  1700. null=True,
  1701. validators=[MinValueValidator(1)],
  1702. help_text="Allocated current draw (watts)"
  1703. )
  1704. _connected_poweroutlet = models.OneToOneField(
  1705. to='dcim.PowerOutlet',
  1706. on_delete=models.SET_NULL,
  1707. related_name='connected_endpoint',
  1708. blank=True,
  1709. null=True
  1710. )
  1711. _connected_powerfeed = models.OneToOneField(
  1712. to='dcim.PowerFeed',
  1713. on_delete=models.SET_NULL,
  1714. related_name='+',
  1715. blank=True,
  1716. null=True
  1717. )
  1718. connection_status = models.NullBooleanField(
  1719. choices=CONNECTION_STATUS_CHOICES,
  1720. blank=True
  1721. )
  1722. objects = NaturalOrderingManager()
  1723. tags = TaggableManager(through=TaggedItem)
  1724. csv_headers = ['device', 'name', 'maximum_draw', 'allocated_draw', 'description']
  1725. class Meta:
  1726. ordering = ['device', 'name']
  1727. unique_together = ['device', 'name']
  1728. def __str__(self):
  1729. return self.name
  1730. def get_absolute_url(self):
  1731. return self.device.get_absolute_url()
  1732. def to_csv(self):
  1733. return (
  1734. self.device.identifier,
  1735. self.name,
  1736. self.maximum_draw,
  1737. self.allocated_draw,
  1738. self.description,
  1739. )
  1740. @property
  1741. def connected_endpoint(self):
  1742. if self._connected_poweroutlet:
  1743. return self._connected_poweroutlet
  1744. return self._connected_powerfeed
  1745. @connected_endpoint.setter
  1746. def connected_endpoint(self, value):
  1747. if value is None:
  1748. self._connected_poweroutlet = None
  1749. self._connected_powerfeed = None
  1750. elif isinstance(value, PowerOutlet):
  1751. self._connected_poweroutlet = value
  1752. self._connected_powerfeed = None
  1753. elif isinstance(value, PowerFeed):
  1754. self._connected_poweroutlet = None
  1755. self._connected_powerfeed = value
  1756. else:
  1757. raise ValueError(
  1758. "Connected endpoint must be a PowerOutlet or PowerFeed, not {}.".format(type(value))
  1759. )
  1760. def get_power_draw(self):
  1761. """
  1762. Return the allocated and maximum power draw (in VA) and child PowerOutlet count for this PowerPort.
  1763. """
  1764. # Calculate aggregate draw of all child power outlets if no numbers have been defined manually
  1765. if self.allocated_draw is None and self.maximum_draw is None:
  1766. outlet_ids = PowerOutlet.objects.filter(power_port=self).values_list('pk', flat=True)
  1767. utilization = PowerPort.objects.filter(_connected_poweroutlet_id__in=outlet_ids).aggregate(
  1768. maximum_draw_total=Sum('maximum_draw'),
  1769. allocated_draw_total=Sum('allocated_draw'),
  1770. )
  1771. ret = {
  1772. 'allocated': utilization['allocated_draw_total'] or 0,
  1773. 'maximum': utilization['maximum_draw_total'] or 0,
  1774. 'outlet_count': len(outlet_ids),
  1775. 'legs': [],
  1776. }
  1777. # Calculate per-leg aggregates for three-phase feeds
  1778. if self._connected_powerfeed and self._connected_powerfeed.phase == POWERFEED_PHASE_3PHASE:
  1779. for leg, leg_name in POWERFEED_LEG_CHOICES:
  1780. outlet_ids = PowerOutlet.objects.filter(power_port=self, feed_leg=leg).values_list('pk', flat=True)
  1781. utilization = PowerPort.objects.filter(_connected_poweroutlet_id__in=outlet_ids).aggregate(
  1782. maximum_draw_total=Sum('maximum_draw'),
  1783. allocated_draw_total=Sum('allocated_draw'),
  1784. )
  1785. ret['legs'].append({
  1786. 'name': leg_name,
  1787. 'allocated': utilization['allocated_draw_total'] or 0,
  1788. 'maximum': utilization['maximum_draw_total'] or 0,
  1789. 'outlet_count': len(outlet_ids),
  1790. })
  1791. return ret
  1792. # Default to administratively defined values
  1793. return {
  1794. 'allocated': self.allocated_draw or 0,
  1795. 'maximum': self.maximum_draw or 0,
  1796. 'outlet_count': PowerOutlet.objects.filter(power_port=self).count(),
  1797. 'legs': [],
  1798. }
  1799. #
  1800. # Power outlets
  1801. #
  1802. class PowerOutlet(CableTermination, ComponentModel):
  1803. """
  1804. A physical power outlet (output) within a Device which provides power to a PowerPort.
  1805. """
  1806. device = models.ForeignKey(
  1807. to='dcim.Device',
  1808. on_delete=models.CASCADE,
  1809. related_name='poweroutlets'
  1810. )
  1811. name = models.CharField(
  1812. max_length=50
  1813. )
  1814. power_port = models.ForeignKey(
  1815. to='dcim.PowerPort',
  1816. on_delete=models.SET_NULL,
  1817. blank=True,
  1818. null=True,
  1819. related_name='poweroutlets'
  1820. )
  1821. feed_leg = models.PositiveSmallIntegerField(
  1822. choices=POWERFEED_LEG_CHOICES,
  1823. blank=True,
  1824. null=True,
  1825. help_text="Phase (for three-phase feeds)"
  1826. )
  1827. connection_status = models.NullBooleanField(
  1828. choices=CONNECTION_STATUS_CHOICES,
  1829. blank=True
  1830. )
  1831. objects = NaturalOrderingManager()
  1832. tags = TaggableManager(through=TaggedItem)
  1833. csv_headers = ['device', 'name', 'power_port', 'feed_leg', 'description']
  1834. class Meta:
  1835. unique_together = ['device', 'name']
  1836. def __str__(self):
  1837. return self.name
  1838. def get_absolute_url(self):
  1839. return self.device.get_absolute_url()
  1840. def to_csv(self):
  1841. return (
  1842. self.device.identifier,
  1843. self.name,
  1844. self.power_port.name if self.power_port else None,
  1845. self.get_feed_leg_display(),
  1846. self.description,
  1847. )
  1848. def clean(self):
  1849. # Validate power port assignment
  1850. if self.power_port and self.power_port.device != self.device:
  1851. raise ValidationError(
  1852. "Parent power port ({}) must belong to the same device".format(self.power_port)
  1853. )
  1854. #
  1855. # Interfaces
  1856. #
  1857. class Interface(CableTermination, ComponentModel):
  1858. """
  1859. A network interface within a Device or VirtualMachine. A physical Interface can connect to exactly one other
  1860. Interface.
  1861. """
  1862. device = models.ForeignKey(
  1863. to='Device',
  1864. on_delete=models.CASCADE,
  1865. related_name='interfaces',
  1866. null=True,
  1867. blank=True
  1868. )
  1869. virtual_machine = models.ForeignKey(
  1870. to='virtualization.VirtualMachine',
  1871. on_delete=models.CASCADE,
  1872. related_name='interfaces',
  1873. null=True,
  1874. blank=True
  1875. )
  1876. name = models.CharField(
  1877. max_length=64
  1878. )
  1879. _connected_interface = models.OneToOneField(
  1880. to='self',
  1881. on_delete=models.SET_NULL,
  1882. related_name='+',
  1883. blank=True,
  1884. null=True
  1885. )
  1886. _connected_circuittermination = models.OneToOneField(
  1887. to='circuits.CircuitTermination',
  1888. on_delete=models.SET_NULL,
  1889. related_name='+',
  1890. blank=True,
  1891. null=True
  1892. )
  1893. connection_status = models.NullBooleanField(
  1894. choices=CONNECTION_STATUS_CHOICES,
  1895. blank=True
  1896. )
  1897. lag = models.ForeignKey(
  1898. to='self',
  1899. on_delete=models.SET_NULL,
  1900. related_name='member_interfaces',
  1901. null=True,
  1902. blank=True,
  1903. verbose_name='Parent LAG'
  1904. )
  1905. type = models.PositiveSmallIntegerField(
  1906. choices=IFACE_TYPE_CHOICES,
  1907. default=IFACE_TYPE_10GE_SFP_PLUS
  1908. )
  1909. enabled = models.BooleanField(
  1910. default=True
  1911. )
  1912. mac_address = MACAddressField(
  1913. null=True,
  1914. blank=True,
  1915. verbose_name='MAC Address'
  1916. )
  1917. mtu = models.PositiveIntegerField(
  1918. blank=True,
  1919. null=True,
  1920. validators=[MinValueValidator(1), MaxValueValidator(65536)],
  1921. verbose_name='MTU'
  1922. )
  1923. mgmt_only = models.BooleanField(
  1924. default=False,
  1925. verbose_name='OOB Management',
  1926. help_text='This interface is used only for out-of-band management'
  1927. )
  1928. mode = models.PositiveSmallIntegerField(
  1929. choices=IFACE_MODE_CHOICES,
  1930. blank=True,
  1931. null=True
  1932. )
  1933. untagged_vlan = models.ForeignKey(
  1934. to='ipam.VLAN',
  1935. on_delete=models.SET_NULL,
  1936. related_name='interfaces_as_untagged',
  1937. null=True,
  1938. blank=True,
  1939. verbose_name='Untagged VLAN'
  1940. )
  1941. tagged_vlans = models.ManyToManyField(
  1942. to='ipam.VLAN',
  1943. related_name='interfaces_as_tagged',
  1944. blank=True,
  1945. verbose_name='Tagged VLANs'
  1946. )
  1947. objects = InterfaceManager()
  1948. tags = TaggableManager(through=TaggedItem)
  1949. csv_headers = [
  1950. 'device', 'virtual_machine', 'name', 'lag', 'type', 'enabled', 'mac_address', 'mtu', 'mgmt_only',
  1951. 'description', 'mode',
  1952. ]
  1953. class Meta:
  1954. ordering = ['device', 'name']
  1955. unique_together = ['device', 'name']
  1956. def __str__(self):
  1957. return self.name
  1958. def get_absolute_url(self):
  1959. return reverse('dcim:interface', kwargs={'pk': self.pk})
  1960. def to_csv(self):
  1961. return (
  1962. self.device.identifier if self.device else None,
  1963. self.virtual_machine.name if self.virtual_machine else None,
  1964. self.name,
  1965. self.lag.name if self.lag else None,
  1966. self.get_type_display(),
  1967. self.enabled,
  1968. self.mac_address,
  1969. self.mtu,
  1970. self.mgmt_only,
  1971. self.description,
  1972. self.get_mode_display(),
  1973. )
  1974. def clean(self):
  1975. # An Interface must belong to a Device *or* to a VirtualMachine
  1976. if self.device and self.virtual_machine:
  1977. raise ValidationError("An interface cannot belong to both a device and a virtual machine.")
  1978. if not self.device and not self.virtual_machine:
  1979. raise ValidationError("An interface must belong to either a device or a virtual machine.")
  1980. # VM interfaces must be virtual
  1981. if self.virtual_machine and self.type is not IFACE_TYPE_VIRTUAL:
  1982. raise ValidationError({
  1983. 'type': "Virtual machines can only have virtual interfaces."
  1984. })
  1985. # Virtual interfaces cannot be connected
  1986. if self.type in NONCONNECTABLE_IFACE_TYPES and (
  1987. self.cable or getattr(self, 'circuit_termination', False)
  1988. ):
  1989. raise ValidationError({
  1990. 'type': "Virtual and wireless interfaces cannot be connected to another interface or circuit. "
  1991. "Disconnect the interface or choose a suitable type."
  1992. })
  1993. # An interface's LAG must belong to the same device (or VC master)
  1994. if self.lag and self.lag.device not in [self.device, self.device.get_vc_master()]:
  1995. raise ValidationError({
  1996. 'lag': "The selected LAG interface ({}) belongs to a different device ({}).".format(
  1997. self.lag.name, self.lag.device.name
  1998. )
  1999. })
  2000. # A virtual interface cannot have a parent LAG
  2001. if self.type in NONCONNECTABLE_IFACE_TYPES and self.lag is not None:
  2002. raise ValidationError({
  2003. 'lag': "{} interfaces cannot have a parent LAG interface.".format(self.get_type_display())
  2004. })
  2005. # Only a LAG can have LAG members
  2006. if self.type != IFACE_TYPE_LAG and self.member_interfaces.exists():
  2007. raise ValidationError({
  2008. 'type': "Cannot change interface type; it has LAG members ({}).".format(
  2009. ", ".join([iface.name for iface in self.member_interfaces.all()])
  2010. )
  2011. })
  2012. # Validate untagged VLAN
  2013. if self.untagged_vlan and self.untagged_vlan.site not in [self.parent.site, None]:
  2014. raise ValidationError({
  2015. 'untagged_vlan': "The untagged VLAN ({}) must belong to the same site as the interface's parent "
  2016. "device/VM, or it must be global".format(self.untagged_vlan)
  2017. })
  2018. def save(self, *args, **kwargs):
  2019. # Remove untagged VLAN assignment for non-802.1Q interfaces
  2020. if self.mode is None:
  2021. self.untagged_vlan = None
  2022. # Only "tagged" interfaces may have tagged VLANs assigned. ("tagged all" implies all VLANs are assigned.)
  2023. if self.pk and self.mode is not IFACE_MODE_TAGGED:
  2024. self.tagged_vlans.clear()
  2025. return super().save(*args, **kwargs)
  2026. def to_objectchange(self, action):
  2027. # Annotate the parent Device/VM
  2028. try:
  2029. parent_obj = self.device or self.virtual_machine
  2030. except ObjectDoesNotExist:
  2031. parent_obj = None
  2032. return ObjectChange(
  2033. changed_object=self,
  2034. object_repr=str(self),
  2035. action=action,
  2036. related_object=parent_obj,
  2037. object_data=serialize_object(self)
  2038. )
  2039. # TODO: Remove in v2.7
  2040. @property
  2041. def form_factor(self):
  2042. """
  2043. Backward-compatibility for form_factor
  2044. """
  2045. return self.type
  2046. # TODO: Remove in v2.7
  2047. @form_factor.setter
  2048. def form_factor(self, value):
  2049. """
  2050. Backward-compatibility for form_factor
  2051. """
  2052. self.type = value
  2053. @property
  2054. def connected_endpoint(self):
  2055. if self._connected_interface:
  2056. return self._connected_interface
  2057. return self._connected_circuittermination
  2058. @connected_endpoint.setter
  2059. def connected_endpoint(self, value):
  2060. from circuits.models import CircuitTermination
  2061. if value is None:
  2062. self._connected_interface = None
  2063. self._connected_circuittermination = None
  2064. elif isinstance(value, Interface):
  2065. self._connected_interface = value
  2066. self._connected_circuittermination = None
  2067. elif isinstance(value, CircuitTermination):
  2068. self._connected_interface = None
  2069. self._connected_circuittermination = value
  2070. else:
  2071. raise ValueError(
  2072. "Connected endpoint must be an Interface or CircuitTermination, not {}.".format(type(value))
  2073. )
  2074. @property
  2075. def parent(self):
  2076. return self.device or self.virtual_machine
  2077. @property
  2078. def is_connectable(self):
  2079. return self.type not in NONCONNECTABLE_IFACE_TYPES
  2080. @property
  2081. def is_virtual(self):
  2082. return self.type in VIRTUAL_IFACE_TYPES
  2083. @property
  2084. def is_wireless(self):
  2085. return self.type in WIRELESS_IFACE_TYPES
  2086. @property
  2087. def is_lag(self):
  2088. return self.type == IFACE_TYPE_LAG
  2089. @property
  2090. def count_ipaddresses(self):
  2091. return self.ip_addresses.count()
  2092. #
  2093. # Pass-through ports
  2094. #
  2095. class FrontPort(CableTermination, ComponentModel):
  2096. """
  2097. A pass-through port on the front of a Device.
  2098. """
  2099. device = models.ForeignKey(
  2100. to='dcim.Device',
  2101. on_delete=models.CASCADE,
  2102. related_name='frontports'
  2103. )
  2104. name = models.CharField(
  2105. max_length=64
  2106. )
  2107. type = models.PositiveSmallIntegerField(
  2108. choices=PORT_TYPE_CHOICES
  2109. )
  2110. rear_port = models.ForeignKey(
  2111. to='dcim.RearPort',
  2112. on_delete=models.CASCADE,
  2113. related_name='frontports'
  2114. )
  2115. rear_port_position = models.PositiveSmallIntegerField(
  2116. default=1,
  2117. validators=[MinValueValidator(1), MaxValueValidator(64)]
  2118. )
  2119. is_path_endpoint = False
  2120. objects = NaturalOrderingManager()
  2121. tags = TaggableManager(through=TaggedItem)
  2122. csv_headers = ['device', 'name', 'type', 'rear_port', 'rear_port_position', 'description']
  2123. class Meta:
  2124. ordering = ['device', 'name']
  2125. unique_together = [
  2126. ['device', 'name'],
  2127. ['rear_port', 'rear_port_position'],
  2128. ]
  2129. def __str__(self):
  2130. return self.name
  2131. def to_csv(self):
  2132. return (
  2133. self.device.identifier,
  2134. self.name,
  2135. self.get_type_display(),
  2136. self.rear_port.name,
  2137. self.rear_port_position,
  2138. self.description,
  2139. )
  2140. def clean(self):
  2141. # Validate rear port assignment
  2142. if self.rear_port.device != self.device:
  2143. raise ValidationError(
  2144. "Rear port ({}) must belong to the same device".format(self.rear_port)
  2145. )
  2146. # Validate rear port position assignment
  2147. if self.rear_port_position > self.rear_port.positions:
  2148. raise ValidationError(
  2149. "Invalid rear port position ({}); rear port {} has only {} positions".format(
  2150. self.rear_port_position, self.rear_port.name, self.rear_port.positions
  2151. )
  2152. )
  2153. class RearPort(CableTermination, ComponentModel):
  2154. """
  2155. A pass-through port on the rear of a Device.
  2156. """
  2157. device = models.ForeignKey(
  2158. to='dcim.Device',
  2159. on_delete=models.CASCADE,
  2160. related_name='rearports'
  2161. )
  2162. name = models.CharField(
  2163. max_length=64
  2164. )
  2165. type = models.PositiveSmallIntegerField(
  2166. choices=PORT_TYPE_CHOICES
  2167. )
  2168. positions = models.PositiveSmallIntegerField(
  2169. default=1,
  2170. validators=[MinValueValidator(1), MaxValueValidator(64)]
  2171. )
  2172. is_path_endpoint = False
  2173. objects = NaturalOrderingManager()
  2174. tags = TaggableManager(through=TaggedItem)
  2175. csv_headers = ['device', 'name', 'type', 'positions', 'description']
  2176. class Meta:
  2177. ordering = ['device', 'name']
  2178. unique_together = ['device', 'name']
  2179. def __str__(self):
  2180. return self.name
  2181. def to_csv(self):
  2182. return (
  2183. self.device.identifier,
  2184. self.name,
  2185. self.get_type_display(),
  2186. self.positions,
  2187. self.description,
  2188. )
  2189. #
  2190. # Device bays
  2191. #
  2192. class DeviceBay(ComponentModel):
  2193. """
  2194. An empty space within a Device which can house a child device
  2195. """
  2196. device = models.ForeignKey(
  2197. to='dcim.Device',
  2198. on_delete=models.CASCADE,
  2199. related_name='device_bays'
  2200. )
  2201. name = models.CharField(
  2202. max_length=50,
  2203. verbose_name='Name'
  2204. )
  2205. installed_device = models.OneToOneField(
  2206. to='dcim.Device',
  2207. on_delete=models.SET_NULL,
  2208. related_name='parent_bay',
  2209. blank=True,
  2210. null=True
  2211. )
  2212. objects = NaturalOrderingManager()
  2213. tags = TaggableManager(through=TaggedItem)
  2214. csv_headers = ['device', 'name', 'installed_device', 'description']
  2215. class Meta:
  2216. ordering = ['device', 'name']
  2217. unique_together = ['device', 'name']
  2218. def __str__(self):
  2219. return '{} - {}'.format(self.device.name, self.name)
  2220. def get_absolute_url(self):
  2221. return self.device.get_absolute_url()
  2222. def to_csv(self):
  2223. return (
  2224. self.device.identifier,
  2225. self.name,
  2226. self.installed_device.identifier if self.installed_device else None,
  2227. self.description,
  2228. )
  2229. def clean(self):
  2230. # Validate that the parent Device can have DeviceBays
  2231. if not self.device.device_type.is_parent_device:
  2232. raise ValidationError("This type of device ({}) does not support device bays.".format(
  2233. self.device.device_type
  2234. ))
  2235. # Cannot install a device into itself, obviously
  2236. if self.device == self.installed_device:
  2237. raise ValidationError("Cannot install a device into itself.")
  2238. # Check that the installed device is not already installed elsewhere
  2239. if self.installed_device:
  2240. current_bay = DeviceBay.objects.filter(installed_device=self.installed_device).first()
  2241. if current_bay and current_bay != self:
  2242. raise ValidationError({
  2243. 'installed_device': "Cannot install the specified device; device is already installed in {}".format(
  2244. current_bay
  2245. )
  2246. })
  2247. #
  2248. # Inventory items
  2249. #
  2250. class InventoryItem(ComponentModel):
  2251. """
  2252. An InventoryItem represents a serialized piece of hardware within a Device, such as a line card or power supply.
  2253. InventoryItems are used only for inventory purposes.
  2254. """
  2255. device = models.ForeignKey(
  2256. to='dcim.Device',
  2257. on_delete=models.CASCADE,
  2258. related_name='inventory_items'
  2259. )
  2260. parent = models.ForeignKey(
  2261. to='self',
  2262. on_delete=models.CASCADE,
  2263. related_name='child_items',
  2264. blank=True,
  2265. null=True
  2266. )
  2267. name = models.CharField(
  2268. max_length=50,
  2269. verbose_name='Name'
  2270. )
  2271. manufacturer = models.ForeignKey(
  2272. to='dcim.Manufacturer',
  2273. on_delete=models.PROTECT,
  2274. related_name='inventory_items',
  2275. blank=True,
  2276. null=True
  2277. )
  2278. part_id = models.CharField(
  2279. max_length=50,
  2280. verbose_name='Part ID',
  2281. blank=True
  2282. )
  2283. serial = models.CharField(
  2284. max_length=50,
  2285. verbose_name='Serial number',
  2286. blank=True
  2287. )
  2288. asset_tag = models.CharField(
  2289. max_length=50,
  2290. unique=True,
  2291. blank=True,
  2292. null=True,
  2293. verbose_name='Asset tag',
  2294. help_text='A unique tag used to identify this item'
  2295. )
  2296. discovered = models.BooleanField(
  2297. default=False,
  2298. verbose_name='Discovered'
  2299. )
  2300. tags = TaggableManager(through=TaggedItem)
  2301. csv_headers = [
  2302. 'device', 'name', 'manufacturer', 'part_id', 'serial', 'asset_tag', 'discovered', 'description',
  2303. ]
  2304. class Meta:
  2305. ordering = ['device__id', 'parent__id', 'name']
  2306. unique_together = ['device', 'parent', 'name']
  2307. def __str__(self):
  2308. return self.name
  2309. def get_absolute_url(self):
  2310. return self.device.get_absolute_url()
  2311. def to_csv(self):
  2312. return (
  2313. self.device.name or '{{{}}}'.format(self.device.pk),
  2314. self.name,
  2315. self.manufacturer.name if self.manufacturer else None,
  2316. self.part_id,
  2317. self.serial,
  2318. self.asset_tag,
  2319. self.discovered,
  2320. self.description,
  2321. )
  2322. #
  2323. # Virtual chassis
  2324. #
  2325. class VirtualChassis(ChangeLoggedModel):
  2326. """
  2327. A collection of Devices which operate with a shared control plane (e.g. a switch stack).
  2328. """
  2329. master = models.OneToOneField(
  2330. to='Device',
  2331. on_delete=models.PROTECT,
  2332. related_name='vc_master_for'
  2333. )
  2334. domain = models.CharField(
  2335. max_length=30,
  2336. blank=True
  2337. )
  2338. tags = TaggableManager(through=TaggedItem)
  2339. csv_headers = ['master', 'domain']
  2340. class Meta:
  2341. ordering = ['master']
  2342. verbose_name_plural = 'virtual chassis'
  2343. def __str__(self):
  2344. return str(self.master) if hasattr(self, 'master') else 'New Virtual Chassis'
  2345. def get_absolute_url(self):
  2346. return self.master.get_absolute_url()
  2347. def clean(self):
  2348. # Verify that the selected master device has been assigned to this VirtualChassis. (Skip when creating a new
  2349. # VirtualChassis.)
  2350. if self.pk and self.master not in self.members.all():
  2351. raise ValidationError({
  2352. 'master': "The selected master is not assigned to this virtual chassis."
  2353. })
  2354. def delete(self, *args, **kwargs):
  2355. # Check for LAG interfaces split across member chassis
  2356. interfaces = Interface.objects.filter(
  2357. device__in=self.members.all(),
  2358. lag__isnull=False
  2359. ).exclude(
  2360. lag__device=F('device')
  2361. )
  2362. if interfaces:
  2363. raise ProtectedError(
  2364. "Unable to delete virtual chassis {}. There are member interfaces which form a cross-chassis "
  2365. "LAG".format(self),
  2366. interfaces
  2367. )
  2368. return super().delete(*args, **kwargs)
  2369. def to_csv(self):
  2370. return (
  2371. self.master,
  2372. self.domain,
  2373. )
  2374. #
  2375. # Power
  2376. #
  2377. class PowerPanel(ChangeLoggedModel):
  2378. """
  2379. A distribution point for electrical power; e.g. a data center RPP.
  2380. """
  2381. site = models.ForeignKey(
  2382. to='Site',
  2383. on_delete=models.PROTECT
  2384. )
  2385. rack_group = models.ForeignKey(
  2386. to='RackGroup',
  2387. on_delete=models.PROTECT,
  2388. blank=True,
  2389. null=True
  2390. )
  2391. name = models.CharField(
  2392. max_length=50
  2393. )
  2394. csv_headers = ['site', 'rack_group_name', 'name']
  2395. class Meta:
  2396. ordering = ['site', 'name']
  2397. unique_together = ['site', 'name']
  2398. def __str__(self):
  2399. return self.name
  2400. def get_absolute_url(self):
  2401. return reverse('dcim:powerpanel', args=[self.pk])
  2402. def to_csv(self):
  2403. return (
  2404. self.site.name,
  2405. self.rack_group.name if self.rack_group else None,
  2406. self.name,
  2407. )
  2408. def clean(self):
  2409. # RackGroup must belong to assigned Site
  2410. if self.rack_group and self.rack_group.site != self.site:
  2411. raise ValidationError("Rack group {} ({}) is in a different site than {}".format(
  2412. self.rack_group, self.rack_group.site, self.site
  2413. ))
  2414. class PowerFeed(ChangeLoggedModel, CableTermination, CustomFieldModel):
  2415. """
  2416. An electrical circuit delivered from a PowerPanel.
  2417. """
  2418. power_panel = models.ForeignKey(
  2419. to='PowerPanel',
  2420. on_delete=models.PROTECT,
  2421. related_name='powerfeeds'
  2422. )
  2423. rack = models.ForeignKey(
  2424. to='Rack',
  2425. on_delete=models.PROTECT,
  2426. blank=True,
  2427. null=True
  2428. )
  2429. connected_endpoint = models.OneToOneField(
  2430. to='dcim.PowerPort',
  2431. on_delete=models.SET_NULL,
  2432. related_name='+',
  2433. blank=True,
  2434. null=True
  2435. )
  2436. connection_status = models.NullBooleanField(
  2437. choices=CONNECTION_STATUS_CHOICES,
  2438. blank=True
  2439. )
  2440. name = models.CharField(
  2441. max_length=50
  2442. )
  2443. status = models.PositiveSmallIntegerField(
  2444. choices=POWERFEED_STATUS_CHOICES,
  2445. default=POWERFEED_STATUS_ACTIVE
  2446. )
  2447. type = models.PositiveSmallIntegerField(
  2448. choices=POWERFEED_TYPE_CHOICES,
  2449. default=POWERFEED_TYPE_PRIMARY
  2450. )
  2451. supply = models.PositiveSmallIntegerField(
  2452. choices=POWERFEED_SUPPLY_CHOICES,
  2453. default=POWERFEED_SUPPLY_AC
  2454. )
  2455. phase = models.PositiveSmallIntegerField(
  2456. choices=POWERFEED_PHASE_CHOICES,
  2457. default=POWERFEED_PHASE_SINGLE
  2458. )
  2459. voltage = models.PositiveSmallIntegerField(
  2460. validators=[MinValueValidator(1)],
  2461. default=120
  2462. )
  2463. amperage = models.PositiveSmallIntegerField(
  2464. validators=[MinValueValidator(1)],
  2465. default=20
  2466. )
  2467. max_utilization = models.PositiveSmallIntegerField(
  2468. validators=[MinValueValidator(1), MaxValueValidator(100)],
  2469. default=80,
  2470. help_text="Maximum permissible draw (percentage)"
  2471. )
  2472. available_power = models.PositiveSmallIntegerField(
  2473. default=0,
  2474. editable=False
  2475. )
  2476. comments = models.TextField(
  2477. blank=True
  2478. )
  2479. custom_field_values = GenericRelation(
  2480. to='extras.CustomFieldValue',
  2481. content_type_field='obj_type',
  2482. object_id_field='obj_id'
  2483. )
  2484. tags = TaggableManager(through=TaggedItem)
  2485. csv_headers = [
  2486. 'site', 'panel_name', 'rack_group', 'rack_name', 'name', 'status', 'type', 'supply', 'phase', 'voltage',
  2487. 'amperage', 'max_utilization', 'comments',
  2488. ]
  2489. class Meta:
  2490. ordering = ['power_panel', 'name']
  2491. unique_together = ['power_panel', 'name']
  2492. def __str__(self):
  2493. return self.name
  2494. def get_absolute_url(self):
  2495. return reverse('dcim:powerfeed', args=[self.pk])
  2496. def to_csv(self):
  2497. return (
  2498. self.power_panel.site.name,
  2499. self.power_panel.name,
  2500. self.rack.group.name if self.rack and self.rack.group else None,
  2501. self.rack.name if self.rack else None,
  2502. self.name,
  2503. self.get_status_display(),
  2504. self.get_type_display(),
  2505. self.get_supply_display(),
  2506. self.get_phase_display(),
  2507. self.voltage,
  2508. self.amperage,
  2509. self.max_utilization,
  2510. self.comments,
  2511. )
  2512. def clean(self):
  2513. # Rack must belong to same Site as PowerPanel
  2514. if self.rack and self.rack.site != self.power_panel.site:
  2515. raise ValidationError("Rack {} ({}) and power panel {} ({}) are in different sites".format(
  2516. self.rack, self.rack.site, self.power_panel, self.power_panel.site
  2517. ))
  2518. def save(self, *args, **kwargs):
  2519. # Cache the available_power property on the instance
  2520. kva = self.voltage * self.amperage * (self.max_utilization / 100)
  2521. if self.phase == POWERFEED_PHASE_3PHASE:
  2522. self.available_power = round(kva * 1.732)
  2523. else:
  2524. self.available_power = round(kva)
  2525. super().save(*args, **kwargs)
  2526. def get_type_class(self):
  2527. return STATUS_CLASSES[self.type]
  2528. def get_status_class(self):
  2529. return STATUS_CLASSES[self.status]
  2530. #
  2531. # Cables
  2532. #
  2533. class Cable(ChangeLoggedModel):
  2534. """
  2535. A physical connection between two endpoints.
  2536. """
  2537. termination_a_type = models.ForeignKey(
  2538. to=ContentType,
  2539. limit_choices_to={'model__in': CABLE_TERMINATION_TYPES},
  2540. on_delete=models.PROTECT,
  2541. related_name='+'
  2542. )
  2543. termination_a_id = models.PositiveIntegerField()
  2544. termination_a = GenericForeignKey(
  2545. ct_field='termination_a_type',
  2546. fk_field='termination_a_id'
  2547. )
  2548. termination_b_type = models.ForeignKey(
  2549. to=ContentType,
  2550. limit_choices_to={'model__in': CABLE_TERMINATION_TYPES},
  2551. on_delete=models.PROTECT,
  2552. related_name='+'
  2553. )
  2554. termination_b_id = models.PositiveIntegerField()
  2555. termination_b = GenericForeignKey(
  2556. ct_field='termination_b_type',
  2557. fk_field='termination_b_id'
  2558. )
  2559. type = models.PositiveSmallIntegerField(
  2560. choices=CABLE_TYPE_CHOICES,
  2561. blank=True,
  2562. null=True
  2563. )
  2564. status = models.BooleanField(
  2565. choices=CONNECTION_STATUS_CHOICES,
  2566. default=CONNECTION_STATUS_CONNECTED
  2567. )
  2568. label = models.CharField(
  2569. max_length=100,
  2570. blank=True
  2571. )
  2572. color = ColorField(
  2573. blank=True
  2574. )
  2575. length = models.PositiveSmallIntegerField(
  2576. blank=True,
  2577. null=True
  2578. )
  2579. length_unit = models.PositiveSmallIntegerField(
  2580. choices=CABLE_LENGTH_UNIT_CHOICES,
  2581. blank=True,
  2582. null=True
  2583. )
  2584. # Stores the normalized length (in meters) for database ordering
  2585. _abs_length = models.DecimalField(
  2586. max_digits=10,
  2587. decimal_places=4,
  2588. blank=True,
  2589. null=True
  2590. )
  2591. # Cache the associated device (where applicable) for the A and B terminations. This enables filtering of Cables by
  2592. # their associated Devices.
  2593. _termination_a_device = models.ForeignKey(
  2594. to=Device,
  2595. on_delete=models.CASCADE,
  2596. related_name='+',
  2597. blank=True,
  2598. null=True
  2599. )
  2600. _termination_b_device = models.ForeignKey(
  2601. to=Device,
  2602. on_delete=models.CASCADE,
  2603. related_name='+',
  2604. blank=True,
  2605. null=True
  2606. )
  2607. csv_headers = [
  2608. 'termination_a_type', 'termination_a_id', 'termination_b_type', 'termination_b_id', 'type', 'status', 'label',
  2609. 'color', 'length', 'length_unit',
  2610. ]
  2611. class Meta:
  2612. ordering = ['pk']
  2613. unique_together = (
  2614. ('termination_a_type', 'termination_a_id'),
  2615. ('termination_b_type', 'termination_b_id'),
  2616. )
  2617. def __init__(self, *args, **kwargs):
  2618. super().__init__(*args, **kwargs)
  2619. # A copy of the PK to be used by __str__ in case the object is deleted
  2620. self._pk = self.pk
  2621. def __str__(self):
  2622. return self.label or '#{}'.format(self._pk)
  2623. def get_absolute_url(self):
  2624. return reverse('dcim:cable', args=[self.pk])
  2625. def clean(self):
  2626. # Validate that termination A exists
  2627. if not hasattr(self, 'termination_a_type'):
  2628. raise ValidationError('Termination A type has not been specified')
  2629. try:
  2630. self.termination_a_type.model_class().objects.get(pk=self.termination_a_id)
  2631. except ObjectDoesNotExist:
  2632. raise ValidationError({
  2633. 'termination_a': 'Invalid ID for type {}'.format(self.termination_a_type)
  2634. })
  2635. # Validate that termination B exists
  2636. if not hasattr(self, 'termination_b_type'):
  2637. raise ValidationError('Termination B type has not been specified')
  2638. try:
  2639. self.termination_b_type.model_class().objects.get(pk=self.termination_b_id)
  2640. except ObjectDoesNotExist:
  2641. raise ValidationError({
  2642. 'termination_b': 'Invalid ID for type {}'.format(self.termination_b_type)
  2643. })
  2644. type_a = self.termination_a_type.model
  2645. type_b = self.termination_b_type.model
  2646. # Validate interface types
  2647. if type_a == 'interface' and self.termination_a.type in NONCONNECTABLE_IFACE_TYPES:
  2648. raise ValidationError({
  2649. 'termination_a_id': 'Cables cannot be terminated to {} interfaces'.format(
  2650. self.termination_a.get_type_display()
  2651. )
  2652. })
  2653. if type_b == 'interface' and self.termination_b.type in NONCONNECTABLE_IFACE_TYPES:
  2654. raise ValidationError({
  2655. 'termination_b_id': 'Cables cannot be terminated to {} interfaces'.format(
  2656. self.termination_b.get_type_display()
  2657. )
  2658. })
  2659. # Check that termination types are compatible
  2660. if type_b not in COMPATIBLE_TERMINATION_TYPES.get(type_a):
  2661. raise ValidationError("Incompatible termination types: {} and {}".format(
  2662. self.termination_a_type, self.termination_b_type
  2663. ))
  2664. # A component with multiple positions must be connected to a component with an equal number of positions
  2665. term_a_positions = getattr(self.termination_a, 'positions', 1)
  2666. term_b_positions = getattr(self.termination_b, 'positions', 1)
  2667. if term_a_positions != term_b_positions:
  2668. raise ValidationError(
  2669. "{} has {} positions and {} has {}. Both terminations must have the same number of positions.".format(
  2670. self.termination_a, term_a_positions, self.termination_b, term_b_positions
  2671. )
  2672. )
  2673. # A termination point cannot be connected to itself
  2674. if self.termination_a == self.termination_b:
  2675. raise ValidationError("Cannot connect {} to itself".format(self.termination_a_type))
  2676. # A front port cannot be connected to its corresponding rear port
  2677. if (
  2678. type_a in ['frontport', 'rearport'] and
  2679. type_b in ['frontport', 'rearport'] and
  2680. (
  2681. getattr(self.termination_a, 'rear_port', None) == self.termination_b or
  2682. getattr(self.termination_b, 'rear_port', None) == self.termination_a
  2683. )
  2684. ):
  2685. raise ValidationError("A front port cannot be connected to it corresponding rear port")
  2686. # Check for an existing Cable connected to either termination object
  2687. if self.termination_a.cable not in (None, self):
  2688. raise ValidationError("{} already has a cable attached (#{})".format(
  2689. self.termination_a, self.termination_a.cable_id
  2690. ))
  2691. if self.termination_b.cable not in (None, self):
  2692. raise ValidationError("{} already has a cable attached (#{})".format(
  2693. self.termination_b, self.termination_b.cable_id
  2694. ))
  2695. # Validate length and length_unit
  2696. if self.length is not None and self.length_unit is None:
  2697. raise ValidationError("Must specify a unit when setting a cable length")
  2698. elif self.length is None:
  2699. self.length_unit = None
  2700. def save(self, *args, **kwargs):
  2701. # Store the given length (if any) in meters for use in database ordering
  2702. if self.length and self.length_unit:
  2703. self._abs_length = to_meters(self.length, self.length_unit)
  2704. else:
  2705. self._abs_length = None
  2706. # Store the parent Device for the A and B terminations (if applicable) to enable filtering
  2707. if hasattr(self.termination_a, 'device'):
  2708. self._termination_a_device = self.termination_a.device
  2709. if hasattr(self.termination_b, 'device'):
  2710. self._termination_b_device = self.termination_b.device
  2711. super().save(*args, **kwargs)
  2712. def to_csv(self):
  2713. return (
  2714. '{}.{}'.format(self.termination_a_type.app_label, self.termination_a_type.model),
  2715. self.termination_a_id,
  2716. '{}.{}'.format(self.termination_b_type.app_label, self.termination_b_type.model),
  2717. self.termination_b_id,
  2718. self.get_type_display(),
  2719. self.get_status_display(),
  2720. self.label,
  2721. self.color,
  2722. self.length,
  2723. self.length_unit,
  2724. )
  2725. def get_status_class(self):
  2726. return 'success' if self.status else 'info'
  2727. def get_compatible_types(self):
  2728. """
  2729. Return all termination types compatible with termination A.
  2730. """
  2731. if self.termination_a is None:
  2732. return
  2733. return COMPATIBLE_TERMINATION_TYPES[self.termination_a._meta.model_name]
  2734. def get_path_endpoints(self):
  2735. """
  2736. Traverse both ends of a cable path and return its connected endpoints. Note that one or both endpoints may be
  2737. None.
  2738. """
  2739. a_path = self.termination_b.trace()
  2740. b_path = self.termination_a.trace()
  2741. # Determine overall path status (connected or planned)
  2742. if self.status == CONNECTION_STATUS_PLANNED:
  2743. path_status = CONNECTION_STATUS_PLANNED
  2744. else:
  2745. path_status = CONNECTION_STATUS_CONNECTED
  2746. for segment in a_path[1:] + b_path[1:]:
  2747. if segment[1] is None or segment[1].status == CONNECTION_STATUS_PLANNED:
  2748. path_status = CONNECTION_STATUS_PLANNED
  2749. break
  2750. a_endpoint = a_path[-1][2]
  2751. b_endpoint = b_path[-1][2]
  2752. return a_endpoint, b_endpoint, path_status