Skip to content

Conversation

jahnvi480
Copy link
Contributor

@jahnvi480 jahnvi480 commented Aug 18, 2025

Work Item / Issue Reference

AB#34893


Summary

This pull request adds comprehensive support for capturing and managing diagnostic messages (such as SQL Server PRINT statements and warnings) in the mssql_python driver's Cursor class, following the DBAPI specification. It introduces a new messages attribute on the cursor, ensures messages are cleared or preserved at the correct times, and provides robust testing for these behaviors. Additionally, it implements the underlying C++ binding for retrieving all diagnostic records from the ODBC driver.

Diagnostic message handling improvements:

  • Added a messages attribute to the Cursor class to store diagnostic messages, and ensured it is cleared before each non-fetch operation (e.g., execute, executemany, close, commit, rollback, scroll, and nextset) to comply with DBAPI expectations. (mssql_python/cursor.py)
  • After each statement execution and fetch operation, diagnostic messages (including informational and warning messages) are collected and appended to the messages list, using a new C++ binding. (mssql_python/cursor.py, mssql_python/pybind/ddbc_bindings.cpp)

Native driver and binding enhancements:

  • Implemented the SQLGetAllDiagRecords function in the C++ pybind layer to retrieve all diagnostic records from an ODBC statement handle, handling both Windows and Unix platforms, and exposed it as DDBCSQLGetAllDiagRecords to Python. (mssql_python/pybind/ddbc_bindings.cpp)

Testing and specification compliance:

  • Added a comprehensive test suite to verify message capturing, clearing, preservation across fetches, handling of multiple messages, message formatting, warning capture, manual clearing, and error scenarios, ensuring compliance with DBAPI and robust behavior. (tests/test_004_cursor.py)

Other cursor improvements:

  • Refactored the skip method to validate arguments more strictly, clear messages before skipping, and improve error handling and documentation. (mssql_python/cursor.py)

These changes significantly improve the usability and correctness of message handling in the driver, making it easier for users to access and manage SQL Server informational and warning messages in Python applications.

@github-actions github-actions bot added the pr-size: medium Moderate update size label Aug 18, 2025
jahnvi480 and others added 2 commits August 27, 2025 11:20
### Work Item / Issue Reference  
<!-- 
IMPORTANT: Please follow the PR template guidelines below.
For mssql-python maintainers: Insert your ADO Work Item ID below (e.g.
AB#37452)
For external contributors: Insert Github Issue number below (e.g. #149)
Only one reference is required - either GitHub issue OR ADO Work Item.
-->

<!-- mssql-python maintainers: ADO Work Item -->
>
[AB#34926](https://sqlclientdrivers.visualstudio.com/c6d89619-62de-46a0-8b46-70b92a84d85e/_workitems/edit/34926)

-------------------------------------------------------------------
### Summary   
This pull request adds a new `tables()` method to the `Cursor` class in
`mssql_python/cursor.py`, providing a way to query metadata about tables
in the database, including support for filtering by name, schema,
catalog, and table type. It also introduces comprehensive test coverage
for this new method in `tests/test_004_cursor.py`. Additionally, the
`skip()` method in the cursor is simplified by delegating to the
existing `scroll()` method.

**New feature: Table metadata querying**
- Added a `tables()` method to the `Cursor` class, enabling users to
retrieve information about tables with support for filtering by table
name (including temporary tables), schema, catalog, and table type
(supports both string and list input). The method returns the cursor
itself for easy chaining and iteration.

**Testing improvements**
- Introduced a suite of tests for the new `tables()` method, covering
basic usage, filtering by name, schema, and type, wildcard support,
combined filters, empty results, iteration, method chaining, and
existence checks. These tests ensure the method works as intended and
handles edge cases.

**Code simplification**
- Refactored the `skip()` method in the cursor to delegate to the
`scroll()` method in 'relative' mode, removing redundant validation and
manual row skipping logic.

---------

Co-authored-by: Jahnvi Thakkar <[email protected]>
@github-actions github-actions bot added pr-size: large Substantial code update and removed pr-size: medium Moderate update size labels Aug 27, 2025
@jahnvi480 jahnvi480 merged commit 3d8304e into jahnvi/cursor_skip Aug 27, 2025
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pr-size: large Substantial code update
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants