Update connection_manager.py

support kingbase database
MarketSquare · Dec 11, 2020 · afddd4b · afddd4b
1 parent ea0b3d4
commit afddd4b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/DatabaseLibrary/connection_manager.py b/src/DatabaseLibrary/connection_manager.py
@@ -139,6 +139,12 @@ def connect_to_database(self, dbapiModuleName=None, dbName=None, dbUsername=None
                 host=dbHost,
                 port=dbPort
             )
+        elif dbapiModuleName in ["ksycopg2"]:
+            dbPort = dbPort or 54321
+            logger.info('Connecting using : %s.connect(database=%s, user=%s, password=%s, host=%s, port=%s) ' % (
+            dbapiModuleName, dbName, dbUsername, dbPassword, dbHost, dbPort))
+            self._dbconnection = db_api_2.connect(database=dbName, user=dbUsername, password=dbPassword, host=dbHost,
+                                                  port=dbPort)
         else:
             logger.info('Connecting using : %s.connect(database=%s, user=%s, password=%s, host=%s, port=%s) ' % (dbapiModuleName, dbName, dbUsername, dbPassword, dbHost, dbPort))
             self._dbconnection = db_api_2.connect(database=dbName, user=dbUsername, password=dbPassword, host=dbHost, port=dbPort)