Skip to content

Commit

Permalink
add support for MSI
Browse files Browse the repository at this point in the history
  • Loading branch information
Ian Seabock (Centific Technologies Inc) committed Jul 5, 2024
1 parent d32fa92 commit 517237c
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions backend/settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -620,20 +620,28 @@ class _AzureSqlServerSettings(BaseSettings, DatasourcePayloadConstructor):
)
_type: Literal["azure_sql_server"] = PrivateAttr(default="azure_sql_server")

connection_string: str = Field(exclude=True)
connection_string: Optional[str] = None
table_schema: Optional[str] = None
schema_max_row: Optional[int] = None
top_n_results: Optional[int] = None
database_server: Optional[str] = None
database_name: Optional[str] = None
port: Optional[int] = None

# Constructed fields
authentication: Optional[dict] = None

@model_validator(mode="after")
def construct_authentication(self) -> Self:
self.authentication = {
"type": "connection_string",
"connection_string": self.connection_string
}
if self.connection_string:
self.authentication = {
"type": "connection_string",
"connection_string": self.connection_string
}
elif self.database_server and self.database_name and self.port:
self.authentication = {
"type": "system_assigned_managed_identity"
}
return self

def construct_payload_configuration(
Expand Down

0 comments on commit 517237c

Please sign in to comment.