120 lines
4.3 KiB
Python
120 lines
4.3 KiB
Python
# Copyright 2015 Mirantis, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from pprint import pprint
|
|
import svgwrite
|
|
from svgwrite import cm
|
|
|
|
|
|
class StdoutViewer(object):
|
|
|
|
def __init__(self, disks_spaces_mapping):
|
|
self.disks_spaces_mapping = disks_spaces_mapping
|
|
|
|
def show_me(self):
|
|
pprint(self.disks_spaces_mapping)
|
|
|
|
|
|
class SVGViewer(object):
|
|
|
|
PALETTE = ['#666547', '#fb2e01', '#6fcb9f', '#ffe28a', '#b9feb3',
|
|
'#404040', '#4cc3d9', '#7bc8a4', '#ffc65d', '#f16745']
|
|
DISK_HEIGHT = 2 * cm
|
|
SPACE_HEIGHT = 2 * cm
|
|
DISKS_INTERVAL = 110
|
|
SPACES_X_INTERVAL = 2
|
|
STYLE = "fill:{color};stroke:black;stroke-width:5;"
|
|
|
|
def __init__(self, disks_spaces_mapping,
|
|
file_path='/tmp/bareon.svg', fit=False):
|
|
|
|
self.disks_spaces_mapping = disks_spaces_mapping
|
|
max_disk_size = max([i['size'] for i in disks_spaces_mapping])
|
|
self.width_multiplier = 450.0 / max_disk_size
|
|
|
|
svg_height = len(disks_spaces_mapping) * (self.DISKS_INTERVAL + 10)
|
|
svg_width = self.width_multiplier * max_disk_size + 200
|
|
options = {}
|
|
if fit:
|
|
options = {
|
|
'preserveAspectRatio': 'none',
|
|
'viewBox': "0 0 {0} {1}".format(svg_width, svg_height)
|
|
if fit else '0 0 maxY maxX'}
|
|
|
|
self.dwg = svgwrite.Drawing(
|
|
filename=file_path,
|
|
debug=True,
|
|
**options)
|
|
|
|
def show_me(self):
|
|
self._add_disk_with_spaces()
|
|
self.dwg.save()
|
|
|
|
def _add_disk_with_spaces(self):
|
|
disk_g = self.dwg.add(self.dwg.g(
|
|
id='disks-group',
|
|
transform="translate({0}, {1})".format(30, 30)))
|
|
|
|
for disk_idx, disk_w_spaces in enumerate(self.disks_spaces_mapping):
|
|
disk_id = disk_w_spaces['disk_id']
|
|
size = disk_w_spaces['size']
|
|
|
|
disk = disk_g.add(self.dwg.g(
|
|
id=disk_id,
|
|
transform="translate(0, {0})".format(
|
|
disk_idx * self.DISKS_INTERVAL)))
|
|
|
|
disk.add(self.dwg.text(
|
|
text='{0} size={1}'.format(disk_id, size),
|
|
fill="black"))
|
|
|
|
disk_rect = disk.add(self.dwg.g(
|
|
transform="translate({0}, {1})".format(0, 10),
|
|
id='in-{0}'.format(disk_id)))
|
|
disk_rect.add(self.dwg.rect(
|
|
style=self.STYLE.format(color='#f5f5f5'),
|
|
ry=5,
|
|
rx=5,
|
|
size=(self.width_multiplier * disk_w_spaces['size'],
|
|
self.DISK_HEIGHT)))
|
|
|
|
last_insert = [0, 0]
|
|
for space_idx, space in enumerate(disk_w_spaces['spaces']):
|
|
palette = self.PALETTE[space_idx % len(self.PALETTE)]
|
|
disk_rect.add(self.dwg.rect(
|
|
style=self.STYLE.format(color=palette),
|
|
ry=5,
|
|
rx=5,
|
|
id=space['space_id'],
|
|
insert=last_insert,
|
|
size=(self.width_multiplier * space['size'],
|
|
self.SPACE_HEIGHT)))
|
|
|
|
last_insert[0] += self.width_multiplier * space['size']
|
|
|
|
spaces_lines = ['{0} size={1}'.format(space['space_id'],
|
|
space['size'])
|
|
for space in disk_w_spaces['spaces']]
|
|
|
|
last_insert[0] = self.width_multiplier * disk_w_spaces['size']
|
|
last_insert[0] += 10
|
|
last_insert[1] += 20
|
|
for space_idx, space_line in enumerate(spaces_lines):
|
|
palette = self.PALETTE[space_idx % len(self.PALETTE)]
|
|
disk_rect.add(self.dwg.text(
|
|
insert=last_insert,
|
|
text=space_line,
|
|
fill=palette))
|
|
last_insert[1] += 20
|