Есть ли элегантный способ отслеживать наборы связанных элементов в Python?

Для определенной части кода мне нужно было найти способ распознать определенные псевдонимы. Дело в том, что заранее не известно, что эти псевдонимы.

Это мои требования:

• Если A и B - псевдонимы, а B и C - псевдонимы, A и C также должны быть алиасами.
• Два набора псевдонимов должны быть объединены, когда они связаны каким-либо образом.
• В каждом наборе псевдонимов должен быть первичный псевдоним.

Я использую следующее решение, используя то, что сводится к словарю множеств:

class Alias(object):
    def __init__(self, initial):
        self._set = {initial}
        self.initial = initial
    def add(self, alias):
        self._set.add(alias)
    def merge(self, other):
        self._set.update(other._set)
    def __iter__(self):
        return iter(self._set)

class AliasDict(object):
    def __init__(self):
        self._dict = {}
    def add(self, one, other):
        if one in self._dict:
            if other in self._dict: #merge!
                self._dict[one].merge(self._dict[other])
                for k in self._dict[other]:
                    self._dict[k] = self._dict[one]
            else:
                self._dict[one].add(other)
        elif other in self._dict:
            self._dict[other].add(one)
        else:
            self._dict[one] = self._dict[other] = Alias(one)
            self._dict[one].add(other)
    def get(self, n):
        return self._dict.get(n)
    def __contains__(self, s):
        return s in self._dict

Можно ли улучшить это? Например, используя класс в стандартной библиотеке (я искал, но, возможно, я пропустил что-то полезное.)