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

strip whitespace from query target input, closes #75

Browse Source
This commit is contained in:
checktheroads
2020-09-28 08:09:48 -07:00
parent fde39bf2ff
commit be7a3645e4
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
hyperglass/api/models/query.py
View File

@ -7,10 +7,9 @@ import secrets
from datetime import datetime from datetime import datetime
# Third Party # Third Party
from pydantic import BaseModel, StrictStr, validator from pydantic import BaseModel, StrictStr, validator, constr
# Project # Project
from hyperglass.log import log
from hyperglass.exceptions import InputInvalid from hyperglass.exceptions import InputInvalid
from hyperglass.configuration import params, devices from hyperglass.configuration import params, devices
from hyperglass.api.models.types import SupportedQuery from hyperglass.api.models.types import SupportedQuery
@ -55,7 +54,7 @@ class Query(BaseModel):
query_location: StrictStr query_location: StrictStr
query_type: SupportedQuery query_type: SupportedQuery
query_vrf: StrictStr query_vrf: StrictStr
query_target: StrictStr query_target: constr(strip_whitespace=True, min_length=1)
class Config: class Config:
"""Pydantic model configuration.""" """Pydantic model configuration."""
@ -229,8 +228,8 @@ class Query(BaseModel):
def validate_query_target(cls, value, values): def validate_query_target(cls, value, values):
"""Validate query target value based on query_type.""" """Validate query target value based on query_type."""
log.debug(values)
query_type = values["query_type"] query_type = values["query_type"]
value = value.strip()
# Use relevant function based on query_type. # Use relevant function based on query_type.
validator_map = { validator_map = {