mirror of
https://github.com/checktheroads/hyperglass
synced 2024-05-11 05:55:08 +00:00
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
"""Validate feature configuration variables."""
|
|
|
|
# Standard Library Imports
|
|
from math import ceil
|
|
|
|
# Third Party Imports
|
|
from pydantic import StrictBool
|
|
from pydantic import StrictInt
|
|
from pydantic import StrictStr
|
|
from pydantic import constr
|
|
|
|
# Project Imports
|
|
from hyperglass.configuration.models._utils import HyperglassModel
|
|
|
|
|
|
class Features(HyperglassModel):
|
|
"""Validation model for params.features."""
|
|
|
|
class BgpRoute(HyperglassModel):
|
|
"""Validation model for params.features.bgp_route."""
|
|
|
|
enable: StrictBool = True
|
|
display_name: StrictStr = "BGP Route"
|
|
|
|
class BgpCommunity(HyperglassModel):
|
|
"""Validation model for params.features.bgp_community."""
|
|
|
|
enable: StrictBool = True
|
|
display_name: StrictStr = "BGP Community"
|
|
|
|
class Regex(HyperglassModel):
|
|
"""Validation model for params.features.bgp_community.regex."""
|
|
|
|
decimal: StrictStr = r"^[0-9]{1,10}$"
|
|
extended_as: StrictStr = r"^([0-9]{0,5})\:([0-9]{1,5})$"
|
|
large: StrictStr = r"^([0-9]{1,10})\:([0-9]{1,10})\:[0-9]{1,10}$"
|
|
|
|
regex: Regex = Regex()
|
|
|
|
class BgpAsPath(HyperglassModel):
|
|
"""Validation model for params.features.bgp_aspath."""
|
|
|
|
enable: StrictBool = True
|
|
display_name: StrictStr = "BGP AS Path"
|
|
|
|
class Regex(HyperglassModel):
|
|
"""Validation model for params.bgp_aspath.regex."""
|
|
|
|
mode: constr(regex="asplain|asdot") = "asplain"
|
|
asplain: StrictStr = r"^(\^|^\_)(\d+\_|\d+\$|\d+\(\_\.\+\_\))+$"
|
|
asdot: StrictStr = (
|
|
r"^(\^|^\_)((\d+\.\d+)\_|(\d+\.\d+)\$|(\d+\.\d+)\(\_\.\+\_\))+$"
|
|
)
|
|
|
|
regex: Regex = Regex()
|
|
|
|
class Ping(HyperglassModel):
|
|
"""Validation model for params.features.ping."""
|
|
|
|
enable: StrictBool = True
|
|
display_name: StrictStr = "Ping"
|
|
|
|
class Traceroute(HyperglassModel):
|
|
"""Validation model for params.features.traceroute."""
|
|
|
|
enable: StrictBool = True
|
|
display_name: StrictStr = "Traceroute"
|
|
|
|
class Cache(HyperglassModel):
|
|
"""Validation model for params.features.cache."""
|
|
|
|
redis_id: StrictInt = 0
|
|
timeout: StrictInt = 120
|
|
show_text: StrictBool = True
|
|
text: StrictStr = "Results will be cached for {timeout} minutes.".format(
|
|
timeout=ceil(timeout / 60)
|
|
)
|
|
|
|
class MaxPrefix(HyperglassModel):
|
|
"""Validation model for params.features.max_prefix."""
|
|
|
|
enable: StrictBool = False
|
|
ipv4: StrictInt = 24
|
|
ipv6: StrictInt = 64
|
|
message: StrictStr = (
|
|
"Prefix length must be smaller than /{m}. <b>{i}</b> is too specific."
|
|
)
|
|
|
|
bgp_route: BgpRoute = BgpRoute()
|
|
bgp_community: BgpCommunity = BgpCommunity()
|
|
bgp_aspath: BgpAsPath = BgpAsPath()
|
|
ping: Ping = Ping()
|
|
traceroute: Traceroute = Traceroute()
|
|
cache: Cache = Cache()
|
|
max_prefix: MaxPrefix = MaxPrefix()
|