iris-pgwire 1.5.4__tar.gz → 1.5.5__tar.gz

{iris_pgwire-1.5.4 → iris_pgwire-1.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iris-pgwire
-Version: 1.5.4
+Version: 1.5.5
 Summary: PostgreSQL Wire Protocol Server for InterSystems IRIS - Connect BI tools, Python frameworks, and PostgreSQL clients to IRIS databases
 Project-URL: Homepage, https://github.com/intersystems-community/iris-pgwire
 Project-URL: Documentation, https://github.com/intersystems-community/iris-pgwire#readme

{iris_pgwire-1.5.4 → iris_pgwire-1.5.5}/src/iris_pgwire/__init__.py

@@ -6,7 +6,7 @@ Based on the specification from docs/iris_pgwire_plan.md and proven patterns fro
 caretdev/sqlalchemy-iris.
 """
 
-__version__ = "1.5.4"
+__version__ = "1.5.5"
 __author__ = "Thomas Dyar <thomas.dyar@intersystems.com>"
 
 # Don't import server/protocol in __init__ to avoid sys.modules conflicts

{iris_pgwire-1.5.4 → iris_pgwire-1.5.5}/src/iris_pgwire/protocol.py

@@ -2930,10 +2930,14 @@ class PGWireProtocol:
                 # For SELECT/SHOW statements, send RowDescription with column metadata
                 # For DDL/DML statements, send NoData
                 query = stmt.get("translated_query", stmt.get("query", ""))
+                # Use original_query for RETURNING detection: the translation pipeline
+                # strips RETURNING before storing translated_query (IRIS doesn't support it
+                # natively), so has_returning would always be False if we use translated_query.
+                original_query = stmt.get("original_query", query)
                 query_upper = query.strip().upper()
 
                 plan = ReturningPlan.from_sql(
-                    query,
+                    original_query,
                     metadata_cache=self.iris_executor.metadata_cache,
                     executor=self.iris_executor,
                 )
@@ -3081,10 +3085,13 @@ class PGWireProtocol:
 
                 stmt = self.prepared_statements[statement_name]
                 query = stmt.get("translated_query", stmt.get("query", ""))
+                # Use original_query for RETURNING detection: translated_query has the
+                # RETURNING clause stripped (IRIS doesn't support it natively).
+                original_query = stmt.get("original_query", query)
                 query_upper = query.strip().upper()
 
                 plan = ReturningPlan.from_sql(
-                    query,
+                    original_query,
                     metadata_cache=self.iris_executor.metadata_cache,
                     executor=self.iris_executor,
                 )