Make code working with new OpenAPI packaging (#78)

.codegen/__init__.py.tmpl

@@ -1,14 +1,21 @@
-{{range .Packages}}
-import databricks.sdk.service.{{.Name}} as {{.Name}}{{end}}
 import databricks.sdk.core as client
 import databricks.sdk.dbutils as dbutils
-import databricks.sdk.mixins.dbfs as dbfs_mixin
-import databricks.sdk.mixins.compute as compute_mixin
+
+from databricks.sdk.mixins.dbfs import DbfsExt
+from databricks.sdk.mixins.compute import ClustersExt
+{{- range .Services}}
+from databricks.sdk.service.{{.Package.Name}} import {{.PascalName}}API{{end}}
 
 {{$args := list "host" "account_id" "username" "password" "client_id" "client_secret"
   "token" "profile" "config_file" "azure_workspace_resource_id" "azure_client_secret"
   "azure_client_id" "azure_tenant_id" "azure_environment" "auth_type" "cluster_id"}}
 
+{{- define "api" -}}
+  {{- $mixins := dict "ClustersAPI" "ClustersExt" "DbfsAPI" "DbfsExt" -}}
+  {{- $genApi := concat .PascalName "API" -}}
+  {{- getOrDefault $mixins $genApi $genApi -}}
+{{- end -}}
+
 class WorkspaceClient:
     def __init__(self, *{{range $args}}, {{.}}: str = None{{end}},
                  debug_truncate_bytes: int = None,
@@ -28,7 +35,7 @@ class WorkspaceClient:
         self.dbutils = dbutils.RemoteDbUtils(self.config)
         self.api_client = client.ApiClient(self.config)
         {{- range .Services}}{{if not .IsAccounts}}
-        self.{{.SnakeName}} = {{if eq .SnakeName "dbfs"}}dbfs_mixin.DbfsExt{{else if eq .SnakeName "clusters"}}compute_mixin.ClustersExt{{else}}{{.Package.Name}}.{{.Name}}API{{end}}(self.api_client){{end -}}{{end}}
+        self.{{.SnakeName}} = {{template "api" .}}(self.api_client){{end -}}{{end}}
 
 class AccountClient:
     def __init__(self, *{{range $args}}, {{.}}: str = None{{end}},
@@ -48,4 +55,4 @@ class AccountClient:
         self.config = config
         self.api_client = client.ApiClient(self.config)
         {{- range .Services}}{{if .IsAccounts}}
-        self.{{.SnakeName}} = {{.Package.Name}}.{{.Name}}API(self.api_client){{end -}}{{end}}
+        self.{{(.TrimPrefix "account").SnakeName}} = {{template "api" .}}(self.api_client){{end -}}{{end}}

.gitattributes

@@ -1,27 +1,16 @@
 databricks/sdk/__init__.py linguist-generated=true
 databricks/sdk/service/billing.py linguist-generated=true
-databricks/sdk/service/clusterpolicies.py linguist-generated=true
-databricks/sdk/service/clusters.py linguist-generated=true
-databricks/sdk/service/commands.py linguist-generated=true
-databricks/sdk/service/dbfs.py linguist-generated=true
-databricks/sdk/service/deployment.py linguist-generated=true
-databricks/sdk/service/endpoints.py linguist-generated=true
-databricks/sdk/service/gitcredentials.py linguist-generated=true
-databricks/sdk/service/globalinitscripts.py linguist-generated=true
-databricks/sdk/service/instancepools.py linguist-generated=true
-databricks/sdk/service/ipaccesslists.py linguist-generated=true
+databricks/sdk/service/catalog.py linguist-generated=true
+databricks/sdk/service/compute.py linguist-generated=true
+databricks/sdk/service/files.py linguist-generated=true
+databricks/sdk/service/iam.py linguist-generated=true
 databricks/sdk/service/jobs.py linguist-generated=true
-databricks/sdk/service/libraries.py linguist-generated=true
-databricks/sdk/service/mlflow.py linguist-generated=true
+databricks/sdk/service/ml.py linguist-generated=true
 databricks/sdk/service/oauth2.py linguist-generated=true
-databricks/sdk/service/permissions.py linguist-generated=true
 databricks/sdk/service/pipelines.py linguist-generated=true
-databricks/sdk/service/repos.py linguist-generated=true
-databricks/sdk/service/scim.py linguist-generated=true
-databricks/sdk/service/secrets.py linguist-generated=true
+databricks/sdk/service/provisioning.py linguist-generated=true
+databricks/sdk/service/serving.py linguist-generated=true
+databricks/sdk/service/settings.py linguist-generated=true
+databricks/sdk/service/sharing.py linguist-generated=true
 databricks/sdk/service/sql.py linguist-generated=true
-databricks/sdk/service/tokenmanagement.py linguist-generated=true
-databricks/sdk/service/tokens.py linguist-generated=true
-databricks/sdk/service/unitycatalog.py linguist-generated=true
 databricks/sdk/service/workspace.py linguist-generated=true
-databricks/sdk/service/workspaceconf.py linguist-generated=true