Create shared models for config-feeder databricks api keys

Granulate · Jul 13, 2023 · 2d91c52 · 2d91c52
1 parent 0ccbec5
commit 2d91c52
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/granulate_utils/config_feeder/core/models/databricks.py b/granulate_utils/config_feeder/core/models/databricks.py
@@ -0,0 +1,28 @@
+import datetime
+
+from pydantic import BaseModel
+
+
+class DatabricksApiKeyBase(BaseModel):
+    service_name: int
+    api_key: str
+
+
+class DatabricksApiKey(BaseModel):
+    id: int
+    client_id: int
+    service_id: int
+    api_key: str
+    ts: datetime.datetime
+
+
+class CreateApiKeyRequest(BaseModel):
+    api_key: DatabricksApiKeyBase
+
+
+class CreateApiKeyResponse(BaseModel):
+    api_key: DatabricksApiKey
+
+
+class GetApiKeyResponse(BaseModel):
+    api_key: DatabricksApiKey