Fix deserialization of repeated enums

.codegen/service.py.tmpl

@@ -8,7 +8,7 @@ import time
 import random
 import logging
 from ..errors import OperationTimeout, OperationFailed
-from ._internal import _enum, _from_dict, _repeated, Wait
+from ._internal import _enum, _from_dict, _repeated_dict, _repeated_enum, Wait
 
 _LOG = logging.getLogger('databricks.sdk')
 
@@ -45,8 +45,9 @@ class {{.PascalName}}{{if eq "List" .PascalName}}Request{{end}}:{{if .Descriptio
 {{- define "from_dict_type" -}}
 	{{- if not .Entity }}None
 	{{- else if .Entity.ArrayValue }}
-		{{- if .Entity.ArrayValue.IsObject }}_repeated(d, '{{.Name}}', {{template "type-nq" .Entity.ArrayValue}})
-		{{- else if .Entity.ArrayValue.IsExternal }}_repeated(d, '{{.Name}}', {{template "type-nq" .Entity.ArrayValue}})
+		{{- if .Entity.ArrayValue.IsObject }}_repeated_dict(d, '{{.Name}}', {{template "type-nq" .Entity.ArrayValue}})
+		{{- else if .Entity.ArrayValue.IsExternal }}_repeated_dict(d, '{{.Name}}', {{.Entity.Package.Name}}.{{template "type-nq" .Entity.ArrayValue}})
+		{{- else if .Entity.ArrayValue.Enum }}_repeated_enum(d, '{{.Name}}', {{template "type-nq" .Entity.ArrayValue}})
 		{{- else}}d.get('{{.Name}}', None){{- end -}}
 	{{- else if .Entity.IsObject }}_from_dict(d, '{{.Name}}', {{.Entity.PascalName}})
 	{{- else if .Entity.IsExternal }}_from_dict(d, '{{.Name}}', {{.Entity.Package.Name}}.{{.Entity.PascalName}})

databricks/sdk/service/_internal.py

@@ -1,5 +1,5 @@
 import datetime
-from typing import Callable, Dict, Generic, Type, TypeVar
+from typing import Callable, Dict, Generic, Optional, Type, TypeVar
 
 
 def _from_dict(d: Dict[str, any], field: str, cls: Type) -> any:
@@ -8,17 +8,34 @@ def _from_dict(d: Dict[str, any], field: str, cls: Type) -> any:
     return getattr(cls, 'from_dict')(d[field])
 
 
-def _repeated(d: Dict[str, any], field: str, cls: Type) -> any:
+def _repeated_dict(d: Dict[str, any], field: str, cls: Type) -> any:
     if field not in d or not d[field]:
         return None
     from_dict = getattr(cls, 'from_dict')
     return [from_dict(v) for v in d[field]]
 
 
+def _get_enum_value(cls: Type, value: str) -> Optional[Type]:
+    return next((v for v in getattr(cls, '__members__').values() if v.value == value), None)
+
+
 def _enum(d: Dict[str, any], field: str, cls: Type) -> any:
+    """Unknown enum values are returned as None."""
+    if field not in d or not d[field]:
+        return None
+    return _get_enum_value(cls, d[field])
+
+
+def _repeated_enum(d: Dict[str, any], field: str, cls: Type) -> any:
+    """For now, unknown enum values are not included in the response."""
     if field not in d or not d[field]:
         return None
-    return next((v for v in getattr(cls, '__members__').values() if v.value == d[field]), None)
+    res = []
+    for e in d[field]:
+        val = _get_enum_value(cls, e)
+        if val:
+            res.append(val)
+    return res
 
 
 ReturnType = TypeVar('ReturnType')