From 27a64c21fe6d814f9b7c191e94430c5e1b7eb817 Mon Sep 17 00:00:00 2001
From: Jonathan Voss <jvoss@onvox.net>
Date: Mon, 17 Aug 2026 04:40:23 +0000
Subject: http-api: T9223: add VRF option to traceroute REST API endpoint

---
 src/services/api/rest/models.py  | 2 ++
 src/services/api/rest/routers.py | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/services/api/rest/models.py b/src/services/api/rest/models.py
index d3ee3a08e..abe7e7726 100644
--- a/src/services/api/rest/models.py
+++ b/src/services/api/rest/models.py
@@ -332,6 +332,7 @@ class PingModel(ApiModel):
 class TracerouteModel(ApiModel):
     op: StrictStr
     host: StrictStr
+    vrf: StrictStr | None = None
 
     class Config:
         schema_extra = {
@@ -339,6 +340,7 @@ class TracerouteModel(ApiModel):
                 'key': 'id_key',
                 'op': 'traceroute',
                 'host': 'host',
+                'vrf': 'vrf',
             }
         }
 
diff --git a/src/services/api/rest/routers.py b/src/services/api/rest/routers.py
index a241cabf8..7b0762084 100644
--- a/src/services/api/rest/routers.py
+++ b/src/services/api/rest/routers.py
@@ -992,10 +992,11 @@ def traceroute_op(data: TracerouteModel):
 
     op = data.op
     host = data.host
+    vrf = data.vrf
 
     try:
         if op == 'traceroute':
-            res = session.traceroute(host)
+            res = session.traceroute(host, vrf)
         else:
             return error(400, f"'{op}' is not a valid operation")
     except ConfigSessionError as e:
-- 
cgit v1.2.3