working uuid

Author: gargsaumya

mssql_python/cursor.py

@@ -465,6 +465,15 @@ def _map_sql_type(self, param, parameters_list, i, min_val=None, max_val=None):
                     0,
                     False
                 )
+        
+        if isinstance(param, uuid.UUID):
+            return (
+                ddbc_sql_const.SQL_GUID.value,
+                ddbc_sql_const.SQL_C_GUID.value,
+                16,
+                0,
+                False,
+            )
 
         if isinstance(param, datetime.datetime):
             if param.tzinfo is not None:
@@ -984,6 +993,8 @@ def execute(
 
         if parameters:
             for i, param in enumerate(parameters):
+                if isinstance(param, uuid.UUID):
+                    parameters[i] = param.bytes
                 paraminfo = self._create_parameter_types_list(
                     param, param_info, parameters, i
                 )

mssql_python/pybind/ddbc_bindings.cpp

@@ -2713,7 +2713,7 @@ SQLRETURN SQLGetData_wrap(SqlHandlePtr StatementHandle, SQLUSMALLINT colCount, p
                 }
                 break;
             }
-#endif
+        #endif
             default:
                 std::ostringstream errorString;
                 errorString << "Unsupported data type for column - " << columnName << ", Type - "

tests/test_004_cursor.py

@@ -7146,6 +7146,62 @@ def test_extreme_uuids(cursor, db_connection):
         cursor.execute(f"DROP TABLE IF EXISTS {table_name}")
         db_connection.commit()
 
+def test_executemany_uuid_insert_and_select(cursor, db_connection):
+    """Test inserting multiple UUIDs using executemany and verifying retrieval."""
+    table_name = "#pytest_uuid_executemany"
+    
+    try:
+        # Drop and create a temporary table for the test
+        cursor.execute(f"DROP TABLE IF EXISTS {table_name}")
+        cursor.execute(f"""
+            CREATE TABLE {table_name} (
+                id UNIQUEIDENTIFIER PRIMARY KEY,
+                description NVARCHAR(50)
+            )
+        """)
+        db_connection.commit()
+
+        # Generate data for insertion
+        data_to_insert = []
+        uuids_to_check = {}
+        for i in range(5):
+            new_uuid = uuid.uuid4()
+            description = f"Item {i}"
+            data_to_insert.append((new_uuid, description))
+            uuids_to_check[description] = new_uuid
+
+        # Insert all data with a single call to executemany
+        sql = f"INSERT INTO {table_name} (id, description) VALUES (?, ?)"
+        cursor.executemany(sql, data_to_insert)
+        db_connection.commit()
+
+        # Verify the number of rows inserted
+        assert cursor.rowcount == 5, f"Expected 5 rows inserted, but got {cursor.rowcount}"
+
+        # Fetch all data from the table
+        cursor.execute(f"SELECT id, description FROM {table_name}")
+        rows = cursor.fetchall()
+        
+        # Verify the number of fetched rows
+        assert len(rows) == len(data_to_insert), "Number of fetched rows does not match."
+        
+        # Verify each fetched row's data and type
+        for row in rows:
+            retrieved_uuid, retrieved_desc = row
+            
+            # Assert the type is correct
+            assert isinstance(retrieved_uuid, uuid.UUID), f"Expected uuid.UUID, got {type(retrieved_uuid)}"
+            
+            # Assert the value matches the original data
+            expected_uuid = uuids_to_check.get(retrieved_desc)
+            assert expected_uuid is not None, f"Retrieved description '{retrieved_desc}' was not in the original data."
+            assert retrieved_uuid == expected_uuid, f"UUID mismatch for '{retrieved_desc}': expected {expected_uuid}, got {retrieved_uuid}"
+
+    finally:
+        # Clean up the temporary table
+        cursor.execute(f"DROP TABLE IF EXISTS {table_name}")
+        db_connection.commit()
+
 def test_decimal_separator_with_multiple_values(cursor, db_connection):
     """Test decimal separator with multiple different decimal values"""
     original_separator = mssql_python.getDecimalSeparator()
@@ -10560,6 +10616,7 @@ def test_decimal_separator_calculations(cursor, db_connection):
 
         # Cleanup
         cursor.execute("DROP TABLE IF EXISTS #pytest_decimal_calc_test")
+        db_connection.commit()
 
 def test_close(db_connection):
     """Test closing the cursor"""