From add0a1af48144c5c3a7279e63d6e598ad4470142 Mon Sep 17 00:00:00 2001
From: checktheroads <matt@allroads.io>
Date: Sun, 13 Dec 2020 01:47:52 -0700
Subject: [PATCH] clean up frontend data model [skip ci]

---
 hyperglass/configuration/main.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/hyperglass/configuration/main.py b/hyperglass/configuration/main.py
index a513f18..e71292d 100644
--- a/hyperglass/configuration/main.py
+++ b/hyperglass/configuration/main.py
@@ -358,7 +358,12 @@ def _build_networks():
                         "display_name": device.display_name,
                         "network": device.network.display_name,
                         "vrfs": [
-                            {"id": vrf.name, "display_name": vrf.display_name}
+                            {
+                                "id": vrf.name,
+                                "display_name": vrf.display_name,
+                                "ipv4": True if vrf.ipv4 else False,  # noqa: IF100
+                                "ipv6": True if vrf.ipv6 else False,  # noqa: IF100
+                            }
                             for vrf in device.vrfs
                         ],
                     }