From c22f238b6a1fb1a328adf99bd3a3873070c6702e Mon Sep 17 00:00:00 2001
From: R_H <118392077+rhgit01@users.noreply.github.com>
Date: Thu, 18 Jul 2024 09:49:21 +0200
Subject: [PATCH] Assign OS390Reflector for Db2 for z/OS

When connecting to a DBMS starting with "DSN" (Db2 for z/OS), assign the OS390Reflector.
---
 ibm_db_sa/base.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ibm_db_sa/base.py b/ibm_db_sa/base.py
index 54b2401..bcdb09c 100644
--- a/ibm_db_sa/base.py
+++ b/ibm_db_sa/base.py
@@ -733,6 +733,8 @@ def initialize(self, connection):
             _reflector_cls = ibm_reflection.DB2Reflector
         elif "IDS/" in self.dbms_name:
             _reflector_cls = ibm_reflection.DB2Reflector
+        elif self.dbms_name.startswith("DSN"):
+            _reflector_cls = ibm_reflection.OS390Reflector            
         self._reflector = _reflector_cls(self)
 
     def get_columns(self, connection, table_name, schema=None, **kw):