Changeset View
Changeset View
Standalone View
Standalone View
swh/graph/client.py
Show All 24 Lines | def raw_verb_lines(self, verb, endpoint, **kwargs): | ||||
response = self.raw_verb(verb, endpoint, stream=True, **kwargs) | response = self.raw_verb(verb, endpoint, stream=True, **kwargs) | ||||
self.raise_for_status(response) | self.raise_for_status(response) | ||||
for line in response.iter_lines(): | for line in response.iter_lines(): | ||||
yield line.decode().lstrip("\n") | yield line.decode().lstrip("\n") | ||||
def get_lines(self, endpoint, **kwargs): | def get_lines(self, endpoint, **kwargs): | ||||
yield from self.raw_verb_lines("get", endpoint, **kwargs) | yield from self.raw_verb_lines("get", endpoint, **kwargs) | ||||
# Web API endpoints without arguments (bad request) | |||||
def leaves_BR(self): | |||||
return self.get("leaves") | |||||
def neighbors_BR(self): | |||||
return self.get("neighbors") | |||||
def visit_nodes_BR(self): | |||||
return self.get("visit/nodes") | |||||
def visit_edges_BR(self): | |||||
return self.get("visit/edges") | |||||
def visit_paths_BR(self): | |||||
return self.get("visit/paths") | |||||
def random_walk_BR(self): | |||||
return self.get("randomwalk") | |||||
def count_leaves_BR(self): | |||||
return self.get("leaves/count") | |||||
def count_neighbors_BR(self): | |||||
return self.get("neighbors/count") | |||||
def count_visit_nodes_BR(self): | |||||
return self.get("visit/nodes/count") | |||||
# Web API endpoints | # Web API endpoints | ||||
def stats(self): | def stats(self): | ||||
return self.get("stats") | return self.get("stats") | ||||
def leaves(self, src, edges="*", direction="forward"): | def leaves(self, src, edges="*", direction="forward"): | ||||
return self.get_lines( | return self.get_lines( | ||||
"leaves/{}".format(src), params={"edges": edges, "direction": direction} | "leaves/{}".format(src), params={"edges": edges, "direction": direction} | ||||
▲ Show 20 Lines • Show All 70 Lines • Show Last 20 Lines |