import time
from timeit import default_timer
import logging
logger = logging.getLogger(__file__)
from collections import OrderedDict, Counter
[docs]def isnotebook():
""" returns True if Jupyter notebook is runnung """
# from https://stackoverflow.com/questions/15411967/how-can-i-check-if-code-is-executed-in-the-ipython-notebook
try:
shell = get_ipython().__class__.__name__
if shell == 'ZMQInteractiveShell':
return True # Jupyter notebook or qtconsole
elif shell == 'TerminalInteractiveShell':
return False # Terminal running IPython
else:
return False # Other type (?)
except NameError:
return False # Probably standard Python interpreter
[docs]def time():
""" returns the time. used instead of time.time for rapid portability"""
return default_timer()
[docs]def get_unique_name_list_from_class_list(cls_list):
"""
returns a list of names using cls.name if unique or cls.name1, cls.name2... otherwise.
Order of the name list matches order of cls_list, such that iterating over zip(cls_list, name_list) is OK
"""
# cls_list is typically
# cls_modules = [rp.HK, rp.AMS, rp.Scope, rp.Sampler, rp.Asg1, rp.Asg2] + \
# [rp.AuxOutput] * 2 + [rp.IQ] * 3 + [rp.Pid] * 4 + [rp.IIR]
# first, map from list of classes to a list of corresponding names
# e.g. all_names = ['hk, ..., 'pwm', 'pwm', ...
all_names = [cls.__name__.lower() for cls in cls_list]
final_names = []
for name in all_names:
# how many times does the name occur?
occurences = all_names.count(name)
if occurences == 1:
# for single names, leave as-is
final_names.append(name)
else:
# for multiple name, assign name+str(lowest_free_number)
for i in range(occurences):
if not name+str(i) in final_names:
final_names.append(name+str(i))
break
return final_names
[docs]def get_class_name_from_module_name(module_name):
""" returns the class name corresponding to a module_name """
return module_name[0].upper() + (module_name[1:]).rstrip('1234567890')
[docs]def get_base_module_class(module):
""" returns the base class of module that has the same name as module """
base_module_class_name = get_class_name_from_module_name(module.name)
for base_module_class in type(module).__mro__:
if base_module_class.__name__ == base_module_class_name:
return base_module_class
# see http://stackoverflow.com/questions/3862310/how-can-i-find-all-subclasses-of-a-class-given-its-name
[docs]def all_subclasses(cls):
""" returns a list of all subclasses of cls """
return cls.__subclasses__() + [g for s in cls.__subclasses__()
for g in all_subclasses(s)]
[docs]def recursive_getattr(root, path):
""" returns root.path (i.e. root.attr1.attr2) """
attribute = root
for name in path.split('.'):
if name != "":
attribute = getattr(attribute, name)
return attribute
[docs]def recursive_setattr(root, path, value):
""" returns root.path = value (i.e. root.attr1.attr2 = value) """
attribute = root
names = path.split('.')
for name in names[:-1]:
attribute = getattr(attribute, name)
setattr(attribute, names[-1], value)
[docs]def setloglevel(level='info', loggername='pyrpl'):
""" sets the log level to the one specified in config file"""
try:
loglevels = {"notset": logging.NOTSET,
"debug": logging.DEBUG,
"info": logging.INFO,
"warning": logging.WARNING,
"error": logging.ERROR,
"critical": logging.CRITICAL}
level = loglevels[level]
except:
pass
else:
logging.getLogger(name=loggername).setLevel(level)
[docs]class DuplicateFilter(logging.Filter):
"""
Prevent multiple repeated logging message from polluting the console
"""
[docs] def filter(self, record):
# add other fields if you need more granular comparison, depends on your app
current_log = (record.module, record.levelno, record.msg)
if current_log != getattr(self, "last_log", None):
self.last_log = current_log
return True
return False
[docs]def sorted_dict(dict_to_sort=None, sort_by_values=True, **kwargs):
if dict_to_sort is None:
dict_to_sort = kwargs
if not sort_by_values:
return OrderedDict(sorted(dict_to_sort.items()))
else:
return OrderedDict(sorted(dict_to_sort.items(), key=lambda x: x[1]))
[docs]def update_with_typeconversion(dictionary, update):
for k, v in update.items():
if k in dictionary:
# perform type conversion if appropriate
v = type(dictionary[k])(v)
dictionary[k] = v
return dictionary
[docs]def unique_list(nonunique_list):
""" Returns a list where each element of nonunique_list occurs exactly once.
The last occurence of an element defines its position in the returned list.
"""
unique_list = []
for attr in reversed(nonunique_list):
# remove all previous occurences
if attr not in unique_list:
unique_list.insert(0, attr)
return unique_list
[docs]class Bijection(dict):
""" This class defines a bijection object based on dict
It can be used exactly like dict, but additionally has a property
'inverse' which contains the inverted {value: key} dict. """
def __init__(self, *args, **kwargs):
super(Bijection, self).__init__(*args, **kwargs)
self.inverse = {v: k for k, v in self.items()}
def __setitem__(self, key, value):
super(Bijection, self).__setitem__(key, value)
self.inverse[value] = key
def __delitem__(self, key):
self.inverse.__delitem__(self.__getitem__(key))
super(Bijection, self).__delitem__(key)
[docs] def pop(self, key):
self.inverse.pop(self.__getitem__(key))
super(Bijection, self).pop(key)
[docs] def update(self, *args, **kwargs):
super(Bijection, self).update(*args, **kwargs)
self.inverse = {v: k for k, v in self.items()}