소스 검색

Enable to specify driver

ogert 3 년 전
부모
커밋
0cb59ef8e5
1개의 변경된 파일11개의 추가작업 그리고 1개의 파일을 삭제
  1. 11 1
      cdplib/db_handlers/SQLHandler.py

+ 11 - 1
cdplib/db_handlers/SQLHandler.py

@@ -65,7 +65,17 @@ class SQLHandler:
 
             from libraries.configuration import default as cfg
 
-            db_uri = "{0}://{1}:{2}@{3}:{4}/{5}?charset=utf8&local_infile=1"\
+            if cfg["SQL"]["DRIVER"]:
+                db_uri = "{0}://{1}:{2}@{3}:{4}/{5}?driver={}&charset=utf8&local_infile=1"\
+                     .format(cfg["SQL"]["SQL_DIALECT"],
+                             cfg["SQL"]["SQL_USER"],
+                             cfg["SQL"]["SQL_PASSWORD"],
+                             cfg["SQL"]["SQL_HOST"],
+                             cfg["SQL"]["SQL_PORT"],
+                             cfg["SQL"]["SQL_DATABASE_NAME"],
+                             cfg["SQL"]["DRIVER"])
+            else:
+                db_uri = "{0}://{1}:{2}@{3}:{4}/{5}?charset=utf8&local_infile=1"\
                      .format(cfg["SQL"]["SQL_DIALECT"],
                              cfg["SQL"]["SQL_USER"],
                              cfg["SQL"]["SQL_PASSWORD"],