80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
|
|
class Alias:
|
|
def __init__(self, alias, channel=None, channeltype=None, parent=None):
|
|
self.alias = alias
|
|
self.channel = channel
|
|
self.channeltype = channeltype
|
|
self.children = []
|
|
self.parent = parent
|
|
|
|
def append(self, subalias):
|
|
assert type(subalias) is Alias, "You can only append aliases to aliases!"
|
|
assert not (
|
|
subalias.alias in [tc.alias for tc in self.children]
|
|
), f"Alias {subalias.alias} exists already!"
|
|
self.children.append(subalias)
|
|
if subalias.parent is None:
|
|
subalias.parent = self
|
|
else:
|
|
print(subalias.parent)
|
|
logger.warning(f'parent of alias {subalias.alias} has been defined already {subalias.parent.alias}.')
|
|
|
|
def get_all(self, joiner="."):
|
|
aa = []
|
|
if self.channel:
|
|
ta = {}
|
|
ta["alias"] = self.alias
|
|
ta["channel"] = self.channel
|
|
if self.channeltype:
|
|
ta["channeltype"] = self.channeltype
|
|
aa.append(ta)
|
|
if self.children:
|
|
for tc in self.children:
|
|
taa = tc.get_all()
|
|
for ta in taa:
|
|
aa.append(
|
|
{
|
|
"alias": joiner.join([self.alias, ta["alias"]]),
|
|
"channel": ta["channel"],
|
|
"channeltype": ta["channeltype"],
|
|
}
|
|
)
|
|
return aa
|
|
|
|
def get_full_name(self,joiner="."):
|
|
name = [self.alias]
|
|
parent = self.parent
|
|
while not parent == None:
|
|
name.append(parent.alias)
|
|
parent = parent.__dict__.get('parent',None)
|
|
if joiner:
|
|
return joiner.join(reversed(name))
|
|
else:
|
|
return name
|
|
|
|
|
|
|
|
# def add_children(self, *args):
|
|
# self.children.append(find_aliases(*args))
|
|
|
|
|
|
#def find_aliases(*args):
|
|
# o = []
|
|
# for obj in args:
|
|
# if isinstance(obj, Alias):
|
|
# o.append(obj)
|
|
# if hasattr(obj, "alias"):
|
|
# obj = obj.alias
|
|
# return tuple(o)
|
|
|
|
|
|
def append_object_to_object(obj_target, obj_init, *args, name=None, **kwargs):
|
|
"""append a new object to another object together with the alias.
|
|
The new object needs to be defined with a name keyword. Theadditional
|
|
args and kwargs are the expected input for the new opject."""
|
|
obj_target.__dict__[name] = obj_init(*args, **kwargs, name=name)
|
|
obj_target.alias.append(obj_target.__dict__[name].alias)
|
|
|
|
|
|
|