feat: change API output

This commit is contained in:
2025-02-20 08:54:03 +01:00
parent 8e91724462
commit 1fa91a7228
4 changed files with 27 additions and 21 deletions

View File

@@ -24,10 +24,10 @@ P = Player
def sociogram_json():
nodes = []
necessary_nodes = set()
links = []
edges = []
with Session(engine) as session:
for p in session.exec(select(P)).fetchall():
nodes.append({"id": p.name, "appearance": 1})
nodes.append({"id": p.name, "label": p.name})
subquery = (
select(C.user, func.max(C.time).label("latest"))
.where(C.time > datetime(2025, 2, 1, 10))
@@ -44,9 +44,11 @@ def sociogram_json():
# G.add_edge(c.user, p)
# p_id = session.exec(select(P.id).where(P.name == p)).one()
necessary_nodes.add(p)
links.append({"source": c.user, "target": p})
edges.append({"from": c.user, "to": p, "relation": "likes"})
for p in c.hate:
edges.append({"from": c.user, "to": p, "relation": "dislikes"})
# nodes = [n for n in nodes if n["name"] in necessary_nodes]
return JSONResponse({"nodes": nodes, "links": links})
return JSONResponse({"nodes": nodes, "edges": edges})
def sociogram_data(show: int | None = 2):