Compare commits
9 Commits
Author | SHA1 | Date | |
---|---|---|---|
94cac54716 | |||
c57763c016 | |||
3c6ea1ffd0 | |||
fde000ec0d | |||
f41f24f626 | |||
c621b31f6e | |||
20b586f96b | |||
08b12aeb9d | |||
00d2d0119c |
@ -3,12 +3,13 @@ datapath: "/data/SDS/" # SC3 Datapath
|
|||||||
networks: ["1Y", "HA", "MK"] # select networks, list or str
|
networks: ["1Y", "HA", "MK"] # select networks, list or str
|
||||||
stations: "*" # select stations, list or str
|
stations: "*" # select stations, list or str
|
||||||
locations: "*" # select locations, list or str
|
locations: "*" # select locations, list or str
|
||||||
stations_blacklist: ["TEST", "EREA", "DOMV"] # exclude these stations
|
stations_blacklist: ["TEST", "EREA", "DOMV", "LFKM", "GR19", "LAKA"] # exclude these stations
|
||||||
networks_blacklist: [] # exclude these networks
|
networks_blacklist: [] # exclude these networks
|
||||||
interval: 60 # Perform checks every x seconds
|
interval: 60 # Perform checks every x seconds
|
||||||
n_track: 360 # wait n_track * intervals before performing an action (i.e. send mail/end highlight status)
|
n_track: 360 # wait n_track * intervals before performing an action (i.e. send mail/end highlight status)
|
||||||
timespan: 3 # Check data of the recent x days
|
timespan: 7 # Check data of the recent x days
|
||||||
verbosity: 0 # verbosity flag
|
verbosity: 0 # verbosity flag for program console output (not logging)
|
||||||
|
logging_level: WARN # set logging level (info, warning, debug)
|
||||||
track_changes: True # tracks all changes since GUI startup by text highlighting (GUI only)
|
track_changes: True # tracks all changes since GUI startup by text highlighting (GUI only)
|
||||||
warn_count: False # show number of warnings and errors in table
|
warn_count: False # show number of warnings and errors in table
|
||||||
min_sample: 5 # minimum samples for raising Warn/FAIL
|
min_sample: 5 # minimum samples for raising Warn/FAIL
|
||||||
@ -39,6 +40,7 @@ POWBOX:
|
|||||||
THRESHOLDS:
|
THRESHOLDS:
|
||||||
pb_thresh: 0.2 # Threshold for PowBox Voltage check +/- (V)
|
pb_thresh: 0.2 # Threshold for PowBox Voltage check +/- (V)
|
||||||
max_temp: 50 # max temperature for temperature warning
|
max_temp: 50 # max temperature for temperature warning
|
||||||
|
critical_temp: 65 # max temperature for critical warning (fail)
|
||||||
low_volt: 12 # min voltage for low voltage warning
|
low_volt: 12 # min voltage for low voltage warning
|
||||||
high_volt: 14.8 # max voltage for over voltage warning
|
high_volt: 14.8 # max voltage for over voltage warning
|
||||||
unclassified: 5 # min voltage samples not classified for warning
|
unclassified: 5 # min voltage samples not classified for warning
|
||||||
@ -55,6 +57,7 @@ THRESHOLDS:
|
|||||||
# For each channel a factor 'unit' for unit conversion (e.g. to SI) can be provided, as well as a 'name'
|
# For each channel a factor 'unit' for unit conversion (e.g. to SI) can be provided, as well as a 'name'
|
||||||
# and 'ticks' [ymin, ymax, ystep] for plotting.
|
# and 'ticks' [ymin, ymax, ystep] for plotting.
|
||||||
# 'warn' and 'fail' plot horizontal lines in corresponding colors (can be str in TRESHOLDS, int/float or iterable)
|
# 'warn' and 'fail' plot horizontal lines in corresponding colors (can be str in TRESHOLDS, int/float or iterable)
|
||||||
|
# keyword "pb_SOH2" or "pb_SOH3" can be used to extract warning values from above POWBOX parameter definition
|
||||||
#
|
#
|
||||||
# 'transform' can be provided for plotting to perform arithmetic operations in given order, e.g.:
|
# 'transform' can be provided for plotting to perform arithmetic operations in given order, e.g.:
|
||||||
# transform: - ["*", 20]
|
# transform: - ["*", 20]
|
||||||
@ -73,12 +76,12 @@ CHANNELS:
|
|||||||
unit: 1e-6
|
unit: 1e-6
|
||||||
name: "PowBox 230V/12V (V)"
|
name: "PowBox 230V/12V (V)"
|
||||||
ticks: [0, 5, 1]
|
ticks: [0, 5, 1]
|
||||||
warn: [2, 3, 4, 4.5, 5]
|
warn: "pb_SOH2"
|
||||||
EX3:
|
EX3:
|
||||||
unit: 1e-6
|
unit: 1e-6
|
||||||
name: "PowBox Router/Charger (V)"
|
name: "PowBox Router/Charger (V)"
|
||||||
ticks: [0, 5, 1]
|
ticks: [0, 5, 1]
|
||||||
warn: [2, 2.5, 3, 4, 5]
|
warn: "pb_SOH3"
|
||||||
VEI:
|
VEI:
|
||||||
unit: 1e-3
|
unit: 1e-3
|
||||||
name: "Datalogger (V)"
|
name: "Datalogger (V)"
|
||||||
@ -121,6 +124,7 @@ add_links:
|
|||||||
# for example: slmon: {"URL": "path/{nw}_{st}.html", "text": "link"}
|
# for example: slmon: {"URL": "path/{nw}_{st}.html", "text": "link"}
|
||||||
slmon: {"URL": "../slmon/{nw}_{st}.html", "text": "show"}
|
slmon: {"URL": "../slmon/{nw}_{st}.html", "text": "show"}
|
||||||
24h-plot: {"URL": "../scheli/{nw}/{st}.png", "text": "plot"}
|
24h-plot: {"URL": "../scheli/{nw}/{st}.png", "text": "plot"}
|
||||||
|
ppsd: {"URL": "../ppsd/{nw}.{st}.html", "text": "show"}
|
||||||
|
|
||||||
# add station-independent links below html table (list items separated with -)
|
# add station-independent links below html table (list items separated with -)
|
||||||
add_global_links:
|
add_global_links:
|
||||||
@ -136,7 +140,7 @@ EMAIL:
|
|||||||
mailserver: "localhost"
|
mailserver: "localhost"
|
||||||
addresses: ["marcel.paffrath@rub.de", "kasper.fischer@rub.de"] # list of mail addresses for info mails
|
addresses: ["marcel.paffrath@rub.de", "kasper.fischer@rub.de"] # list of mail addresses for info mails
|
||||||
sender: "webmaster@geophysik.ruhr-uni-bochum.de" # mail sender
|
sender: "webmaster@geophysik.ruhr-uni-bochum.de" # mail sender
|
||||||
stations_blacklist: ['GR33'] # do not send emails for specific stations
|
stations_blacklist: [] # do not send emails for specific stations
|
||||||
networks_blacklist: [] # do not send emails for specific network
|
networks_blacklist: [] # do not send emails for specific network
|
||||||
# specify recipients for single stations in a yaml: key = email-address, val = station list (e.g. [1Y.GR01, 1Y.GR02])
|
# specify recipients for single stations in a yaml: key = email-address, val = station list (e.g. [1Y.GR01, 1Y.GR02])
|
||||||
external_mail_list: "mailing_list.yaml"
|
external_mail_list: "mailing_list.yaml"
|
||||||
|
204
survBot.py
204
survBot.py
@ -7,6 +7,7 @@ __author__ = 'Marcel Paffrath'
|
|||||||
import os
|
import os
|
||||||
import io
|
import io
|
||||||
import copy
|
import copy
|
||||||
|
import logging
|
||||||
import traceback
|
import traceback
|
||||||
import yaml
|
import yaml
|
||||||
import argparse
|
import argparse
|
||||||
@ -31,27 +32,48 @@ try:
|
|||||||
|
|
||||||
mail_functionality = True
|
mail_functionality = True
|
||||||
except ImportError:
|
except ImportError:
|
||||||
print('Could not import smtplib or mail. Disabled sending mails.')
|
logging.warning('Could not import smtplib or mail. Disabled sending mails.')
|
||||||
mail_functionality = False
|
mail_functionality = False
|
||||||
|
|
||||||
pjoin = os.path.join
|
pjoin = os.path.join
|
||||||
UP = "\x1B[{length}A"
|
UP = "\x1B[{length}A"
|
||||||
CLR = "\x1B[0K"
|
CLR = "\x1B[0K"
|
||||||
deg_str = '\N{DEGREE SIGN}C'
|
DEG_STR = '\N{DEGREE SIGN}C'
|
||||||
|
|
||||||
|
|
||||||
def read_yaml(file_path, n_read=3):
|
def read_yaml(file_path: str, n_read: int = 3) -> dict:
|
||||||
for index in range(n_read):
|
for index in range(n_read):
|
||||||
try:
|
try:
|
||||||
with open(file_path, "r") as f:
|
with open(file_path, "r") as f:
|
||||||
params = yaml.safe_load(f)
|
params = yaml.safe_load(f)
|
||||||
|
set_logging_level(params)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f'Could not read parameters file: {e}.\nWill try again {n_read - index - 1} time(s).')
|
logging.warning(f'Could not read parameters file: {e}.\nWill try again {n_read - index - 1} time(s).')
|
||||||
time.sleep(10)
|
time.sleep(10)
|
||||||
continue
|
continue
|
||||||
return params
|
return params
|
||||||
|
|
||||||
|
|
||||||
|
def set_logging_level(params: dict) -> None:
|
||||||
|
logging_levels = {'info': logging.INFO,
|
||||||
|
'warning': logging.WARNING,
|
||||||
|
'warn': logging.WARNING,
|
||||||
|
'debug': logging.DEBUG,
|
||||||
|
'error': logging.ERROR,
|
||||||
|
'critical': logging.CRITICAL}
|
||||||
|
logging_level_str = params.get('logging_level')
|
||||||
|
if not logging_level_str:
|
||||||
|
logging.warning('Could not set logging level. Parameter not set')
|
||||||
|
return
|
||||||
|
if not isinstance(logging_level_str, str):
|
||||||
|
logging.warning(
|
||||||
|
f'Could not set logging level. Parameter logging_level = {logging_level_str} could not be interpreted.')
|
||||||
|
return
|
||||||
|
logging.info(f'Setting logging level to {logging_level_str}')
|
||||||
|
logging_level = logging_levels.get(logging_level_str.lower())
|
||||||
|
logging.basicConfig(level=logging_level)
|
||||||
|
|
||||||
|
|
||||||
def nsl_from_id(nwst_id):
|
def nsl_from_id(nwst_id):
|
||||||
nwst_id = get_full_seed_id(nwst_id)
|
nwst_id = get_full_seed_id(nwst_id)
|
||||||
network, station, location = nwst_id.split('.')
|
network, station, location = nwst_id.split('.')
|
||||||
@ -115,7 +137,6 @@ class SurveillanceBot(object):
|
|||||||
self.parameters['channels'] = channels
|
self.parameters['channels'] = channels
|
||||||
self.reread_parameters = self.parameters.get('reread_parameters')
|
self.reread_parameters = self.parameters.get('reread_parameters')
|
||||||
self.dt_thresh = [int(val) for val in self.parameters.get('dt_thresh')]
|
self.dt_thresh = [int(val) for val in self.parameters.get('dt_thresh')]
|
||||||
self.verbosity = self.parameters.get('verbosity')
|
|
||||||
self.stations_blacklist = self.parameters.get('stations_blacklist')
|
self.stations_blacklist = self.parameters.get('stations_blacklist')
|
||||||
self.networks_blacklist = self.parameters.get('networks_blacklist')
|
self.networks_blacklist = self.parameters.get('networks_blacklist')
|
||||||
self.refresh_period = self.parameters.get('interval')
|
self.refresh_period = self.parameters.get('interval')
|
||||||
@ -196,8 +217,7 @@ class SurveillanceBot(object):
|
|||||||
for filename in self.filenames:
|
for filename in self.filenames:
|
||||||
# if file already read and last modification time is the same as of last read operation: continue
|
# if file already read and last modification time is the same as of last read operation: continue
|
||||||
if self.filenames_read_last_modif.get(filename) == os.path.getmtime(filename):
|
if self.filenames_read_last_modif.get(filename) == os.path.getmtime(filename):
|
||||||
if self.verbosity > 0:
|
logging.info(f'Continue on file {filename}')
|
||||||
print('Continue on file', filename)
|
|
||||||
continue
|
continue
|
||||||
try:
|
try:
|
||||||
# read only header of wf_data
|
# read only header of wf_data
|
||||||
@ -207,7 +227,7 @@ class SurveillanceBot(object):
|
|||||||
st_new = read(filename, dtype=float)
|
st_new = read(filename, dtype=float)
|
||||||
self.filenames_read_last_modif[filename] = os.path.getmtime(filename)
|
self.filenames_read_last_modif[filename] = os.path.getmtime(filename)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f'Could not read file {filename}:', e)
|
logging.warning(f'Could not read file {filename}: {e}')
|
||||||
continue
|
continue
|
||||||
self.dataStream += st_new
|
self.dataStream += st_new
|
||||||
self.gaps = self.dataStream.get_gaps(min_gap=self.parameters['THRESHOLDS'].get('min_gap'))
|
self.gaps = self.dataStream.get_gaps(min_gap=self.parameters['THRESHOLDS'].get('min_gap'))
|
||||||
@ -234,8 +254,7 @@ class SurveillanceBot(object):
|
|||||||
if stream:
|
if stream:
|
||||||
nsl = nsl_from_id(nwst_id)
|
nsl = nsl_from_id(nwst_id)
|
||||||
station_qc = StationQC(self, stream, nsl, self.parameters, self.keys, qc_starttime,
|
station_qc = StationQC(self, stream, nsl, self.parameters, self.keys, qc_starttime,
|
||||||
self.verbosity, print_func=self.print,
|
print_func=self.print, status_track=self.status_track.get(nwst_id))
|
||||||
status_track=self.status_track.get(nwst_id))
|
|
||||||
analysis_print_result = station_qc.return_print_analysis()
|
analysis_print_result = station_qc.return_print_analysis()
|
||||||
station_dict = station_qc.return_analysis()
|
station_dict = station_qc.return_analysis()
|
||||||
else:
|
else:
|
||||||
@ -388,13 +407,13 @@ class SurveillanceBot(object):
|
|||||||
st = modify_stream_for_plot(st, parameters=self.parameters)
|
st = modify_stream_for_plot(st, parameters=self.parameters)
|
||||||
st.plot(fig=fig, show=False, draw=False, block=False, equal_scale=False, method='full',
|
st.plot(fig=fig, show=False, draw=False, block=False, equal_scale=False, method='full',
|
||||||
starttime=starttime, endtime=endtime)
|
starttime=starttime, endtime=endtime)
|
||||||
# set_axis_ylabels(fig, self.parameters, self.verbosity)
|
# set_axis_ylabels(fig, self.parameters)
|
||||||
set_axis_yticks(fig, self.parameters, self.verbosity)
|
set_axis_yticks(fig, self.parameters)
|
||||||
set_axis_color(fig)
|
set_axis_color(fig)
|
||||||
plot_axis_thresholds(fig, self.parameters, self.verbosity)
|
plot_axis_thresholds(fig, self.parameters)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f'Could not generate plot for {nwst_id}:')
|
logging.error(f'Could not generate plot for {nwst_id}: {e}')
|
||||||
print(traceback.format_exc())
|
logging.error(traceback.format_exc())
|
||||||
if len(fig.axes) > 0:
|
if len(fig.axes) > 0:
|
||||||
ax = fig.axes[0]
|
ax = fig.axes[0]
|
||||||
ax.set_title(f'Plot refreshed at (UTC) {UTCDateTime.now().strftime("%Y-%m-%d %H:%M:%S")}. '
|
ax.set_title(f'Plot refreshed at (UTC) {UTCDateTime.now().strftime("%Y-%m-%d %H:%M:%S")}. '
|
||||||
@ -402,7 +421,10 @@ class SurveillanceBot(object):
|
|||||||
for ax in fig.axes:
|
for ax in fig.axes:
|
||||||
ax.grid(True, alpha=0.1)
|
ax.grid(True, alpha=0.1)
|
||||||
for fnout in fnames_out:
|
for fnout in fnames_out:
|
||||||
|
try:
|
||||||
fig.savefig(fnout, dpi=150., bbox_inches='tight')
|
fig.savefig(fnout, dpi=150., bbox_inches='tight')
|
||||||
|
except IOError as e:
|
||||||
|
logging.warning('Could not save figure with IO error. Disk quota exceeded?\nError message: {e}')
|
||||||
# if needed save figure as virtual object (e.g. for mailing)
|
# if needed save figure as virtual object (e.g. for mailing)
|
||||||
if save_bytes:
|
if save_bytes:
|
||||||
fnames_out[-1].seek(0)
|
fnames_out[-1].seek(0)
|
||||||
@ -458,7 +480,7 @@ class SurveillanceBot(object):
|
|||||||
# add degree sign for temp
|
# add degree sign for temp
|
||||||
if check_key == 'temp':
|
if check_key == 'temp':
|
||||||
if not type(message) in [str]:
|
if not type(message) in [str]:
|
||||||
message = str(message) + deg_str
|
message = str(message) + DEG_STR
|
||||||
|
|
||||||
html_class = self.get_html_class(hide_keys_mobile, status=status, check_key=check_key)
|
html_class = self.get_html_class(hide_keys_mobile, status=status, check_key=check_key)
|
||||||
item = dict(text=str(message), tooltip=str(detailed_message), color=bg_color,
|
item = dict(text=str(message), tooltip=str(detailed_message), color=bg_color,
|
||||||
@ -515,17 +537,16 @@ class SurveillanceBot(object):
|
|||||||
# write footer with optional logo
|
# write footer with optional logo
|
||||||
logo_file = self.parameters.get('html_logo')
|
logo_file = self.parameters.get('html_logo')
|
||||||
if not os.path.isfile(pjoin(self.outpath_html, logo_file)):
|
if not os.path.isfile(pjoin(self.outpath_html, logo_file)):
|
||||||
print(f'Specified file {logo_file} not found.')
|
logging.info(f'Specified file {logo_file} not found.')
|
||||||
logo_file = None
|
logo_file = None
|
||||||
|
|
||||||
outfile.write(html_footer(footer_logo=logo_file))
|
outfile.write(html_footer(footer_logo=logo_file))
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f'Could not write HTML table to {fnout}:')
|
logging.info(f'Could not write HTML table to {fnout}:')
|
||||||
print(traceback.format_exc())
|
logging.debug(traceback.format_exc())
|
||||||
|
|
||||||
if self.verbosity:
|
logging.info(f'Wrote html table to {fnout}')
|
||||||
print(f'Wrote html table to {fnout}')
|
|
||||||
|
|
||||||
def update_status_message(self):
|
def update_status_message(self):
|
||||||
timespan = timedelta(seconds=int(self.parameters.get('timespan') * 24 * 3600))
|
timespan = timedelta(seconds=int(self.parameters.get('timespan') * 24 * 3600))
|
||||||
@ -540,7 +561,6 @@ class SurveillanceBot(object):
|
|||||||
string.replace('\n', clear_end)
|
string.replace('\n', clear_end)
|
||||||
print(string, end=clear_end, **kwargs)
|
print(string, end=clear_end, **kwargs)
|
||||||
self.print_count += n_nl + 1 # number of newlines + actual print with end='\n' (no check for kwargs end!)
|
self.print_count += n_nl + 1 # number of newlines + actual print with end='\n' (no check for kwargs end!)
|
||||||
# print('pc:', self.print_count)
|
|
||||||
|
|
||||||
def clear_prints(self):
|
def clear_prints(self):
|
||||||
print(UP.format(length=self.print_count), end='')
|
print(UP.format(length=self.print_count), end='')
|
||||||
@ -548,14 +568,12 @@ class SurveillanceBot(object):
|
|||||||
|
|
||||||
|
|
||||||
class StationQC(object):
|
class StationQC(object):
|
||||||
def __init__(self, parent, stream, nsl, parameters, keys, starttime, verbosity, print_func, status_track=None):
|
def __init__(self, parent, stream, nsl, parameters, keys, starttime, print_func, status_track=None):
|
||||||
"""
|
"""
|
||||||
Station Quality Check class.
|
Station Quality Check class.
|
||||||
:param nsl: dictionary containing network, station and location (key: str)
|
:param nsl: dictionary containing network, station and location (key: str)
|
||||||
:param parameters: parameters dictionary from parameters.yaml file
|
:param parameters: parameters dictionary from parameters.yaml file
|
||||||
"""
|
"""
|
||||||
if status_track is None:
|
|
||||||
status_track = {}
|
|
||||||
self.parent = parent
|
self.parent = parent
|
||||||
self.stream = stream
|
self.stream = stream
|
||||||
self.nsl = nsl
|
self.nsl = nsl
|
||||||
@ -565,7 +583,6 @@ class StationQC(object):
|
|||||||
# make a copy of parameters object to prevent accidental changes
|
# make a copy of parameters object to prevent accidental changes
|
||||||
self.parameters = copy.deepcopy(parameters)
|
self.parameters = copy.deepcopy(parameters)
|
||||||
self.program_starttime = starttime
|
self.program_starttime = starttime
|
||||||
self.verbosity = verbosity
|
|
||||||
self.last_active = False
|
self.last_active = False
|
||||||
self.print = print_func
|
self.print = print_func
|
||||||
|
|
||||||
@ -576,6 +593,8 @@ class StationQC(object):
|
|||||||
status_track = {}
|
status_track = {}
|
||||||
self.status_track = status_track
|
self.status_track = status_track
|
||||||
|
|
||||||
|
self.powbox_active = self.is_pbox_activated_check()
|
||||||
|
|
||||||
self.start()
|
self.start()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
@ -598,8 +617,7 @@ class StationQC(object):
|
|||||||
current_status = self.status_dict.get(key)
|
current_status = self.status_dict.get(key)
|
||||||
|
|
||||||
# change this to something more useful, SMS/EMAIL/PUSH
|
# change this to something more useful, SMS/EMAIL/PUSH
|
||||||
if self.verbosity:
|
logging.info(f'{UTCDateTime()}: {detailed_message}')
|
||||||
self.print(f'{UTCDateTime()}: {detailed_message}', flush=False)
|
|
||||||
|
|
||||||
# if error, do not overwrite with warning
|
# if error, do not overwrite with warning
|
||||||
if current_status.is_error:
|
if current_status.is_error:
|
||||||
@ -628,7 +646,8 @@ class StationQC(object):
|
|||||||
send_mail = False
|
send_mail = False
|
||||||
new_error = StatusError(count=count, show_count=self.parameters.get('warn_count'))
|
new_error = StatusError(count=count, show_count=self.parameters.get('warn_count'))
|
||||||
if disc:
|
if disc:
|
||||||
new_error.set_disconnected()
|
msg = disc if type(disc) == str else None
|
||||||
|
new_error.set_disconnected(msg)
|
||||||
current_status = self.status_dict.get(key)
|
current_status = self.status_dict.get(key)
|
||||||
if current_status.is_error:
|
if current_status.is_error:
|
||||||
current_status.count += count
|
current_status.count += count
|
||||||
@ -638,8 +657,7 @@ class StationQC(object):
|
|||||||
if self.status_track.get(key) and not self.status_track.get(key)[-1]:
|
if self.status_track.get(key) and not self.status_track.get(key)[-1]:
|
||||||
self.parent.write_html_figure(self.nwst_id, save_bytes=True)
|
self.parent.write_html_figure(self.nwst_id, save_bytes=True)
|
||||||
|
|
||||||
if self.verbosity:
|
logging.info(f'{UTCDateTime()}: {detailed_message}')
|
||||||
self.print(f'{UTCDateTime()}: {detailed_message}', flush=False)
|
|
||||||
|
|
||||||
# do not send error mail if this is the first run (e.g. program startup) or state was already error (unchanged)
|
# do not send error mail if this is the first run (e.g. program startup) or state was already error (unchanged)
|
||||||
if self.search_previous_errors(key) is True:
|
if self.search_previous_errors(key) is True:
|
||||||
@ -671,7 +689,7 @@ class StationQC(object):
|
|||||||
|
|
||||||
# simulate an error specified in json file (dictionary: {nwst_id: key} )
|
# simulate an error specified in json file (dictionary: {nwst_id: key} )
|
||||||
if self._simulated_error_check(key) is True:
|
if self._simulated_error_check(key) is True:
|
||||||
print(f'Simulating Error on {self.nwst_id}, {key}')
|
logging.info(f'Simulating Error on {self.nwst_id}, {key}')
|
||||||
return True
|
return True
|
||||||
|
|
||||||
previous_errors = self.status_track.get(key)
|
previous_errors = self.status_track.get(key)
|
||||||
@ -696,26 +714,22 @@ class StationQC(object):
|
|||||||
def send_mail(self, key, status_type, additional_message=''):
|
def send_mail(self, key, status_type, additional_message=''):
|
||||||
""" Send info mail using parameters specified in parameters file """
|
""" Send info mail using parameters specified in parameters file """
|
||||||
if not mail_functionality:
|
if not mail_functionality:
|
||||||
if self.verbosity:
|
logging.info('Mail functionality disabled. Return')
|
||||||
print('Mail functionality disabled. Return')
|
|
||||||
return
|
return
|
||||||
|
|
||||||
mail_params = self.parameters.get('EMAIL')
|
mail_params = self.parameters.get('EMAIL')
|
||||||
if not mail_params:
|
if not mail_params:
|
||||||
if self.verbosity:
|
logging.info('parameter "EMAIL" not set in parameter file. Return')
|
||||||
print('parameter "EMAIL" not set in parameter file. Return')
|
|
||||||
return
|
return
|
||||||
|
|
||||||
stations_blacklist = mail_params.get('stations_blacklist')
|
stations_blacklist = mail_params.get('stations_blacklist')
|
||||||
if stations_blacklist and self.station in stations_blacklist:
|
if stations_blacklist and self.station in stations_blacklist:
|
||||||
if self.verbosity:
|
logging.info(f'Station {self.station} listed in blacklist. Return')
|
||||||
print(f'Station {self.station} listed in blacklist. Return')
|
|
||||||
return
|
return
|
||||||
|
|
||||||
networks_blacklist = mail_params.get('networks_blacklist')
|
networks_blacklist = mail_params.get('networks_blacklist')
|
||||||
if networks_blacklist and self.network in networks_blacklist:
|
if networks_blacklist and self.network in networks_blacklist:
|
||||||
if self.verbosity:
|
logging.info(f'Station {self.station} of network {self.network} listed in blacklist. Return')
|
||||||
print(f'Station {self.station} of network {self.network} listed in blacklist. Return')
|
|
||||||
return
|
return
|
||||||
|
|
||||||
sender = mail_params.get('sender')
|
sender = mail_params.get('sender')
|
||||||
@ -726,8 +740,7 @@ class StationQC(object):
|
|||||||
addresses = addresses[:] + list(add_addresses)
|
addresses = addresses[:] + list(add_addresses)
|
||||||
server = mail_params.get('mailserver')
|
server = mail_params.get('mailserver')
|
||||||
if not sender or not addresses:
|
if not sender or not addresses:
|
||||||
if self.verbosity:
|
logging.info('Mail sender or addresses not (correctly) defined. Return')
|
||||||
print('Mail sender or addresses not (correctly) defined. Return')
|
|
||||||
return
|
return
|
||||||
dt = self.get_dt_for_action()
|
dt = self.get_dt_for_action()
|
||||||
text = f'{key}: Status {status_type} longer than {dt}: ' + additional_message
|
text = f'{key}: Status {status_type} longer than {dt}: ' + additional_message
|
||||||
@ -790,20 +803,16 @@ class StationQC(object):
|
|||||||
yield address
|
yield address
|
||||||
# file not existing
|
# file not existing
|
||||||
except FileNotFoundError as e:
|
except FileNotFoundError as e:
|
||||||
if self.verbosity:
|
logging.warning(e)
|
||||||
print(e)
|
|
||||||
# no dictionary
|
# no dictionary
|
||||||
except AttributeError as e:
|
except AttributeError as e:
|
||||||
if self.verbosity:
|
logging.warning(f'Could not read dictionary from file {eml_filename}: {e}')
|
||||||
print(f'Could not read dictionary from file {eml_filename}: {e}')
|
|
||||||
# other exceptions
|
# other exceptions
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if self.verbosity:
|
logging.warning(f'Could not open file {eml_filename}: {e}')
|
||||||
print(f'Could not open file {eml_filename}: {e}')
|
|
||||||
# no file specified
|
# no file specified
|
||||||
else:
|
else:
|
||||||
if self.verbosity:
|
logging.info('No external mail list set.')
|
||||||
print('No external mail list set.')
|
|
||||||
|
|
||||||
return []
|
return []
|
||||||
|
|
||||||
@ -877,9 +886,8 @@ class StationQC(object):
|
|||||||
timespan = self.parameters.get('timespan') * 24 * 3600
|
timespan = self.parameters.get('timespan') * 24 * 3600
|
||||||
self.analysis_starttime = self.program_starttime - timespan
|
self.analysis_starttime = self.program_starttime - timespan
|
||||||
|
|
||||||
if self.verbosity > 0:
|
logging.info(150 * '#')
|
||||||
self.print(150 * '#')
|
logging.info('This is StationQC. Calculating quality for station'
|
||||||
self.print('This is StationQT. Calculating quality for station'
|
|
||||||
' {network}.{station}.{location}'.format(**self.nsl))
|
' {network}.{station}.{location}'.format(**self.nsl))
|
||||||
self.voltage_analysis()
|
self.voltage_analysis()
|
||||||
self.pb_temp_analysis()
|
self.pb_temp_analysis()
|
||||||
@ -907,7 +915,7 @@ class StationQC(object):
|
|||||||
if key == 'last active':
|
if key == 'last active':
|
||||||
items.append(fancy_timestr(message))
|
items.append(fancy_timestr(message))
|
||||||
elif key == 'temp':
|
elif key == 'temp':
|
||||||
items.append(str(message) + deg_str)
|
items.append(str(message) + DEG_STR)
|
||||||
else:
|
else:
|
||||||
items.append(str(message))
|
items.append(str(message))
|
||||||
return items
|
return items
|
||||||
@ -946,9 +954,8 @@ class StationQC(object):
|
|||||||
clock_quality_warn_level = self.parameters.get('THRESHOLDS').get('clockquality_warn')
|
clock_quality_warn_level = self.parameters.get('THRESHOLDS').get('clockquality_warn')
|
||||||
clock_quality_fail_level = self.parameters.get('THRESHOLDS').get('clockquality_fail')
|
clock_quality_fail_level = self.parameters.get('THRESHOLDS').get('clockquality_fail')
|
||||||
|
|
||||||
if self.verbosity > 1:
|
logging.info(40 * '-')
|
||||||
self.print(40 * '-')
|
logging.info('Performing Clock Quality check')
|
||||||
self.print('Performing Clock Quality check', flush=False)
|
|
||||||
|
|
||||||
clockQuality_warn = np.where(clock_quality < clock_quality_warn_level)[0]
|
clockQuality_warn = np.where(clock_quality < clock_quality_warn_level)[0]
|
||||||
clockQuality_fail = np.where(clock_quality < clock_quality_fail_level)[0]
|
clockQuality_fail = np.where(clock_quality < clock_quality_fail_level)[0]
|
||||||
@ -992,9 +999,8 @@ class StationQC(object):
|
|||||||
low_volt = self.parameters.get('THRESHOLDS').get('low_volt')
|
low_volt = self.parameters.get('THRESHOLDS').get('low_volt')
|
||||||
high_volt = self.parameters.get('THRESHOLDS').get('high_volt')
|
high_volt = self.parameters.get('THRESHOLDS').get('high_volt')
|
||||||
|
|
||||||
if self.verbosity > 1:
|
logging.info(40 * '-')
|
||||||
self.print(40 * '-')
|
logging.info('Performing Voltage check')
|
||||||
self.print('Performing Voltage check', flush=False)
|
|
||||||
|
|
||||||
overvolt = np.where(voltage > high_volt)[0]
|
overvolt = np.where(voltage > high_volt)[0]
|
||||||
undervolt = np.where(voltage < low_volt)[0]
|
undervolt = np.where(voltage < low_volt)[0]
|
||||||
@ -1020,9 +1026,12 @@ class StationQC(object):
|
|||||||
self.warn(key, detailed_message=detailed_message, count=n_undervolt,
|
self.warn(key, detailed_message=detailed_message, count=n_undervolt,
|
||||||
last_occurrence=self.get_last_occurrence(trace, undervolt))
|
last_occurrence=self.get_last_occurrence(trace, undervolt))
|
||||||
|
|
||||||
def pb_temp_analysis(self, channel='EX1'):
|
def pb_temp_analysis(self, channel='EX1', t_max_default=50, t_crit_default=70):
|
||||||
""" Analyse PowBox temperature output. """
|
""" Analyse PowBox temperature output. """
|
||||||
key = 'temp'
|
key = 'temp'
|
||||||
|
if not self.powbox_active:
|
||||||
|
self.set_pbox_inactive_error(key)
|
||||||
|
return
|
||||||
st = self.stream.select(channel=channel)
|
st = self.stream.select(channel=channel)
|
||||||
trace = self.get_trace(st, key)
|
trace = self.get_trace(st, key)
|
||||||
if not trace:
|
if not trace:
|
||||||
@ -1033,23 +1042,31 @@ class StationQC(object):
|
|||||||
# average temp
|
# average temp
|
||||||
timespan = min([self.parameters.get('timespan') * 24 * 3600, int(len(temp) / trace.stats.sampling_rate)])
|
timespan = min([self.parameters.get('timespan') * 24 * 3600, int(len(temp) / trace.stats.sampling_rate)])
|
||||||
nsamp_av = int(trace.stats.sampling_rate) * timespan
|
nsamp_av = int(trace.stats.sampling_rate) * timespan
|
||||||
av_temp_str = str(round(np.nanmean(temp[-nsamp_av:]), 1)) + deg_str
|
av_temp_str = str(round(np.nanmean(temp[-nsamp_av:]), 1)) + DEG_STR
|
||||||
# dt of average
|
# dt of average
|
||||||
dt_t_str = str(timedelta(seconds=int(timespan))).replace(', 0:00:00', '')
|
dt_t_str = str(timedelta(seconds=int(timespan))).replace(', 0:00:00', '')
|
||||||
# current temp
|
# current temp
|
||||||
cur_temp = round(temp[-1], 1)
|
cur_temp = round(temp[-1], 1)
|
||||||
if self.verbosity > 1:
|
logging.info(40 * '-')
|
||||||
self.print(40 * '-')
|
logging.info('Performing PowBox temperature check (EX1)')
|
||||||
self.print('Performing PowBox temperature check (EX1)', flush=False)
|
logging.info(f'Average temperature at {np.nanmean(temp)}\N{DEGREE SIGN}')
|
||||||
self.print(f'Average temperature at {np.nanmean(temp)}\N{DEGREE SIGN}', flush=False)
|
logging.info(f'Peak temperature at {max(temp)}\N{DEGREE SIGN}')
|
||||||
self.print(f'Peak temperature at {max(temp)}\N{DEGREE SIGN}', flush=False)
|
logging.info(f'Min temperature at {min(temp)}\N{DEGREE SIGN}')
|
||||||
self.print(f'Min temperature at {min(temp)}\N{DEGREE SIGN}', flush=False)
|
max_temp = thresholds.get('max_temp', t_max_default)
|
||||||
max_temp = thresholds.get('max_temp')
|
max_temp_crit = thresholds.get('critical_temp', t_crit_default)
|
||||||
t_check = np.where(temp > max_temp)[0]
|
t_check = np.where(temp > max_temp)[0]
|
||||||
if len(t_check) > 0:
|
t_check_crit = np.where(temp > max_temp_crit)[0]
|
||||||
|
tcheck_message_template = ('Trace {id}: Temperature over {tmax}' + f'\N{DEGREE SIGN}'
|
||||||
|
+ '! Current temperature: {temp}' + f'\N{DEGREE SIGN}')
|
||||||
|
if len(t_check_crit) > 0:
|
||||||
|
self.error(key=key,
|
||||||
|
detailed_message=tcheck_message_template.format(id=trace.get_id(), tmax=max_temp, temp=cur_temp)
|
||||||
|
+ self.get_last_occurrence_timestring(trace, t_check_crit),
|
||||||
|
last_occurrence=self.get_last_occurrence(trace, t_check_crit))
|
||||||
|
elif len(t_check) > 0:
|
||||||
self.warn(key=key,
|
self.warn(key=key,
|
||||||
detailed_message=f'Trace {trace.get_id()}: '
|
detailed_message=tcheck_message_template.format(id=trace.get_id(), tmax=max_temp_crit,
|
||||||
f'Temperature over {max_temp}\N{DEGREE SIGN} at {trace.get_id()}!'
|
temp=cur_temp)
|
||||||
+ self.get_last_occurrence_timestring(trace, t_check),
|
+ self.get_last_occurrence_timestring(trace, t_check),
|
||||||
last_occurrence=self.get_last_occurrence(trace, t_check))
|
last_occurrence=self.get_last_occurrence(trace, t_check))
|
||||||
else:
|
else:
|
||||||
@ -1098,23 +1115,26 @@ class StationQC(object):
|
|||||||
self.error(key=key,
|
self.error(key=key,
|
||||||
detailed_message=f'Fail status for mass centering. Highest val (abs) {common_highest_val}V',)
|
detailed_message=f'Fail status for mass centering. Highest val (abs) {common_highest_val}V',)
|
||||||
|
|
||||||
if self.verbosity > 1:
|
logging.info(40 * '-')
|
||||||
self.print(40 * '-')
|
logging.info('Performing mass position check')
|
||||||
self.print('Performing mass position check', flush=False)
|
logging.info(f'Average mass position at {common_highest_val}')
|
||||||
self.print(f'Average mass position at {common_highest_val}', flush=False)
|
|
||||||
|
|
||||||
def pb_power_analysis(self, channel='EX2', pb_dict_key='pb_SOH2'):
|
def pb_power_analysis(self, channel='EX2', pb_dict_key='pb_SOH2'):
|
||||||
""" Analyse EX2 channel of PowBox """
|
""" Analyse EX2 channel of PowBox """
|
||||||
keys = ['230V', '12V']
|
keys = ['230V', '12V']
|
||||||
|
if not self.powbox_active:
|
||||||
|
for key in keys:
|
||||||
|
self.set_pbox_inactive_error(key)
|
||||||
|
return
|
||||||
|
|
||||||
st = self.stream.select(channel=channel)
|
st = self.stream.select(channel=channel)
|
||||||
trace = self.get_trace(st, keys)
|
trace = self.get_trace(st, keys)
|
||||||
if not trace:
|
if not trace:
|
||||||
return
|
return
|
||||||
|
|
||||||
voltage = trace.data * self.get_unit_factor(channel)
|
voltage = trace.data * self.get_unit_factor(channel)
|
||||||
if self.verbosity > 1:
|
logging.info(40 * '-')
|
||||||
self.print(40 * '-')
|
logging.info('Performing PowBox 12V/230V check (EX2)')
|
||||||
self.print('Performing PowBox 12V/230V check (EX2)', flush=False)
|
|
||||||
voltage_check, voltage_dict, last_val = self.pb_voltage_ok(trace, voltage, pb_dict_key, channel=channel)
|
voltage_check, voltage_dict, last_val = self.pb_voltage_ok(trace, voltage, pb_dict_key, channel=channel)
|
||||||
|
|
||||||
if voltage_check:
|
if voltage_check:
|
||||||
@ -1128,16 +1148,19 @@ class StationQC(object):
|
|||||||
def pb_rout_charge_analysis(self, channel='EX3', pb_dict_key='pb_SOH3'):
|
def pb_rout_charge_analysis(self, channel='EX3', pb_dict_key='pb_SOH3'):
|
||||||
""" Analyse EX3 channel of PowBox """
|
""" Analyse EX3 channel of PowBox """
|
||||||
keys = ['router', 'charger']
|
keys = ['router', 'charger']
|
||||||
pb_thresh = self.parameters.get('THRESHOLDS').get('pb_1v')
|
if not self.powbox_active:
|
||||||
|
for key in keys:
|
||||||
|
self.set_pbox_inactive_error(key)
|
||||||
|
return
|
||||||
|
|
||||||
st = self.stream.select(channel=channel)
|
st = self.stream.select(channel=channel)
|
||||||
trace = self.get_trace(st, keys)
|
trace = self.get_trace(st, keys)
|
||||||
if not trace:
|
if not trace:
|
||||||
return
|
return
|
||||||
|
|
||||||
voltage = trace.data * self.get_unit_factor(channel)
|
voltage = trace.data * self.get_unit_factor(channel)
|
||||||
if self.verbosity > 1:
|
logging.info(40 * '-')
|
||||||
self.print(40 * '-')
|
logging.info('Performing PowBox Router/Charger check (EX3)')
|
||||||
self.print('Performing PowBox Router/Charger check (EX3)', flush=False)
|
|
||||||
voltage_check, voltage_dict, last_val = self.pb_voltage_ok(trace, voltage, pb_dict_key, channel=channel)
|
voltage_check, voltage_dict, last_val = self.pb_voltage_ok(trace, voltage, pb_dict_key, channel=channel)
|
||||||
|
|
||||||
if voltage_check:
|
if voltage_check:
|
||||||
@ -1243,6 +1266,10 @@ class StationQC(object):
|
|||||||
with each voltage value associated to the different steps specified in POWBOX > pb_steps. Also raises
|
with each voltage value associated to the different steps specified in POWBOX > pb_steps. Also raises
|
||||||
self.warn in case there are unassociated voltage values recorded.
|
self.warn in case there are unassociated voltage values recorded.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
if not self.powbox_active:
|
||||||
|
return
|
||||||
|
|
||||||
pb_thresh = self.parameters.get('THRESHOLDS').get('pb_thresh')
|
pb_thresh = self.parameters.get('THRESHOLDS').get('pb_thresh')
|
||||||
pb_ok = self.parameters.get('POWBOX').get('pb_ok')
|
pb_ok = self.parameters.get('POWBOX').get('pb_ok')
|
||||||
# possible voltage levels are keys of pb voltage level dict
|
# possible voltage levels are keys of pb voltage level dict
|
||||||
@ -1305,6 +1332,13 @@ class StationQC(object):
|
|||||||
""" get UTCDateTime from trace and index"""
|
""" get UTCDateTime from trace and index"""
|
||||||
return trace.stats.starttime + trace.stats.delta * index
|
return trace.stats.starttime + trace.stats.delta * index
|
||||||
|
|
||||||
|
def is_pbox_activated_check(self):
|
||||||
|
return self.station not in self.parameters.get('no_pbox_stations', [])
|
||||||
|
|
||||||
|
def set_pbox_inactive_error(self, key):
|
||||||
|
msg = self.parameters.get('no_pbox_stations')[self.station]
|
||||||
|
self.error(key, detailed_message=f'PowBox not connected', disc=msg)
|
||||||
|
|
||||||
|
|
||||||
class Status(object):
|
class Status(object):
|
||||||
""" Basic Status class. All status classes are derived from this class."""
|
""" Basic Status class. All status classes are derived from this class."""
|
||||||
@ -1372,8 +1406,10 @@ class StatusError(Status):
|
|||||||
self.set_error()
|
self.set_error()
|
||||||
self.default_message = message
|
self.default_message = message
|
||||||
|
|
||||||
def set_disconnected(self, message='DCN'):
|
def set_disconnected(self, message=None):
|
||||||
self.connection_error = True
|
self.connection_error = True
|
||||||
|
if not message:
|
||||||
|
message = 'DCN'
|
||||||
self.message = message
|
self.message = message
|
||||||
|
|
||||||
def set_connected(self):
|
def set_connected(self):
|
||||||
|
@ -11,6 +11,8 @@ import os
|
|||||||
import sys
|
import sys
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from PySide2 import QtGui, QtCore, QtWidgets
|
from PySide2 import QtGui, QtCore, QtWidgets
|
||||||
except ImportError:
|
except ImportError:
|
||||||
@ -41,7 +43,7 @@ try:
|
|||||||
from rest_api.rest_api_utils import get_last_messages, send_message, get_default_params
|
from rest_api.rest_api_utils import get_last_messages, send_message, get_default_params
|
||||||
sms_funcs = True
|
sms_funcs = True
|
||||||
except ImportError:
|
except ImportError:
|
||||||
print('Could not load rest_api utils, SMS functionality disabled.')
|
logging.warning('Could not load rest_api utils, SMS functionality disabled.')
|
||||||
sms_funcs = False
|
sms_funcs = False
|
||||||
|
|
||||||
deg_str = '\N{DEGREE SIGN}C'
|
deg_str = '\N{DEGREE SIGN}C'
|
||||||
@ -54,10 +56,9 @@ class Thread(QtCore.QThread):
|
|||||||
"""
|
"""
|
||||||
update = QtCore.Signal()
|
update = QtCore.Signal()
|
||||||
|
|
||||||
def __init__(self, parent, runnable, verbosity=0):
|
def __init__(self, parent, runnable):
|
||||||
super(Thread, self).__init__(parent=parent)
|
super(Thread, self).__init__(parent=parent)
|
||||||
self.setParent(parent)
|
self.setParent(parent)
|
||||||
self.verbosity = verbosity
|
|
||||||
self.runnable = runnable
|
self.runnable = runnable
|
||||||
self.is_active = True
|
self.is_active = True
|
||||||
|
|
||||||
@ -69,11 +70,10 @@ class Thread(QtCore.QThread):
|
|||||||
self.update.emit()
|
self.update.emit()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.is_active = False
|
self.is_active = False
|
||||||
print(e)
|
logging.error(e)
|
||||||
print(traceback.format_exc())
|
logging.debug(traceback.format_exc())
|
||||||
finally:
|
finally:
|
||||||
if self.verbosity > 0:
|
logging.info(f'Time for Thread execution: {UTCDateTime() - t0}')
|
||||||
print(f'Time for Thread execution: {UTCDateTime() - t0}')
|
|
||||||
|
|
||||||
|
|
||||||
class MainWindow(QtWidgets.QMainWindow):
|
class MainWindow(QtWidgets.QMainWindow):
|
||||||
@ -195,7 +195,7 @@ class MainWindow(QtWidgets.QMainWindow):
|
|||||||
station = nwst_id.split('.')[1]
|
station = nwst_id.split('.')[1]
|
||||||
iccid = get_station_iccid(station)
|
iccid = get_station_iccid(station)
|
||||||
if not iccid:
|
if not iccid:
|
||||||
print('Could not find iccid for station', nwst_id)
|
logging.info(f'Could not find iccid for station: {nwst_id}')
|
||||||
return
|
return
|
||||||
sms_widget = ReadSMSWidget(parent=self, iccid=iccid)
|
sms_widget = ReadSMSWidget(parent=self, iccid=iccid)
|
||||||
sms_widget.setWindowTitle(f'Recent SMS of station: {nwst_id}')
|
sms_widget.setWindowTitle(f'Recent SMS of station: {nwst_id}')
|
||||||
|
73
utils.py
73
utils.py
@ -1,12 +1,22 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
import matplotlib
|
import matplotlib
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
from obspy import Stream
|
from obspy import Stream
|
||||||
|
|
||||||
|
|
||||||
|
COLORS_DICT = {'FAIL': (195, 29, 14, 255),
|
||||||
|
'NO DATA': (255, 255, 125, 255),
|
||||||
|
'WARN': (250, 192, 63, 255),
|
||||||
|
'OK': (185, 245, 145, 255),
|
||||||
|
'undefined': (240, 240, 240, 255),
|
||||||
|
'disc': (126, 127, 131, 255), }
|
||||||
|
|
||||||
|
|
||||||
def get_bg_color(check_key, status, dt_thresh=None, hex=False):
|
def get_bg_color(check_key, status, dt_thresh=None, hex=False):
|
||||||
message = status.message
|
message = status.message
|
||||||
if check_key == 'last active':
|
if check_key == 'last active':
|
||||||
@ -41,13 +51,9 @@ def get_color(key):
|
|||||||
# 'OK': (173, 255, 133, 255),
|
# 'OK': (173, 255, 133, 255),
|
||||||
# 'undefined': (230, 230, 230, 255),
|
# 'undefined': (230, 230, 230, 255),
|
||||||
# 'disc': (255, 160, 40, 255),}
|
# 'disc': (255, 160, 40, 255),}
|
||||||
colors_dict = {'FAIL': (195, 29, 14, 255),
|
if not key in COLORS_DICT.keys():
|
||||||
'NO DATA': (255, 255, 125, 255),
|
key = 'undefined'
|
||||||
'WARN': (250, 192, 63, 255),
|
return COLORS_DICT.get(key)
|
||||||
'OK': (185, 245, 145, 255),
|
|
||||||
'undefined': (240, 240, 240, 255),
|
|
||||||
'disc': (126, 127, 131, 255), }
|
|
||||||
return colors_dict.get(key)
|
|
||||||
|
|
||||||
|
|
||||||
def get_color_mpl(key):
|
def get_color_mpl(key):
|
||||||
@ -82,6 +88,8 @@ def get_mass_color(message):
|
|||||||
def get_temp_color(temp, vmin=-10, vmax=60, cmap='coolwarm'):
|
def get_temp_color(temp, vmin=-10, vmax=60, cmap='coolwarm'):
|
||||||
""" Get an rgba temperature value back from specified cmap, linearly interpolated between vmin and vmax. """
|
""" Get an rgba temperature value back from specified cmap, linearly interpolated between vmin and vmax. """
|
||||||
if type(temp) in [str]:
|
if type(temp) in [str]:
|
||||||
|
if temp in COLORS_DICT.keys():
|
||||||
|
return get_color(temp)
|
||||||
return get_color('undefined')
|
return get_color('undefined')
|
||||||
cmap = matplotlib.cm.get_cmap(cmap)
|
cmap = matplotlib.cm.get_cmap(cmap)
|
||||||
val = (temp - vmin) / (vmax - vmin)
|
val = (temp - vmin) / (vmax - vmin)
|
||||||
@ -170,7 +178,7 @@ def transform_trace(data, transf):
|
|||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
def set_axis_ylabels(fig, parameters, verbosity=0):
|
def set_axis_ylabels(fig, parameters):
|
||||||
"""
|
"""
|
||||||
Adds channel names to y-axis if defined in parameters.
|
Adds channel names to y-axis if defined in parameters.
|
||||||
"""
|
"""
|
||||||
@ -178,24 +186,25 @@ def set_axis_ylabels(fig, parameters, verbosity=0):
|
|||||||
if not names: # or not len(st.traces):
|
if not names: # or not len(st.traces):
|
||||||
return
|
return
|
||||||
if not len(names) == len(fig.axes):
|
if not len(names) == len(fig.axes):
|
||||||
if verbosity:
|
logging.info('Mismatch in axis and label lengths. Not adding plot labels')
|
||||||
print('Mismatch in axis and label lengths. Not adding plot labels')
|
|
||||||
return
|
return
|
||||||
for channel_name, ax in zip(names, fig.axes):
|
for channel_name, ax in zip(names, fig.axes):
|
||||||
if channel_name:
|
if channel_name:
|
||||||
ax.set_ylabel(channel_name)
|
ax.set_ylabel(channel_name)
|
||||||
|
|
||||||
|
|
||||||
def set_axis_color(fig, color='0.8'):
|
def set_axis_color(fig, color='0.8', shade_color='0.95'):
|
||||||
"""
|
"""
|
||||||
Set all axes of figure to specific color
|
Set all axes (frame) of figure to specific color. Shade every second axis.
|
||||||
"""
|
"""
|
||||||
for ax in fig.axes:
|
for i, ax in enumerate(fig.axes):
|
||||||
for key in ['bottom', 'top', 'right', 'left']:
|
for key in ['bottom', 'top', 'right', 'left']:
|
||||||
ax.spines[key].set_color(color)
|
ax.spines[key].set_color(color)
|
||||||
|
if i % 2:
|
||||||
|
ax.set_facecolor(shade_color)
|
||||||
|
|
||||||
|
|
||||||
def set_axis_yticks(fig, parameters, verbosity=0):
|
def set_axis_yticks(fig, parameters):
|
||||||
"""
|
"""
|
||||||
Adds channel names to y-axis if defined in parameters.
|
Adds channel names to y-axis if defined in parameters.
|
||||||
"""
|
"""
|
||||||
@ -203,8 +212,7 @@ def set_axis_yticks(fig, parameters, verbosity=0):
|
|||||||
if not ticks:
|
if not ticks:
|
||||||
return
|
return
|
||||||
if not len(ticks) == len(fig.axes):
|
if not len(ticks) == len(fig.axes):
|
||||||
if verbosity:
|
logging.info('Mismatch in axis tick and label lengths. Not changing plot ticks.')
|
||||||
print('Mismatch in axis tick and label lengths. Not changing plot ticks.')
|
|
||||||
return
|
return
|
||||||
for ytick_tripple, ax in zip(ticks, fig.axes):
|
for ytick_tripple, ax in zip(ticks, fig.axes):
|
||||||
if not ytick_tripple:
|
if not ytick_tripple:
|
||||||
@ -216,12 +224,11 @@ def set_axis_yticks(fig, parameters, verbosity=0):
|
|||||||
ax.set_ylim(ymin - 0.33 * step, ymax + 0.33 * step)
|
ax.set_ylim(ymin - 0.33 * step, ymax + 0.33 * step)
|
||||||
|
|
||||||
|
|
||||||
def plot_axis_thresholds(fig, parameters, verbosity=0):
|
def plot_axis_thresholds(fig, parameters):
|
||||||
"""
|
"""
|
||||||
Adds channel thresholds (warn, fail) to y-axis if defined in parameters.
|
Adds channel thresholds (warn, fail) to y-axis if defined in parameters.
|
||||||
"""
|
"""
|
||||||
if verbosity > 0:
|
logging.info('Plotting trace thresholds')
|
||||||
print('Plotting trace thresholds')
|
|
||||||
|
|
||||||
keys_colors = {'warn': dict(color=0.8 * get_color_mpl('WARN'), linestyle=(0, (5, 10)), alpha=0.5, linewidth=0.7),
|
keys_colors = {'warn': dict(color=0.8 * get_color_mpl('WARN'), linestyle=(0, (5, 10)), alpha=0.5, linewidth=0.7),
|
||||||
'fail': dict(color=0.8 * get_color_mpl('FAIL'), linestyle='solid', alpha=0.5, linewidth=0.7)}
|
'fail': dict(color=0.8 * get_color_mpl('FAIL'), linestyle='solid', alpha=0.5, linewidth=0.7)}
|
||||||
@ -235,6 +242,10 @@ def plot_axis_thresholds(fig, parameters, verbosity=0):
|
|||||||
|
|
||||||
def plot_threshold_lines(fig, channel_threshold_list, parameters, **kwargs):
|
def plot_threshold_lines(fig, channel_threshold_list, parameters, **kwargs):
|
||||||
for channel_thresholds, ax in zip(channel_threshold_list, fig.axes):
|
for channel_thresholds, ax in zip(channel_threshold_list, fig.axes):
|
||||||
|
if channel_thresholds in ['pb_SOH2', 'pb_SOH3']:
|
||||||
|
annotate_voltage_states(ax, parameters, channel_thresholds)
|
||||||
|
channel_thresholds = get_warn_states_pbox(channel_thresholds, parameters)
|
||||||
|
|
||||||
if not channel_thresholds:
|
if not channel_thresholds:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
@ -244,5 +255,27 @@ def plot_threshold_lines(fig, channel_threshold_list, parameters, **kwargs):
|
|||||||
for warn_thresh in channel_thresholds:
|
for warn_thresh in channel_thresholds:
|
||||||
if isinstance(warn_thresh, str):
|
if isinstance(warn_thresh, str):
|
||||||
warn_thresh = parameters.get('THRESHOLDS').get(warn_thresh)
|
warn_thresh = parameters.get('THRESHOLDS').get(warn_thresh)
|
||||||
if type(warn_thresh in (float, int)):
|
if isinstance(warn_thresh, (float, int)):
|
||||||
ax.axhline(warn_thresh, **kwargs)
|
ax.axhline(warn_thresh, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
def get_warn_states_pbox(soh_key: str, parameters: dict) -> list:
|
||||||
|
pb_dict = parameters.get('POWBOX').get(soh_key)
|
||||||
|
if not pb_dict:
|
||||||
|
return []
|
||||||
|
return [key for key in pb_dict.keys() if key > 1]
|
||||||
|
|
||||||
|
|
||||||
|
def annotate_voltage_states(ax, parameters, pb_key, color='0.75'):
|
||||||
|
for voltage, voltage_dict in parameters.get('POWBOX').get(pb_key).items():
|
||||||
|
if float(voltage) < 1:
|
||||||
|
continue
|
||||||
|
out_string = ''
|
||||||
|
for key, val in voltage_dict.items():
|
||||||
|
if val != 'OK':
|
||||||
|
if out_string:
|
||||||
|
out_string += ' | '
|
||||||
|
out_string += f'{key}: {val}'
|
||||||
|
|
||||||
|
ax.annotate(out_string, (ax.get_xlim()[-1], voltage), color=color, fontsize='xx-small',
|
||||||
|
horizontalalignment='right')
|
||||||
|
Loading…
Reference in New Issue
Block a user