From 6902ffdca6f28584bca7c01ba124e3204742b6d2 Mon Sep 17 00:00:00 2001
From: julius <julius@0124816.xyz>
Date: Wed, 16 Apr 2025 12:59:50 +0200
Subject: [PATCH] fix: weigh dislike negatively for weighted degree

---
 cutt/analysis.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/cutt/analysis.py b/cutt/analysis.py
index 4d4876d..30da656 100644
--- a/cutt/analysis.py
+++ b/cutt/analysis.py
@@ -58,6 +58,7 @@ def graph_json(
     request: Annotated[
         TeamScopedRequest, Security(verify_team_scope, scopes=["analysis"])
     ],
+    networkx_graph: bool = False,
 ):
     nodes = []
     edges = []
@@ -124,11 +125,22 @@ def graph_json(
         )
     G = nx.DiGraph()
     G.add_nodes_from([n["id"] for n in nodes])
-    G.add_weighted_edges_from([(e["source"], e["target"], e["size"]) for e in edges])
+    G.add_weighted_edges_from(
+        [
+            (
+                e["source"],
+                e["target"],
+                e["size"] if e["data"]["relation"] == 2 else -e["size"],
+            )
+            for e in edges
+        ]
+    )
     in_degrees = G.in_degree(weight="weight")
     nodes = [
         dict(node, **{"data": {"inDegree": in_degrees[node["id"]]}}) for node in nodes
     ]
+    if networkx_graph:
+        return G
     return JSONResponse({"nodes": nodes, "edges": edges})