diff --git a/src/sakia/data/graphs/base_graph.py b/src/sakia/data/graphs/base_graph.py index f635596311ea4cabe8a62d033e26bda92205ef18..a7c4b3c4e0e4753c59228fd97c8f66b9a9697a79 100644 --- a/src/sakia/data/graphs/base_graph.py +++ b/src/sakia/data/graphs/base_graph.py @@ -139,34 +139,6 @@ class BaseGraph(QObject): self.nx_graph.add_edge(identity.pubkey, certified.pubkey, attr_dict=arc) - def add_offline_certifier_list(self, certifier_list, identity): - """ - Add list of certifiers to graph - :param List[sakia.data.entities.Certification] certifier_list: List of certified from api - :param sakia.data.entities.Identity identity: identity instance which is certified - :return: - """ - # add certifiers of uid - for certification in tuple(certifier_list): - certifier = self.identities_service.get_identity(certification.certifier) - if certifier: - node_status = self.node_status(certifier) - self.add_certifier_node(certifier, identity, certification, node_status) - - def add_offline_certified_list(self, certified_list, identity): - """ - Add list of certified from api to graph - :param List[sakia.data.entities.Certification] certified_list: List of certified from api - :param identity identity: identity instance which is certifier - :return: - """ - # add certified by uid - for certification in tuple(certified_list): - certified = self.identities_service.get_identity(certification.certified) - if certified: - node_status = self.node_status(certified) - self.add_certified_node(identity, certified, certification, node_status) - def add_certifier_list(self, certifier_list, identity): """ Add list of certifiers to graph diff --git a/src/sakia/data/graphs/wot_graph.py b/src/sakia/data/graphs/wot_graph.py index 5c94e63e7ed24728906e89ff31f059060ff94f29..6d18e4b42bacd685ec19453b4f2b8b7a19ca7370 100644 --- a/src/sakia/data/graphs/wot_graph.py +++ b/src/sakia/data/graphs/wot_graph.py @@ -44,7 +44,9 @@ class WoTGraph(BaseGraph): # populate graph with certifiers-of certifier_list = self.identities_service.certifications_received(center_identity.pubkey) - self.add_offline_certifier_list(certifier_list, center_identity) + certifier_list = {c: self.identities_service.get_identity(c.certifier) for c in certifier_list} + self.add_certifier_list(certifier_list, center_identity) # populate graph with certified-by certified_list = self.identities_service.certifications_sent(center_identity.pubkey) - self.add_offline_certified_list(certified_list, center_identity) + certified_list = {c: self.identities_service.get_identity(c.certified) for c in certified_list} + self.add_certified_list(certified_list, center_identity)