replace getters with property tag

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

src/databricks/sql/backend/types.py

@@ -139,7 +139,7 @@ def __str__(self) -> str:
                 if isinstance(self.secret, bytes)
                 else str(self.secret)
             )
-            return f"{self.get_hex_guid()}|{secret_hex}"
+            return f"{self.hex_guid}|{secret_hex}"
         return str(self.guid)
 
     @classmethod
@@ -217,14 +217,8 @@ def to_sea_session_id(self):
 
         return self.guid
 
-    def get_guid(self) -> Any:
-        """
-        Get the ID of the session.
-        """
-
-        return self.guid
-
-    def get_hex_guid(self) -> str:
+    @property
+    def hex_guid(self) -> str:
         """
         Get a hexadecimal string representation of the session ID.
 
@@ -237,7 +231,8 @@ def get_hex_guid(self) -> str:
         else:
             return str(self.guid)
 
-    def get_protocol_version(self):
+    @property
+    def protocol_version(self):
         """
         Get the server protocol version for this session.
 

src/databricks/sql/client.py

@@ -304,11 +304,11 @@ def __del__(self):
 
     def get_session_id(self):
         """Get the raw session ID (backend-specific)"""
-        return self.session.get_id()
+        return self.session.guid
 
     def get_session_id_hex(self):
         """Get the session ID in hex format"""
-        return self.session.get_id_hex()
+        return self.session.guid_hex
 
     @staticmethod
     def server_parameterized_queries_enabled(protocolVersion):
@@ -784,7 +784,7 @@ def execute(
         self._close_and_clear_active_result_set()
         self.active_result_set = self.backend.execute_command(
             operation=prepared_operation,
-            session_id=self.connection.session.get_session_id(),
+            session_id=self.connection.session.session_id,
             max_rows=self.arraysize,
             max_bytes=self.buffer_size_bytes,
             lz4_compression=self.connection.lz4_compression,
@@ -840,7 +840,7 @@ def execute_async(
         self._close_and_clear_active_result_set()
         self.backend.execute_command(
             operation=prepared_operation,
-            session_id=self.connection.session.get_session_id(),
+            session_id=self.connection.session.session_id,
             max_rows=self.arraysize,
             max_bytes=self.buffer_size_bytes,
             lz4_compression=self.connection.lz4_compression,
@@ -927,7 +927,7 @@ def catalogs(self) -> "Cursor":
         self._check_not_closed()
         self._close_and_clear_active_result_set()
         self.active_result_set = self.backend.get_catalogs(
-            session_id=self.connection.session.get_session_id(),
+            session_id=self.connection.session.session_id,
             max_rows=self.arraysize,
             max_bytes=self.buffer_size_bytes,
             cursor=self,
@@ -946,7 +946,7 @@ def schemas(
         self._check_not_closed()
         self._close_and_clear_active_result_set()
         self.active_result_set = self.backend.get_schemas(
-            session_id=self.connection.session.get_session_id(),
+            session_id=self.connection.session.session_id,
             max_rows=self.arraysize,
             max_bytes=self.buffer_size_bytes,
             cursor=self,
@@ -972,7 +972,7 @@ def tables(
         self._close_and_clear_active_result_set()
 
         self.active_result_set = self.backend.get_tables(
-            session_id=self.connection.session.get_session_id(),
+            session_id=self.connection.session.session_id,
             max_rows=self.arraysize,
             max_bytes=self.buffer_size_bytes,
             cursor=self,
@@ -1000,7 +1000,7 @@ def columns(
         self._close_and_clear_active_result_set()
 
         self.active_result_set = self.backend.get_columns(
-            session_id=self.connection.session.get_session_id(),
+            session_id=self.connection.session.session_id,
             max_rows=self.arraysize,
             max_bytes=self.buffer_size_bytes,
             cursor=self,

src/databricks/sql/session.py

@@ -95,11 +95,11 @@ def open(self):
         )
         self.protocol_version = self.get_protocol_version(self._session_id)
         self.is_open = True
-        logger.info("Successfully opened session %s", str(self.get_id_hex()))
+        logger.info("Successfully opened session %s", str(self.guid_hex))
 
     @staticmethod
     def get_protocol_version(session_id: SessionId):
-        return session_id.get_protocol_version()
+        return session_id.protocol_version
 
     @staticmethod
     def server_parameterized_queries_enabled(protocolVersion):
@@ -111,21 +111,24 @@ def server_parameterized_queries_enabled(protocolVersion):
         else:
             return False
 
-    def get_session_id(self) -> SessionId:
+    @property
+    def session_id(self) -> SessionId:
         """Get the normalized session ID"""
         return self._session_id
 
-    def get_id(self):
+    @property
+    def guid(self) -> Any:
         """Get the raw session ID (backend-specific)"""
-        return self._session_id.get_guid()
+        return self._session_id.guid
 
-    def get_id_hex(self) -> str:
+    @property
+    def guid_hex(self) -> str:
         """Get the session ID in hex format"""
-        return self._session_id.get_hex_guid()
+        return self._session_id.hex_guid
 
     def close(self) -> None:
         """Close the underlying session."""
-        logger.info("Closing session %s", self.get_id_hex())
+        logger.info("Closing session %s", self.guid_hex)
         if not self.is_open:
             logger.debug("Session appears to have been closed already")
             return