1
0
mirror of https://github.com/checktheroads/hyperglass synced 2024-05-11 05:55:08 +00:00

allow NoneType for AFIs

This commit is contained in:
checktheroads
2019-09-13 08:31:46 -07:00
parent 17d7648b9c
commit 4f00762ec9

View File

@@ -31,34 +31,6 @@ class AfiMap(BaseSettings):
ipv4_vpn: Union[str, None] = None
ipv6_vpn: Union[str, None] = None
@validator("ipv4", always=True)
def validate_ipv4(cls, v): # noqa: N805
"""If a map field is undefined, get a default value"""
if v is None:
v = afi_nos_map.get("default").get("ipv4")
return v
@validator("ipv6", always=True)
def validate_ipv6(cls, v): # noqa: N805
"""If a map field is undefined, get a default value"""
if v is None:
v = afi_nos_map.get("default").get("ipv6")
return v
@validator("ipv4_vpn", always=True)
def validate_ipv4_vpn(cls, v): # noqa: N805
"""If a map field is undefined, get a default value"""
if v is None:
v = afi_nos_map.get("default").get("ipv4_vpn")
return v
@validator("ipv6_vpn", always=True)
def validate_ipv6_vpn(cls, v): # noqa: N805
"""If a map field is undefined, get a default value"""
if v is None:
v = afi_nos_map.get("default").get("ipv6_vpn")
return v
class Router(BaseSettings):
"""Model for per-router config in devices.yaml."""