From ee59c077191010206bf6e66c5f13e6c7bcbe89a1 Mon Sep 17 00:00:00 2001 From: qazal Date: Wed, 25 Sep 2024 17:36:09 +0800 Subject: [PATCH] fix matcher with ctx --- viz/serve.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/viz/serve.py b/viz/serve.py index 848e3f061f..709ba299cc 100755 --- a/viz/serve.py +++ b/viz/serve.py @@ -28,7 +28,8 @@ class RewriteLocation: fp, lineno = ctx.loc p = r"graph_rewrite\([^,]+,\s*([^>]+)\)" match = re.search(p, code:=lines(fp)[lineno-1].strip()) - return RewriteLocation(f"{fp.split('/')[-1]}:{lineno}", code, match.group(1) if match is not None else None, len(graph_rewrites(ctx))) + return RewriteLocation(f"{fp.split('/')[-1]}:{lineno}", code, match.group(1).split(",")[0] if match is not None else None, + len(graph_rewrites(ctx))) def to_json(self): return asdict(self) @dataclass(frozen=True)