DO NOT MERGE Preview Well Known Types

.codegen/_openapi_sha

@@ -1 +1 @@
-b95c2c6e21bec9551ec7d7d51ddf2dfe390b4522
+file:/home/hector.castejon/universe/bazel-bin/openapi/all-internal.json

.gitattributes

@@ -10,8 +10,11 @@ databricks/sdk/service/compute.py linguist-generated=true
 databricks/sdk/service/dashboards.py linguist-generated=true
 databricks/sdk/service/database.py linguist-generated=true
 databricks/sdk/service/files.py linguist-generated=true
+databricks/sdk/service/httpcallv2.py linguist-generated=true
 databricks/sdk/service/iam.py linguist-generated=true
+databricks/sdk/service/iamv2.py linguist-generated=true
 databricks/sdk/service/jobs.py linguist-generated=true
+databricks/sdk/service/jsonmarshallv2.py linguist-generated=true
 databricks/sdk/service/marketplace.py linguist-generated=true
 databricks/sdk/service/ml.py linguist-generated=true
 databricks/sdk/service/oauth2.py linguist-generated=true
@@ -26,3 +29,5 @@ databricks/sdk/service/sql.py linguist-generated=true
 databricks/sdk/service/tags.py linguist-generated=true
 databricks/sdk/service/vectorsearch.py linguist-generated=true
 databricks/sdk/service/workspace.py linguist-generated=true
+test_http_call.py linguist-generated=true
+test_json_marshall.py linguist-generated=true

databricks/sdk/common/types/fieldmask.py

@@ -0,0 +1,15 @@
+class FieldMask(object):
+    """Class for FieldMask message type."""
+
+    def ToJsonString(self):
+        """Converts FieldMask to string."""
+        return ",".join(self.paths)
+
+    def FromJsonString(self, value):
+        """Converts string to FieldMask."""
+        if not isinstance(value, str):
+            raise ValueError("FieldMask JSON value not a string: {!r}".format(value))
+        if value:
+            self.paths = value.split(",")
+        else:
+            self.paths = []

databricks/sdk/service/_internal.py

@@ -1,6 +1,11 @@
 import datetime
 import urllib.parse
-from typing import Callable, Dict, Generic, Optional, Type, TypeVar
+from typing import Callable, Dict, Generic, List, Optional, Type, TypeVar
+
+from google.protobuf.duration_pb2 import Duration
+from google.protobuf.timestamp_pb2 import Timestamp
+
+from databricks.sdk.common.types.fieldmask import FieldMask
 
 
 def _from_dict(d: Dict[str, any], field: str, cls: Type) -> any:
@@ -46,6 +51,63 @@ def _escape_multi_segment_path_parameter(param: str) -> str:
     return urllib.parse.quote(param)
 
 
+def _timestamp(d: Dict[str, any], field: str) -> Optional[Timestamp]:
+    if field not in d or not d[field]:
+        return None
+    ts = Timestamp()
+    ts.FromJsonString(d[field])
+    return ts
+
+
+def _repeated_timestamp(d: Dict[str, any], field: str) -> Optional[List[Timestamp]]:
+    if field not in d or not d[field]:
+        return None
+    result = []
+    for v in d[field]:
+        ts = Timestamp()
+        ts.FromJsonString(v)
+        result.append(ts)
+    return result
+
+
+def _duration(d: Dict[str, any], field: str) -> Optional[Duration]:
+    if field not in d or not d[field]:
+        return None
+    dur = Duration()
+    dur.FromJsonString(d[field])
+    return dur
+
+
+def _repeated_duration(d: Dict[str, any], field: str) -> Optional[List[Duration]]:
+    if field not in d or not d[field]:
+        return None
+    result = []
+    for v in d[field]:
+        dur = Duration()
+        dur.FromJsonString(v)
+        result.append(dur)
+    return result
+
+
+def _fieldmask(d: Dict[str, any], field: str) -> Optional[FieldMask]:
+    if field not in d or not d[field]:
+        return None
+    fm = FieldMask()
+    fm.FromJsonString(d[field])
+    return fm
+
+
+def _repeated_fieldmask(d: Dict[str, any], field: str) -> Optional[List[FieldMask]]:
+    if field not in d or not d[field]:
+        return None
+    result = []
+    for v in d[field]:
+        fm = FieldMask()
+        fm.FromJsonString(v)
+        result.append(fm)
+    return result
+
+
 ReturnType = TypeVar("ReturnType")