mirror of
https://github.com/checktheroads/hyperglass
synced 2024-05-11 05:55:08 +00:00
35 lines
889 B
Python
35 lines
889 B
Python
"""Validate credential configuration variables."""
|
|
|
|
# Third Party
|
|
from pydantic import SecretStr
|
|
|
|
# Project
|
|
from hyperglass.configuration.models._utils import HyperglassModel, clean_name
|
|
|
|
|
|
class Credential(HyperglassModel):
|
|
"""Model for per-credential config in devices.yaml."""
|
|
|
|
username: str
|
|
password: SecretStr
|
|
|
|
|
|
class Credentials(HyperglassModel):
|
|
"""Base model for credentials class."""
|
|
|
|
@classmethod
|
|
def import_params(cls, input_params):
|
|
"""Import credentials with corrected field names.
|
|
|
|
Arguments:
|
|
input_params {dict} -- Credential definition
|
|
|
|
Returns:
|
|
{object} -- Validated credential object
|
|
"""
|
|
obj = Credentials()
|
|
for (credname, params) in input_params.items():
|
|
cred = clean_name(credname)
|
|
setattr(Credentials, cred, Credential(**params))
|
|
return obj
|