mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-23 04:22:01 -06:00
Refactor SVG module
This commit is contained in:
jeremystretch
parent
3362bc3106
commit
cf7a091319
2
netbox/dcim/svg/__init__.py
Normal file
2
netbox/dcim/svg/__init__.py
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
from .cables import *
|
||||||
|
from .racks import *
|
||||||
@ -4,269 +4,15 @@ from svgwrite.shapes import Line, Rect
|
|||||||
from svgwrite.text import Text
|
from svgwrite.text import Text
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.urls import reverse
|
|
||||||
from django.utils.http import urlencode
|
|
||||||
|
|
||||||
from utilities.utils import foreground_color
|
from utilities.utils import foreground_color
|
||||||
from .choices import DeviceFaceChoices
|
|
||||||
from .constants import RACK_ELEVATION_BORDER_WIDTH
|
|
||||||
|
|
||||||
|
|
||||||
__all__ = (
|
__all__ = (
|
||||||
'CableTraceSVG',
|
'CableTraceSVG',
|
||||||
'RackElevationSVG',
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def get_device_name(device):
|
|
||||||
if device.virtual_chassis:
|
|
||||||
return f'{device.virtual_chassis.name}:{device.vc_position}'
|
|
||||||
elif device.name:
|
|
||||||
return device.name
|
|
||||||
else:
|
|
||||||
return str(device.device_type)
|
|
||||||
|
|
||||||
|
|
||||||
class RackElevationSVG:
|
|
||||||
"""
|
|
||||||
Use this class to render a rack elevation as an SVG image.
|
|
||||||
|
|
||||||
:param rack: A NetBox Rack instance
|
|
||||||
:param user: User instance. If specified, only devices viewable by this user will be fully displayed.
|
|
||||||
:param include_images: If true, the SVG document will embed front/rear device face images, where available
|
|
||||||
:param base_url: Base URL for links within the SVG document. If none, links will be relative.
|
|
||||||
"""
|
|
||||||
def __init__(self, rack, user=None, include_images=True, base_url=None):
|
|
||||||
self.rack = rack
|
|
||||||
self.include_images = include_images
|
|
||||||
if base_url is not None:
|
|
||||||
self.base_url = base_url.rstrip('/')
|
|
||||||
else:
|
|
||||||
self.base_url = ''
|
|
||||||
|
|
||||||
# Determine the subset of devices within this rack that are viewable by the user, if any
|
|
||||||
permitted_devices = self.rack.devices
|
|
||||||
if user is not None:
|
|
||||||
permitted_devices = permitted_devices.restrict(user, 'view')
|
|
||||||
self.permitted_device_ids = permitted_devices.values_list('pk', flat=True)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _get_device_description(device):
|
|
||||||
return '{} ({}) — {} {} ({}U) {} {}'.format(
|
|
||||||
device.name,
|
|
||||||
device.device_role,
|
|
||||||
device.device_type.manufacturer.name,
|
|
||||||
device.device_type.model,
|
|
||||||
device.device_type.u_height,
|
|
||||||
device.asset_tag or '',
|
|
||||||
device.serial or ''
|
|
||||||
)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _add_gradient(drawing, id_, color):
|
|
||||||
gradient = drawing.linearGradient(
|
|
||||||
start=(0, 0),
|
|
||||||
end=(0, 25),
|
|
||||||
spreadMethod='repeat',
|
|
||||||
id_=id_,
|
|
||||||
gradientTransform='rotate(45, 0, 0)',
|
|
||||||
gradientUnits='userSpaceOnUse'
|
|
||||||
)
|
|
||||||
gradient.add_stop_color(offset='0%', color='#f7f7f7')
|
|
||||||
gradient.add_stop_color(offset='50%', color='#f7f7f7')
|
|
||||||
gradient.add_stop_color(offset='50%', color=color)
|
|
||||||
gradient.add_stop_color(offset='100%', color=color)
|
|
||||||
drawing.defs.add(gradient)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _setup_drawing(width, height):
|
|
||||||
drawing = svgwrite.Drawing(size=(width, height))
|
|
||||||
|
|
||||||
# add the stylesheet
|
|
||||||
with open('{}/rack_elevation.css'.format(settings.STATIC_ROOT)) as css_file:
|
|
||||||
drawing.defs.add(drawing.style(css_file.read()))
|
|
||||||
|
|
||||||
# add gradients
|
|
||||||
RackElevationSVG._add_gradient(drawing, 'reserved', '#c7c7ff')
|
|
||||||
RackElevationSVG._add_gradient(drawing, 'occupied', '#d7d7d7')
|
|
||||||
RackElevationSVG._add_gradient(drawing, 'blocked', '#ffc0c0')
|
|
||||||
|
|
||||||
return drawing
|
|
||||||
|
|
||||||
def _draw_device_front(self, drawing, device, start, end, text):
|
|
||||||
name = get_device_name(device)
|
|
||||||
if device.devicebay_count:
|
|
||||||
name += ' ({}/{})'.format(device.get_children().count(), device.devicebay_count)
|
|
||||||
|
|
||||||
color = device.device_role.color
|
|
||||||
link = drawing.add(
|
|
||||||
drawing.a(
|
|
||||||
href='{}{}'.format(self.base_url, reverse('dcim:device', kwargs={'pk': device.pk})),
|
|
||||||
target='_top',
|
|
||||||
fill='black'
|
|
||||||
)
|
|
||||||
)
|
|
||||||
link.set_desc(self._get_device_description(device))
|
|
||||||
link.add(drawing.rect(start, end, style='fill: #{}'.format(color), class_='slot'))
|
|
||||||
hex_color = '#{}'.format(foreground_color(color))
|
|
||||||
link.add(drawing.text(str(name), insert=text, fill=hex_color))
|
|
||||||
|
|
||||||
# Embed front device type image if one exists
|
|
||||||
if self.include_images and device.device_type.front_image:
|
|
||||||
image = drawing.image(
|
|
||||||
href=device.device_type.front_image.url,
|
|
||||||
insert=start,
|
|
||||||
size=end,
|
|
||||||
class_='device-image'
|
|
||||||
)
|
|
||||||
image.fit(scale='slice')
|
|
||||||
link.add(image)
|
|
||||||
link.add(drawing.text(str(name), insert=text, stroke='black',
|
|
||||||
stroke_width='0.2em', stroke_linejoin='round', class_='device-image-label'))
|
|
||||||
link.add(drawing.text(str(name), insert=text, fill='white', class_='device-image-label'))
|
|
||||||
|
|
||||||
def _draw_device_rear(self, drawing, device, start, end, text):
|
|
||||||
link = drawing.add(
|
|
||||||
drawing.a(
|
|
||||||
href='{}{}'.format(self.base_url, reverse('dcim:device', kwargs={'pk': device.pk})),
|
|
||||||
target='_top',
|
|
||||||
fill='black'
|
|
||||||
)
|
|
||||||
)
|
|
||||||
link.set_desc(self._get_device_description(device))
|
|
||||||
link.add(drawing.rect(start, end, class_="slot blocked"))
|
|
||||||
link.add(drawing.text(get_device_name(device), insert=text))
|
|
||||||
|
|
||||||
# Embed rear device type image if one exists
|
|
||||||
if self.include_images and device.device_type.rear_image:
|
|
||||||
image = drawing.image(
|
|
||||||
href=device.device_type.rear_image.url,
|
|
||||||
insert=start,
|
|
||||||
size=end,
|
|
||||||
class_='device-image'
|
|
||||||
)
|
|
||||||
image.fit(scale='slice')
|
|
||||||
link.add(image)
|
|
||||||
link.add(drawing.text(get_device_name(device), insert=text, stroke='black',
|
|
||||||
stroke_width='0.2em', stroke_linejoin='round', class_='device-image-label'))
|
|
||||||
link.add(drawing.text(get_device_name(device), insert=text, fill='white', class_='device-image-label'))
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _draw_empty(drawing, rack, start, end, text, id_, face_id, class_, reservation):
|
|
||||||
link_url = '{}?{}'.format(
|
|
||||||
reverse('dcim:device_add'),
|
|
||||||
urlencode({
|
|
||||||
'site': rack.site.pk,
|
|
||||||
'location': rack.location.pk if rack.location else '',
|
|
||||||
'rack': rack.pk,
|
|
||||||
'face': face_id,
|
|
||||||
'position': id_
|
|
||||||
})
|
|
||||||
)
|
|
||||||
link = drawing.add(
|
|
||||||
drawing.a(href=link_url, target='_top')
|
|
||||||
)
|
|
||||||
if reservation:
|
|
||||||
link.set_desc('{} — {} · {}'.format(
|
|
||||||
reservation.description, reservation.user, reservation.created
|
|
||||||
))
|
|
||||||
link.add(drawing.rect(start, end, class_=class_))
|
|
||||||
link.add(drawing.text("add device", insert=text, class_='add-device'))
|
|
||||||
|
|
||||||
def merge_elevations(self, face):
|
|
||||||
elevation = self.rack.get_rack_units(face=face, expand_devices=False)
|
|
||||||
if face == DeviceFaceChoices.FACE_REAR:
|
|
||||||
other_face = DeviceFaceChoices.FACE_FRONT
|
|
||||||
else:
|
|
||||||
other_face = DeviceFaceChoices.FACE_REAR
|
|
||||||
other = self.rack.get_rack_units(face=other_face)
|
|
||||||
|
|
||||||
unit_cursor = 0
|
|
||||||
for u in elevation:
|
|
||||||
o = other[unit_cursor]
|
|
||||||
if not u['device'] and o['device'] and o['device'].device_type.is_full_depth:
|
|
||||||
u['device'] = o['device']
|
|
||||||
u['height'] = 1
|
|
||||||
unit_cursor += u.get('height', 1)
|
|
||||||
|
|
||||||
return elevation
|
|
||||||
|
|
||||||
def render(self, face, unit_width, unit_height, legend_width):
|
|
||||||
"""
|
|
||||||
Return an SVG document representing a rack elevation.
|
|
||||||
"""
|
|
||||||
drawing = self._setup_drawing(
|
|
||||||
unit_width + legend_width + RACK_ELEVATION_BORDER_WIDTH * 2,
|
|
||||||
unit_height * self.rack.u_height + RACK_ELEVATION_BORDER_WIDTH * 2
|
|
||||||
)
|
|
||||||
reserved_units = self.rack.get_reserved_units()
|
|
||||||
|
|
||||||
unit_cursor = 0
|
|
||||||
for ru in range(0, self.rack.u_height):
|
|
||||||
start_y = ru * unit_height
|
|
||||||
position_coordinates = (legend_width / 2, start_y + unit_height / 2 + RACK_ELEVATION_BORDER_WIDTH)
|
|
||||||
unit = ru + 1 if self.rack.desc_units else self.rack.u_height - ru
|
|
||||||
drawing.add(
|
|
||||||
drawing.text(str(unit), position_coordinates, class_="unit")
|
|
||||||
)
|
|
||||||
|
|
||||||
for unit in self.merge_elevations(face):
|
|
||||||
|
|
||||||
# Loop through all units in the elevation
|
|
||||||
device = unit['device']
|
|
||||||
height = unit.get('height', 1)
|
|
||||||
|
|
||||||
# Setup drawing coordinates
|
|
||||||
x_offset = legend_width + RACK_ELEVATION_BORDER_WIDTH
|
|
||||||
y_offset = unit_cursor * unit_height + RACK_ELEVATION_BORDER_WIDTH
|
|
||||||
end_y = unit_height * height
|
|
||||||
start_cordinates = (x_offset, y_offset)
|
|
||||||
end_cordinates = (unit_width, end_y)
|
|
||||||
text_cordinates = (x_offset + (unit_width / 2), y_offset + end_y / 2)
|
|
||||||
|
|
||||||
# Draw the device
|
|
||||||
if device and device.face == face and device.pk in self.permitted_device_ids:
|
|
||||||
self._draw_device_front(drawing, device, start_cordinates, end_cordinates, text_cordinates)
|
|
||||||
elif device and device.device_type.is_full_depth and device.pk in self.permitted_device_ids:
|
|
||||||
self._draw_device_rear(drawing, device, start_cordinates, end_cordinates, text_cordinates)
|
|
||||||
elif device:
|
|
||||||
# Devices which the user does not have permission to view are rendered only as unavailable space
|
|
||||||
drawing.add(drawing.rect(start_cordinates, end_cordinates, class_='blocked'))
|
|
||||||
else:
|
|
||||||
# Draw shallow devices, reservations, or empty units
|
|
||||||
class_ = 'slot'
|
|
||||||
reservation = reserved_units.get(unit["id"])
|
|
||||||
if device:
|
|
||||||
class_ += ' occupied'
|
|
||||||
if reservation:
|
|
||||||
class_ += ' reserved'
|
|
||||||
self._draw_empty(
|
|
||||||
drawing,
|
|
||||||
self.rack,
|
|
||||||
start_cordinates,
|
|
||||||
end_cordinates,
|
|
||||||
text_cordinates,
|
|
||||||
unit["id"],
|
|
||||||
face,
|
|
||||||
class_,
|
|
||||||
reservation
|
|
||||||
)
|
|
||||||
|
|
||||||
unit_cursor += height
|
|
||||||
|
|
||||||
# Wrap the drawing with a border
|
|
||||||
border_width = RACK_ELEVATION_BORDER_WIDTH
|
|
||||||
border_offset = RACK_ELEVATION_BORDER_WIDTH / 2
|
|
||||||
frame = drawing.rect(
|
|
||||||
insert=(legend_width + border_offset, border_offset),
|
|
||||||
size=(unit_width + border_width, self.rack.u_height * unit_height + border_width),
|
|
||||||
class_='rack'
|
|
||||||
)
|
|
||||||
drawing.add(frame)
|
|
||||||
|
|
||||||
return drawing
|
|
||||||
|
|
||||||
|
|
||||||
OFFSET = 0.5
|
OFFSET = 0.5
|
||||||
PADDING = 10
|
PADDING = 10
|
||||||
LINE_HEIGHT = 20
|
LINE_HEIGHT = 20
|
||||||
263
netbox/dcim/svg/racks.py
Normal file
263
netbox/dcim/svg/racks.py
Normal file
@ -0,0 +1,263 @@
|
|||||||
|
import svgwrite
|
||||||
|
|
||||||
|
from django.conf import settings
|
||||||
|
from django.urls import reverse
|
||||||
|
from django.utils.http import urlencode
|
||||||
|
|
||||||
|
from utilities.utils import foreground_color
|
||||||
|
from dcim.choices import DeviceFaceChoices
|
||||||
|
from dcim.constants import RACK_ELEVATION_BORDER_WIDTH
|
||||||
|
|
||||||
|
|
||||||
|
__all__ = (
|
||||||
|
'RackElevationSVG',
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def get_device_name(device):
|
||||||
|
if device.virtual_chassis:
|
||||||
|
return f'{device.virtual_chassis.name}:{device.vc_position}'
|
||||||
|
elif device.name:
|
||||||
|
return device.name
|
||||||
|
else:
|
||||||
|
return str(device.device_type)
|
||||||
|
|
||||||
|
|
||||||
|
class RackElevationSVG:
|
||||||
|
"""
|
||||||
|
Use this class to render a rack elevation as an SVG image.
|
||||||
|
|
||||||
|
:param rack: A NetBox Rack instance
|
||||||
|
:param user: User instance. If specified, only devices viewable by this user will be fully displayed.
|
||||||
|
:param include_images: If true, the SVG document will embed front/rear device face images, where available
|
||||||
|
:param base_url: Base URL for links within the SVG document. If none, links will be relative.
|
||||||
|
"""
|
||||||
|
def __init__(self, rack, user=None, include_images=True, base_url=None):
|
||||||
|
self.rack = rack
|
||||||
|
self.include_images = include_images
|
||||||
|
if base_url is not None:
|
||||||
|
self.base_url = base_url.rstrip('/')
|
||||||
|
else:
|
||||||
|
self.base_url = ''
|
||||||
|
|
||||||
|
# Determine the subset of devices within this rack that are viewable by the user, if any
|
||||||
|
permitted_devices = self.rack.devices
|
||||||
|
if user is not None:
|
||||||
|
permitted_devices = permitted_devices.restrict(user, 'view')
|
||||||
|
self.permitted_device_ids = permitted_devices.values_list('pk', flat=True)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _get_device_description(device):
|
||||||
|
return '{} ({}) — {} {} ({}U) {} {}'.format(
|
||||||
|
device.name,
|
||||||
|
device.device_role,
|
||||||
|
device.device_type.manufacturer.name,
|
||||||
|
device.device_type.model,
|
||||||
|
device.device_type.u_height,
|
||||||
|
device.asset_tag or '',
|
||||||
|
device.serial or ''
|
||||||
|
)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _add_gradient(drawing, id_, color):
|
||||||
|
gradient = drawing.linearGradient(
|
||||||
|
start=(0, 0),
|
||||||
|
end=(0, 25),
|
||||||
|
spreadMethod='repeat',
|
||||||
|
id_=id_,
|
||||||
|
gradientTransform='rotate(45, 0, 0)',
|
||||||
|
gradientUnits='userSpaceOnUse'
|
||||||
|
)
|
||||||
|
gradient.add_stop_color(offset='0%', color='#f7f7f7')
|
||||||
|
gradient.add_stop_color(offset='50%', color='#f7f7f7')
|
||||||
|
gradient.add_stop_color(offset='50%', color=color)
|
||||||
|
gradient.add_stop_color(offset='100%', color=color)
|
||||||
|
drawing.defs.add(gradient)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _setup_drawing(width, height):
|
||||||
|
drawing = svgwrite.Drawing(size=(width, height))
|
||||||
|
|
||||||
|
# add the stylesheet
|
||||||
|
with open('{}/rack_elevation.css'.format(settings.STATIC_ROOT)) as css_file:
|
||||||
|
drawing.defs.add(drawing.style(css_file.read()))
|
||||||
|
|
||||||
|
# add gradients
|
||||||
|
RackElevationSVG._add_gradient(drawing, 'reserved', '#c7c7ff')
|
||||||
|
RackElevationSVG._add_gradient(drawing, 'occupied', '#d7d7d7')
|
||||||
|
RackElevationSVG._add_gradient(drawing, 'blocked', '#ffc0c0')
|
||||||
|
|
||||||
|
return drawing
|
||||||
|
|
||||||
|
def _draw_device_front(self, drawing, device, start, end, text):
|
||||||
|
name = get_device_name(device)
|
||||||
|
if device.devicebay_count:
|
||||||
|
name += ' ({}/{})'.format(device.get_children().count(), device.devicebay_count)
|
||||||
|
|
||||||
|
color = device.device_role.color
|
||||||
|
link = drawing.add(
|
||||||
|
drawing.a(
|
||||||
|
href='{}{}'.format(self.base_url, reverse('dcim:device', kwargs={'pk': device.pk})),
|
||||||
|
target='_top',
|
||||||
|
fill='black'
|
||||||
|
)
|
||||||
|
)
|
||||||
|
link.set_desc(self._get_device_description(device))
|
||||||
|
link.add(drawing.rect(start, end, style='fill: #{}'.format(color), class_='slot'))
|
||||||
|
hex_color = '#{}'.format(foreground_color(color))
|
||||||
|
link.add(drawing.text(str(name), insert=text, fill=hex_color))
|
||||||
|
|
||||||
|
# Embed front device type image if one exists
|
||||||
|
if self.include_images and device.device_type.front_image:
|
||||||
|
image = drawing.image(
|
||||||
|
href=device.device_type.front_image.url,
|
||||||
|
insert=start,
|
||||||
|
size=end,
|
||||||
|
class_='device-image'
|
||||||
|
)
|
||||||
|
image.fit(scale='slice')
|
||||||
|
link.add(image)
|
||||||
|
link.add(drawing.text(str(name), insert=text, stroke='black',
|
||||||
|
stroke_width='0.2em', stroke_linejoin='round', class_='device-image-label'))
|
||||||
|
link.add(drawing.text(str(name), insert=text, fill='white', class_='device-image-label'))
|
||||||
|
|
||||||
|
def _draw_device_rear(self, drawing, device, start, end, text):
|
||||||
|
link = drawing.add(
|
||||||
|
drawing.a(
|
||||||
|
href='{}{}'.format(self.base_url, reverse('dcim:device', kwargs={'pk': device.pk})),
|
||||||
|
target='_top',
|
||||||
|
fill='black'
|
||||||
|
)
|
||||||
|
)
|
||||||
|
link.set_desc(self._get_device_description(device))
|
||||||
|
link.add(drawing.rect(start, end, class_="slot blocked"))
|
||||||
|
link.add(drawing.text(get_device_name(device), insert=text))
|
||||||
|
|
||||||
|
# Embed rear device type image if one exists
|
||||||
|
if self.include_images and device.device_type.rear_image:
|
||||||
|
image = drawing.image(
|
||||||
|
href=device.device_type.rear_image.url,
|
||||||
|
insert=start,
|
||||||
|
size=end,
|
||||||
|
class_='device-image'
|
||||||
|
)
|
||||||
|
image.fit(scale='slice')
|
||||||
|
link.add(image)
|
||||||
|
link.add(drawing.text(get_device_name(device), insert=text, stroke='black',
|
||||||
|
stroke_width='0.2em', stroke_linejoin='round', class_='device-image-label'))
|
||||||
|
link.add(drawing.text(get_device_name(device), insert=text, fill='white', class_='device-image-label'))
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _draw_empty(drawing, rack, start, end, text, id_, face_id, class_, reservation):
|
||||||
|
link_url = '{}?{}'.format(
|
||||||
|
reverse('dcim:device_add'),
|
||||||
|
urlencode({
|
||||||
|
'site': rack.site.pk,
|
||||||
|
'location': rack.location.pk if rack.location else '',
|
||||||
|
'rack': rack.pk,
|
||||||
|
'face': face_id,
|
||||||
|
'position': id_
|
||||||
|
})
|
||||||
|
)
|
||||||
|
link = drawing.add(
|
||||||
|
drawing.a(href=link_url, target='_top')
|
||||||
|
)
|
||||||
|
if reservation:
|
||||||
|
link.set_desc('{} — {} · {}'.format(
|
||||||
|
reservation.description, reservation.user, reservation.created
|
||||||
|
))
|
||||||
|
link.add(drawing.rect(start, end, class_=class_))
|
||||||
|
link.add(drawing.text("add device", insert=text, class_='add-device'))
|
||||||
|
|
||||||
|
def merge_elevations(self, face):
|
||||||
|
elevation = self.rack.get_rack_units(face=face, expand_devices=False)
|
||||||
|
if face == DeviceFaceChoices.FACE_REAR:
|
||||||
|
other_face = DeviceFaceChoices.FACE_FRONT
|
||||||
|
else:
|
||||||
|
other_face = DeviceFaceChoices.FACE_REAR
|
||||||
|
other = self.rack.get_rack_units(face=other_face)
|
||||||
|
|
||||||
|
unit_cursor = 0
|
||||||
|
for u in elevation:
|
||||||
|
o = other[unit_cursor]
|
||||||
|
if not u['device'] and o['device'] and o['device'].device_type.is_full_depth:
|
||||||
|
u['device'] = o['device']
|
||||||
|
u['height'] = 1
|
||||||
|
unit_cursor += u.get('height', 1)
|
||||||
|
|
||||||
|
return elevation
|
||||||
|
|
||||||
|
def render(self, face, unit_width, unit_height, legend_width):
|
||||||
|
"""
|
||||||
|
Return an SVG document representing a rack elevation.
|
||||||
|
"""
|
||||||
|
drawing = self._setup_drawing(
|
||||||
|
unit_width + legend_width + RACK_ELEVATION_BORDER_WIDTH * 2,
|
||||||
|
unit_height * self.rack.u_height + RACK_ELEVATION_BORDER_WIDTH * 2
|
||||||
|
)
|
||||||
|
reserved_units = self.rack.get_reserved_units()
|
||||||
|
|
||||||
|
unit_cursor = 0
|
||||||
|
for ru in range(0, self.rack.u_height):
|
||||||
|
start_y = ru * unit_height
|
||||||
|
position_coordinates = (legend_width / 2, start_y + unit_height / 2 + RACK_ELEVATION_BORDER_WIDTH)
|
||||||
|
unit = ru + 1 if self.rack.desc_units else self.rack.u_height - ru
|
||||||
|
drawing.add(
|
||||||
|
drawing.text(str(unit), position_coordinates, class_="unit")
|
||||||
|
)
|
||||||
|
|
||||||
|
for unit in self.merge_elevations(face):
|
||||||
|
|
||||||
|
# Loop through all units in the elevation
|
||||||
|
device = unit['device']
|
||||||
|
height = unit.get('height', 1)
|
||||||
|
|
||||||
|
# Setup drawing coordinates
|
||||||
|
x_offset = legend_width + RACK_ELEVATION_BORDER_WIDTH
|
||||||
|
y_offset = unit_cursor * unit_height + RACK_ELEVATION_BORDER_WIDTH
|
||||||
|
end_y = unit_height * height
|
||||||
|
start_cordinates = (x_offset, y_offset)
|
||||||
|
end_cordinates = (unit_width, end_y)
|
||||||
|
text_cordinates = (x_offset + (unit_width / 2), y_offset + end_y / 2)
|
||||||
|
|
||||||
|
# Draw the device
|
||||||
|
if device and device.face == face and device.pk in self.permitted_device_ids:
|
||||||
|
self._draw_device_front(drawing, device, start_cordinates, end_cordinates, text_cordinates)
|
||||||
|
elif device and device.device_type.is_full_depth and device.pk in self.permitted_device_ids:
|
||||||
|
self._draw_device_rear(drawing, device, start_cordinates, end_cordinates, text_cordinates)
|
||||||
|
elif device:
|
||||||
|
# Devices which the user does not have permission to view are rendered only as unavailable space
|
||||||
|
drawing.add(drawing.rect(start_cordinates, end_cordinates, class_='blocked'))
|
||||||
|
else:
|
||||||
|
# Draw shallow devices, reservations, or empty units
|
||||||
|
class_ = 'slot'
|
||||||
|
reservation = reserved_units.get(unit["id"])
|
||||||
|
if device:
|
||||||
|
class_ += ' occupied'
|
||||||
|
if reservation:
|
||||||
|
class_ += ' reserved'
|
||||||
|
self._draw_empty(
|
||||||
|
drawing,
|
||||||
|
self.rack,
|
||||||
|
start_cordinates,
|
||||||
|
end_cordinates,
|
||||||
|
text_cordinates,
|
||||||
|
unit["id"],
|
||||||
|
face,
|
||||||
|
class_,
|
||||||
|
reservation
|
||||||
|
)
|
||||||
|
|
||||||
|
unit_cursor += height
|
||||||
|
|
||||||
|
# Wrap the drawing with a border
|
||||||
|
border_width = RACK_ELEVATION_BORDER_WIDTH
|
||||||
|
border_offset = RACK_ELEVATION_BORDER_WIDTH / 2
|
||||||
|
frame = drawing.rect(
|
||||||
|
insert=(legend_width + border_offset, border_offset),
|
||||||
|
size=(unit_width + border_width, self.rack.u_height * unit_height + border_width),
|
||||||
|
class_='rack'
|
||||||
|
)
|
||||||
|
drawing.add(frame)
|
||||||
|
|
||||||
|
return drawing
|
||||||
Loading…
Reference in New Issue
Block a user