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