ids -> backend/types

Author: varun-edachali-dbx

src/databricks/sql/backend/databricks_client.py

@@ -2,7 +2,7 @@
 from typing import Dict, Tuple, List, Optional, Any, Union
 
 from databricks.sql.thrift_api.TCLIService import ttypes
-from databricks.sql.ids import SessionId, CommandId
+from databricks.sql.backend.types import SessionId, CommandId
 from databricks.sql.utils import ExecuteResponse
 from databricks.sql.types import SSLOptions
 

src/databricks/sql/backend/thrift_backend.py

@@ -8,7 +8,7 @@
 from typing import List, Union, Any
 
 from databricks.sql.thrift_api.TCLIService.ttypes import TOperationState
-from databricks.sql.ids import SessionId, CommandId, BackendType, guid_to_hex_id
+from databricks.sql.backend.types import SessionId, CommandId, BackendType, guid_to_hex_id
 
 try:
     import pyarrow

src/databricks/sql/ids.py renamed to src/databricks/sql/backend/types.py

@@ -47,7 +47,7 @@
 from databricks.sql.auth.auth import get_python_sql_connector_auth_provider
 from databricks.sql.experimental.oauth_persistence import OAuthPersistence
 from databricks.sql.session import Session
-from databricks.sql.ids import CommandId, BackendType
+from databricks.sql.backend.types import CommandId, BackendType
 
 from databricks.sql.thrift_api.TCLIService.ttypes import (
     TSparkParameter,

src/databricks/sql/client.py

@@ -9,7 +9,7 @@
 from databricks.sql import USER_AGENT_NAME
 from databricks.sql.backend.thrift_backend import ThriftDatabricksClient
 from databricks.sql.backend.databricks_client import DatabricksClient
-from databricks.sql.ids import SessionId, BackendType
+from databricks.sql.backend.types import SessionId, BackendType
 
 logger = logging.getLogger(__name__)
 

src/databricks/sql/session.py

@@ -26,7 +26,7 @@
     TSparkRowSetType,
 )
 from databricks.sql.types import SSLOptions
-from databricks.sql.ids import CommandId
+from databricks.sql.backend.types import CommandId
 
 from databricks.sql.parameters.native import ParameterStructure, TDbsqlParameter
 

src/databricks/sql/utils.py

@@ -7,7 +7,7 @@
     TSessionHandle,
     THandleIdentifier,
 )
-from databricks.sql.ids import SessionId, BackendType
+from databricks.sql.backend.types import SessionId, BackendType
 
 import databricks.sql
 

tests/unit/test_session.py

@@ -18,7 +18,7 @@
 from databricks.sql import *
 from databricks.sql.auth.authenticators import AuthProvider
 from databricks.sql.backend.thrift_backend import ThriftDatabricksClient
-from databricks.sql.ids import CommandId, SessionId, BackendType
+from databricks.sql.backend.types import CommandId, SessionId, BackendType
 
 
 def retry_policy_factory():