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

allow int or str for primary_asn

This commit is contained in:
checktheroads
2020-01-28 10:06:43 -07:00
parent 0cfa3472f7
commit 492e7917a1

View File

@@ -31,7 +31,7 @@ class Params(HyperglassModel):
# Top Level Params # Top Level Params
debug: StrictBool = False debug: StrictBool = False
developer_mode: StrictBool = False developer_mode: StrictBool = False
primary_asn: StrictStr = "65001" primary_asn: Union[StrictInt, StrictStr] = "65001"
org_name: StrictStr = "The Company" org_name: StrictStr = "The Company"
site_title: StrictStr = "hyperglass" site_title: StrictStr = "hyperglass"
site_description: StrictStr = "{org_name} Network Looking Glass" site_description: StrictStr = "{org_name} Network Looking Glass"
@@ -121,3 +121,17 @@ class Params(HyperglassModel):
f'/tmp/hyperglass_{now.strftime(r"%Y%M%d-%H%M%S")}.log' # noqa: S108 f'/tmp/hyperglass_{now.strftime(r"%Y%M%d-%H%M%S")}.log' # noqa: S108
) )
return value return value
@validator("primary_asn")
def validate_primary_asn(cls, value):
"""Stringify primary_asn if passed as an integer.
Arguments:
value {str|int} -- Unvalidated Primary ASN
Returns:
{str} -- Stringified Primary ASN.
"""
if not isinstance(value, str):
value = str(value)
return value