Skip to content

Commit f82658a

Browse files
make _parse methods in models internal
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 1edc80a commit f82658a

File tree

2 files changed

+10
-12
lines changed

2 files changed

+10
-12
lines changed

src/databricks/sql/backend/sea/backend.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,7 @@
4141
GetStatementResponse,
4242
CreateSessionResponse,
4343
)
44-
from databricks.sql.backend.sea.models.responses import (
45-
GetChunksResponse,
46-
)
44+
from databricks.sql.backend.sea.models.responses import GetChunksResponse
4745

4846
logger = logging.getLogger(__name__)
4947

src/databricks/sql/backend/sea/models/responses.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
)
1919

2020

21-
def parse_status(data: Dict[str, Any]) -> StatementStatus:
21+
def _parse_status(data: Dict[str, Any]) -> StatementStatus:
2222
"""Parse status from response data."""
2323
status_data = data.get("status", {})
2424
error = None
@@ -40,7 +40,7 @@ def parse_status(data: Dict[str, Any]) -> StatementStatus:
4040
)
4141

4242

43-
def parse_manifest(data: Dict[str, Any]) -> ResultManifest:
43+
def _parse_manifest(data: Dict[str, Any]) -> ResultManifest:
4444
"""Parse manifest from response data."""
4545

4646
manifest_data = data.get("manifest", {})
@@ -69,7 +69,7 @@ def parse_manifest(data: Dict[str, Any]) -> ResultManifest:
6969
)
7070

7171

72-
def parse_result(data: Dict[str, Any]) -> ResultData:
72+
def _parse_result(data: Dict[str, Any]) -> ResultData:
7373
"""Parse result data from response data."""
7474
result_data = data.get("result", {})
7575
external_links = None
@@ -118,9 +118,9 @@ def from_dict(cls, data: Dict[str, Any]) -> "ExecuteStatementResponse":
118118
"""Create an ExecuteStatementResponse from a dictionary."""
119119
return cls(
120120
statement_id=data.get("statement_id", ""),
121-
status=parse_status(data),
122-
manifest=parse_manifest(data),
123-
result=parse_result(data),
121+
status=_parse_status(data),
122+
manifest=_parse_manifest(data),
123+
result=_parse_result(data),
124124
)
125125

126126

@@ -138,9 +138,9 @@ def from_dict(cls, data: Dict[str, Any]) -> "GetStatementResponse":
138138
"""Create a GetStatementResponse from a dictionary."""
139139
return cls(
140140
statement_id=data.get("statement_id", ""),
141-
status=parse_status(data),
142-
manifest=parse_manifest(data),
143-
result=parse_result(data),
141+
status=_parse_status(data),
142+
manifest=_parse_manifest(data),
143+
result=_parse_result(data),
144144
)
145145

146146

0 commit comments

Comments
 (0)