cellinfo
/
cocluto
2
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

reformatted code to pep8 convention

This commit is contained in:
Guillaume Raffy 2018-08-27 14:54:55 +00:00
parent 197e94d320
commit e9f3c5a784
1 changed files with 253 additions and 258 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

511
PowerDiagram.py
View File

@ -1,299 +1,294 @@
''' '''
The goal of this application is to generate a power diagram that will help system administrators to: The goal of this application is to generate a power diagram that will help system administrators to:
- document the power supply architecture - document the power supply architecture
- easily spot potential power overloads (for example if the power consumption exceeds the capacity of a cable) - easily spot potential power overloads (for example if the power consumption exceeds the capacity of a cable)
This application takes its input from a database, currently in the form of an sql dump, but it could easily be adapted to read directly from a mysql database This application takes its input from a database, currently in the form of an sql dump, but it could easily be adapted to read directly from a mysql database
''' '''
import sqlite3
import os
import re import re
import sys
import pygraphviz import pygraphviz
from inventory import Inventory from inventory import Inventory
from SimpaDbUtil import SqlFile, SqlDatabaseReader from SimpaDbUtil import SqlFile, SqlDatabaseReader
def add_capacity_constraints(capacity1, capacity2): def add_capacity_constraints(capacity1, capacity2):
""" """
combines 2 capacity constraints (max amperes) together combines 2 capacity constraints (max amperes) together
:param float capacity1: max amperes for the first capacity, None if there are no constraints :param float capacity1: max amperes for the first capacity, None if there are no constraints
:param float capacity2: max amperes for the second capacity, None if there are no constraints :param float capacity2: max amperes for the second capacity, None if there are no constraints
:return float: max amperes for the combined capacity, None if there are no constraints :return float: max amperes for the combined capacity, None if there are no constraints
""" """
if capacity1 is None: if capacity1 is None:
return capacity2 return capacity2
else: else:
if capacity2 is None: if capacity2 is None:
return capacity1 return capacity1
else: else:
return min(capacity1, capacity2) return min(capacity1, capacity2)
class Machine(object): class Machine(object):
""" """
represents a device with input and output power plugs. It could represent a power consumer such as a server (in which case, it has no output plug), or a power distrubtion unit (in which case, it has one input plug and more than one output plugs), or even something else... represents a device with input and output power plugs. It could represent a power consumer such as a server (in which case, it has no output plug), or a power distrubtion unit (in which case, it has one input plug and more than one output plugs), or even something else...
""" """
def __init__(self, name, power_config): def __init__(self, name, power_config):
self.name = name self.name = name
self.input_plugs = {} self.input_plugs = {}
self.output_plugs = {} self.output_plugs = {}
self.current_capacity_constraint = None # the maximum amperes in this connection self.current_capacity_constraint = None # the maximum amperes in this connection
self.power_config = power_config self.power_config = power_config
self.power_consumption = 0.0 self.power_consumption = 0.0
def get_plug(self, plug_name): def get_plug(self, plug_name):
plugs = {'i': self.input_plugs, 'o':self.output_plugs}[plug_name[0]] plugs = {'i': self.input_plugs, 'o': self.output_plugs}[plug_name[0]]
if plug_name not in plugs: if plug_name not in plugs:
plugs[plug_name] = Plug(plug_name, self, self.power_config) plugs[plug_name] = Plug(plug_name, self, self.power_config)
return plugs[plug_name] return plugs[plug_name]
def get_max_amperes(self): def get_max_amperes(self):
capacity = None capacity = None
if len(self.input_plugs) > 0: if len(self.input_plugs) > 0:
capacity = self.input_plugs.values()[0].get_max_amperes() capacity = self.input_plugs.values()[0].get_max_amperes()
capacity = add_capacity_constraints (capacity, self.current_capacity_constraint) capacity = add_capacity_constraints(capacity, self.current_capacity_constraint)
return capacity return capacity
def get_outgoing_connections(self): def get_outgoing_connections(self):
outgoing_connections = [] outgoing_connections = []
for conn in self.power_config.connections: for conn in self.power_config.connections:
if conn.from_plug.machine == self: if conn.from_plug.machine == self:
outgoing_connections.append(conn) outgoing_connections.append(conn)
return outgoing_connections return outgoing_connections
def get_power_consumption(self): def get_power_consumption(self):
power_consumption = self.power_consumption power_consumption = self.power_consumption
for conn in self.get_outgoing_connections(): for conn in self.get_outgoing_connections():
power_consumption += conn.get_power_consumption() power_consumption += conn.get_power_consumption()
# print("machine %s : power_consumption += %f" % (self.name, conn.get_power_consumption())) # print("machine %s : power_consumption += %f" % (self.name, conn.get_power_consumption()))
return power_consumption return power_consumption
class Plug(object): class Plug(object):
""" """
represents a power plug (input or output) of a device represents a power plug (input or output) of a device
""" """
def __init__(self, name, machine, power_config): def __init__(self, name, machine, power_config):
self.name = name self.name = name
self.machine = machine self.machine = machine
self.current_capacity_constraint = None # the maximum amperes in this connection self.current_capacity_constraint = None # the maximum amperes in this connection
self.power_config = power_config self.power_config = power_config
def __str__(self): def __str__(self):
return self.machine.name + '.' + self.name return self.machine.name + '.' + self.name
def get_incoming_connection(self): def get_incoming_connection(self):
return self.power_config.get_connection_to(self) return self.power_config.get_connection_to(self)
# def get_outgoing_connections(self): # def get_outgoing_connections(self):
# return self.power_config.get_connections_from(self) # return self.power_config.get_connections_from(self)
def is_input_plug(self): def is_input_plug(self):
return self.name[0] == 'i' return self.name[0] == 'i'
def set_current_capacity_constraint(self, max_amps): def set_current_capacity_constraint(self, max_amps):
self.current_capacity_constraint = max_amps self.current_capacity_constraint = max_amps
def get_max_amperes(self): def get_max_amperes(self):
capacity = None capacity = None
if self.is_input_plug(): if self.is_input_plug():
in_con = self.get_incoming_connection() in_con = self.get_incoming_connection()
if in_con: if in_con:
# apply incoming connection amperes limitation # apply incoming connection amperes limitation
capacity = add_capacity_constraints (capacity, in_con.get_max_amperes()) capacity = add_capacity_constraints(capacity, in_con.get_max_amperes())
# print(str(self)+ 'after incoming connection amperes limitation, capacity = ' + str(capacity)) # print(str(self)+ 'after incoming connection amperes limitation, capacity = ' + str(capacity))
else: else:
# apply the machine containing this plug's amperes limitation # apply the machine containing this plug's amperes limitation
capacity = add_capacity_constraints (capacity, self.machine.get_max_amperes()) capacity = add_capacity_constraints(capacity, self.machine.get_max_amperes())
# print(str(self)+'apply the machine containing this plug s amperes limitation, capacity = ' + str(capacity)) # print(str(self)+'apply the machine containing this plug s amperes limitation, capacity = ' + str(capacity))
# apply this plug's amperes limitation # apply this plug's amperes limitation
capacity = add_capacity_constraints (capacity, self.current_capacity_constraint) capacity = add_capacity_constraints(capacity, self.current_capacity_constraint)
# print(str(self)+'after apply this plug s amperes limitation, capacity = ' + str(capacity)) # print(str(self)+'after apply this plug s amperes limitation, capacity = ' + str(capacity))
return capacity return capacity
# def get_power_consumption(self): # def get_power_consumption(self):
# power_consumption = 0.0 # power_consumption = 0.0
# for conn in self.get_outgoing_connections(): # for conn in self.get_outgoing_connections():
# power_consumption += conn.get_power_consumption() # power_consumption += conn.get_power_consumption()
# power_consumption += self.get_power_consumption() # power_consumption += self.get_power_consumption()
# return self.from_plug.get_power_consumption() # return self.from_plug.get_power_consumption()
class Connection(object): class Connection(object):
""" """
a power cable connecting an input power plug to an output power plug a power cable connecting an input power plug to an output power plug
""" """
def __init__(self, from_plug, to_plug): def __init__(self, from_plug, to_plug):
self.from_plug = from_plug self.from_plug = from_plug
self.to_plug = to_plug self.to_plug = to_plug
self.current_capacity_constraint = None # the maximum amperes in this connection self.current_capacity_constraint = None # the maximum amperes in this connection
def __str__(self): def __str__(self):
return str(self.from_plug) + ' -> ' + str(self.to_plug) + ' (' + str(self.from_plug.get_max_amperes()) + str(self.get_max_amperes()) + 'A)' return str(self.from_plug) + ' -> ' + str(self.to_plug) + ' (' + str(self.from_plug.get_max_amperes()) + str(self.get_max_amperes()) + 'A)'
def get_max_amperes(self): def get_max_amperes(self):
capacity = self.from_plug.get_max_amperes() capacity = self.from_plug.get_max_amperes()
capacity = add_capacity_constraints (capacity, self.to_plug.current_capacity_constraint) capacity = add_capacity_constraints(capacity, self.to_plug.current_capacity_constraint)
if self.current_capacity_constraint is not None: if self.current_capacity_constraint is not None:
capacity = min(capacity, self.current_capacity_constraint) capacity = min(capacity, self.current_capacity_constraint)
return capacity return capacity
def is_redundancy_cable(self): def is_redundancy_cable(self):
return not (self.to_plug.name == 'i' or self.to_plug.name == 'i1') return not (self.to_plug.name == 'i' or self.to_plug.name == 'i1')
def get_power_consumption(self): def get_power_consumption(self):
# at the moment, this program doesn't handle redundant power supplies properly: # at the moment, this program doesn't handle redundant power supplies properly:
# the energy dragged by a power supply depends whether both power supplies are connnected to the same provider or not # the energy dragged by a power supply depends whether both power supplies are connnected to the same provider or not
if self.is_redundancy_cable(): if self.is_redundancy_cable():
return 0.0 # consider secondary power supplies to drag 0 power return 0.0 # consider secondary power supplies to drag 0 power
else: else:
return self.to_plug.machine.get_power_consumption() return self.to_plug.machine.get_power_consumption()
class PowerConfig(object): class PowerConfig(object):
""" """
the description of how machines are connected together (in terms of electric power) the description of how machines are connected together (in terms of electric power)
""" """
def __init__(self, simpa_db_sql_file_path): def __init__(self, simpa_db_sql_file_path):
self.machines = {} self.machines = {}
self.connections = [] self.connections = []
sql_source = SqlFile(simpa_db_sql_file_path)
sql_reader = SqlDatabaseReader(sql_source)
inventory = Inventory(sql_reader)
self._parse_from_inventory(inventory)
def _parse_from_inventory(self, inventory):
"""
:param Inventory inventory:
"""
rows = inventory.query("SELECT * FROM machine_to_power")
for row in rows:
(to_plug_as_str, from_plug_as_str, powercordid)=row
if to_plug_as_str != '':
conn = self._add_connection(from_plug_as_str, to_plug_as_str)
for plug in (conn.from_plug, conn.to_plug):
plug_capacity = inventory.read_plug_capacity(plug)
if plug_capacity is not None:
plug.set_current_capacity_constraint(plug_capacity)
rows = inventory.query("SELECT * FROM power_output_specs")
for row in rows:
# print row
# ('ups1_o1', 16.0),
(to_plug_as_str, max_amps_as_str)=row
to_plug = self._get_plug(to_plug_as_str)
to_plug.set_current_capacity_constraint(float(max_amps_as_str))
for machine in self.machines.values():
machine_name = machine.name
if re.match('simpatix.._..', machine_name):
machine_name = '_'.join(machine_name.split('_')[0:-1])
# print(machine_name)
machine_spec_id = inventory.machine_name_to_machine_spec_id(machine_name)
if machine_spec_id is not None:
power_consumption = inventory.machine_spec_id_to_power_consumption(machine_spec_id)
if power_consumption is not None:
machine.power_consumption = power_consumption
def get_connection_to(self, to_plug):
for connection in self.connections:
if connection.to_plug == to_plug:
return connection
return None
def _get_machine(self, machine_name):
if machine_name not in self.machines:
self.machines[machine_name] = Machine(machine_name, self)
return self.machines[machine_name]
def _get_plug(self, plug_as_str): sql_source = SqlFile(simpa_db_sql_file_path)
elements = plug_as_str.split('_') sql_reader = SqlDatabaseReader(sql_source)
plug_name = elements[-1] inventory = Inventory(sql_reader)
machine_name = plug_as_str[0:-(len(plug_name)+1)] self._parse_from_inventory(inventory)
machine = self._get_machine(machine_name)
return machine.get_plug(plug_name)
def _add_connection(self, from_plug_as_str, to_plug_as_str): def _parse_from_inventory(self, inventory):
from_plug = self._get_plug(from_plug_as_str) """
to_plug = self._get_plug(to_plug_as_str) :param Inventory inventory:
conn = Connection(from_plug, to_plug) """
self.connections.append(conn)
return conn rows = inventory.query("SELECT * FROM machine_to_power")
def __str__(self): for row in rows:
s = '' (to_plug_as_str, from_plug_as_str, powercordid) = row
for c in self.connections: if to_plug_as_str != '':
s += str(c) + '\n' conn = self._add_connection(from_plug_as_str, to_plug_as_str)
return s for plug in (conn.from_plug, conn.to_plug):
plug_capacity = inventory.read_plug_capacity(plug)
if plug_capacity is not None:
plug.set_current_capacity_constraint(plug_capacity)
rows = inventory.query("SELECT * FROM power_output_specs")
for row in rows:
# print row
# ('ups1_o1', 16.0),
(to_plug_as_str, max_amps_as_str) = row
to_plug = self._get_plug(to_plug_as_str)
to_plug.set_current_capacity_constraint(float(max_amps_as_str))
for machine in self.machines.values():
machine_name = machine.name
if re.match('simpatix.._..', machine_name):
machine_name = '_'.join(machine_name.split('_')[0:-1])
# print(machine_name)
machine_spec_id = inventory.machine_name_to_machine_spec_id(machine_name)
if machine_spec_id is not None:
power_consumption = inventory.machine_spec_id_to_power_consumption(machine_spec_id)
if power_consumption is not None:
machine.power_consumption = power_consumption
def get_connection_to(self, to_plug):
for connection in self.connections:
if connection.to_plug == to_plug:
return connection
return None
def _get_machine(self, machine_name):
if machine_name not in self.machines:
self.machines[machine_name] = Machine(machine_name, self)
return self.machines[machine_name]
def _get_plug(self, plug_as_str):
elements = plug_as_str.split('_')
plug_name = elements[-1]
machine_name = plug_as_str[0:-(len(plug_name) + 1)]
machine = self._get_machine(machine_name)
return machine.get_plug(plug_name)
def _add_connection(self, from_plug_as_str, to_plug_as_str):
from_plug = self._get_plug(from_plug_as_str)
to_plug = self._get_plug(to_plug_as_str)
conn = Connection(from_plug, to_plug)
self.connections.append(conn)
return conn
def __str__(self):
s = ''
for c in self.connections:
s += str(c) + '\n'
return s
def power_config_to_svg( power_config, svg_file_path ): def power_config_to_svg(power_config, svg_file_path):
""" """
creates a svg diagram representing the input power configuration creates a svg diagram representing the input power configuration
:param PowerConfig power_config: the input power config :param PowerConfig power_config: the input power config
""" """
graph = pygraphviz.AGraph() graph = pygraphviz.AGraph()
graph.graph_attr['overlap']='false' graph.graph_attr['overlap'] = 'false'
graph.graph_attr['splines']='true' graph.graph_attr['splines'] = 'true'
graph.edge_attr['colorscheme'] = 'rdylgn9' # 'brbg11'
graph.node_attr['shape'] = 'box'
graph.node_attr['height'] = 0.3 # default 0.5 inches
graph.node_attr['fontname'] = 'Helvetica' # default : Times-Roman
graph.edge_attr['fontsize'] = 10 # default : 14 pt
graph.edge_attr['len'] = 1.5 # default : 1.0
def hotness_to_hsv_color(hotness):
"""
:param float hotness: temperature of the wire ratio (0.0 : cold -> 1.0 : hot)
"""
clamped_hotness = max(min(hotness, 1.0), 0.0)
return "%f, 1.0, 0.8" % ((1.0 - clamped_hotness) * 0.33)
graph.edge_attr['colorscheme']='rdylgn9' # 'brbg11' # graph.add_node('a')
graph.node_attr['shape']='box' # graph.add_node('b')
graph.node_attr['height']=0.3 # default 0.5 inches # graph.add_edge(u'a',u'b',color='blue')
graph.node_attr['fontname']='Helvetica' # default : Times-Roman # graph.add_edge(u'b',u'a',color='blue')
graph.edge_attr['fontsize']=10 # default : 14 pt for con in power_config.connections:
graph.edge_attr['len']=1.5 # default : 1.0 # print(con.from_plug.machine.name, con.to_plug.machine.name)
if not con.is_redundancy_cable(): # don't display redundancy cables, as they might overlap and hide the main one
def hotness_to_hsv_color(hotness): power_consumption = con.get_power_consumption()
""" amperes = power_consumption / 220.0
:param float hotness: temperature of the wire ratio (0.0 : cold -> 1.0 : hot) color = 'green'
""" capacity = con.get_max_amperes()
clamped_hotness = max(min(hotness, 1.0), 0.0) if capacity is None:
return "%f, 1.0, 0.8" % ((1.0-clamped_hotness)*0.33) label = '?'
else:
#graph.add_node('a') label = str(capacity) + 'A'
#graph.add_node('b') if amperes / capacity > 1.0:
#graph.add_edge(u'a',u'b',color='blue') color = 'red'
#graph.add_edge(u'b',u'a',color='blue') elif amperes / capacity > 0.75:
for con in power_config.connections: color = 'orange'
# print(con.from_plug.machine.name, con.to_plug.machine.name) else:
if not con.is_redundancy_cable(): # don't display redundancy cables, as they might overlap and hide the main one color = 'green'
power_consumption = con.get_power_consumption() label = "%.1f/%s" % (amperes, label)
amperes = power_consumption/220.0 # color='//%d' % int(9.0-amperes/capacity*8)
color='green'
capacity = con.get_max_amperes() color = hotness_to_hsv_color(pow(amperes / capacity, 4.0))
if capacity == None: graph.add_edge(con.from_plug.machine.name, con.to_plug.machine.name, color=color, label=label, penwidth=capacity * 0.25)
label = '?' graph.layout(prog='twopi')
else: graph.draw(svg_file_path)
label = str(capacity) + 'A'
if amperes/capacity > 1.0:
color='red'
elif amperes/capacity > 0.75:
color='orange'
else:
color='green'
label = "%.1f/%s" % (amperes, label)
#color='//%d' % int(9.0-amperes/capacity*8)
color = hotness_to_hsv_color(pow(amperes/capacity, 4.0))
graph.add_edge(con.from_plug.machine.name, con.to_plug.machine.name, color=color, label=label, penwidth=capacity*0.25)
graph.layout(prog='twopi')
graph.draw(svg_file_path)