closes #58 - fix custom command validation

This commit is contained in:
checktheroads 2020-07-20 07:19:56 -07:00
parent 84f7ce1c32
commit 78d4a75b46
2 changed files with 2 additions and 3 deletions

View file

@ -20,7 +20,6 @@ from hyperglass.configuration.models.vrfs import Vrf, Info
from hyperglass.configuration.models.proxies import Proxy
from hyperglass.configuration.models.networks import Network
from hyperglass.configuration.models.credentials import Credential
from hyperglass.configuration.models.commands.common import CommandGroup
_default_vrf = {
"name": "default",
@ -51,7 +50,7 @@ class Router(HyperglassModel):
port: StrictInt
ssl: Optional[Ssl]
nos: StrictStr
commands: Optional[CommandGroup]
commands: Optional[StrictStr]
vrfs: List[Vrf] = [_default_vrf]
display_vrfs: List[StrictStr] = []
vrf_names: List[StrictStr] = []

View file

@ -134,7 +134,7 @@ class Construct:
self.query_data.query_type,
)
else:
cmd_paths = (self.device.nos, afi.protocol, self.query_data.query_type)
cmd_paths = (self.device.commands, afi.protocol, self.query_data.query_type)
command = attrgetter(".".join(cmd_paths))(commands)
return command.format(