Class EdgeLabel

python/graphscope/framework/graph_schema.py:204–246

class EdgeLabel(Label):
    __slots__ = ["_relations"]

    def __init__(self, name, label_id=0):
        super().__init__(name, label_id)
        self._relations: List[Relation] = []

    @property
    def type_enum(self):
        return graph_def_pb2.TypeEnumPb.EDGE

    def source(self, label):
        self._relations.append(Relation(label, ""))
        return self

    def destination(self, label):
        assert (
            self._relations
        ), "Found empty relation, maybe you should use `source` first."
        assert not self._relations[-1].destination, "An destination is already exists."
        self._relations[-1] = self._relations[-1]._replace(destination=label)
        return self

    @property
    def relations(self) -> List[Relation]:
        return self._relations

    def __repr__(self) -> str:
        s = super().__repr__()
        if self._relations:
            s += f"Relations: {self.relations}"
        return s

    def to_dict(self) -> dict:
        # super dict
        sd = super().to_dict()
        relations = []
        for r in self._relations:
            relations.append(
                {"source_vertex": r.source, "destination_vertex": r.destination}
            )
        sd.update({"vertex_type_pair_relations": relations})
        return sd


class GraphSchema: