datamodel: stub-zone: added StubServerSchema

- added to be in line with forward-zones

diff --git a/manager/knot_resolver_manager/datamodel/stub_zone_schema.py b/manager/knot_resolver_manager/datamodel/stub_zone_schema.py
index 64e61c152535bf41e2990c2f10f1b5548115f2a8..ffd43bc787d9412a4d4b70cd4d5c7a2447304871 100644 (file)
--- a/manager/knot_resolver_manager/datamodel/stub_zone_schema.py
+++ b/manager/knot_resolver_manager/datamodel/stub_zone_schema.py
@@ -1,11 +1,15 @@
-from typing import List, Optional
+from typing import List, Optional, Union
 
 from knot_resolver_manager.datamodel.types import IPAddressPort
 from knot_resolver_manager.datamodel.view_schema import FlagsEnum
 from knot_resolver_manager.utils import SchemaNode
 
 
+class StubServerSchema(SchemaNode):
+    address: IPAddressPort
+
+
 class StubZoneSchema(SchemaNode):
-    servers: List[IPAddressPort]
+    servers: Union[List[IPAddressPort], List[StubServerSchema]]
     views: Optional[List[str]] = None
     options: Optional[List[FlagsEnum]] = None