http-api: T9223: add VRF option to traceroute REST API endpoint

Author: jvoss

python/vyos/configsession.py

@@ -494,6 +494,9 @@ def ping(self, host, count: int = 5, vrf: str | None = None):
         out = self.__run_command(cmd)
         return out
 
-    def traceroute(self, host):
-        out = self.__run_command(TRACEROUTE + [host])
+    def traceroute(self, host, vrf: str | None = None):
+        cmd = TRACEROUTE + [host]
+        if vrf is not None:
+            cmd += ['--vrf', vrf]
+        out = self.__run_command(cmd)
         return out

src/services/api/rest/models.py

@@ -332,13 +332,15 @@ class Config:
 class TracerouteModel(ApiModel):
     op: StrictStr
     host: StrictStr
+    vrf: StrictStr | None = None
 
     class Config:
         schema_extra = {
             'example': {
                 'key': 'id_key',
                 'op': 'traceroute',
                 'host': 'host',
+                'vrf': 'vrf',
             }
         }
 

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: