数据库自动切换

tables.py

@@ -21,73 +21,43 @@ json_file = os.path.join(relative_path, "config.json")
 # json_file = "config.json"
 
 with open(json_file) as f:
-    mysql_config = json.load(f)["mysql"]
+    load_vale = json.load(f)
 
-with open(json_file) as f:
-    postgre_config = json.load(f)["postgresql"]
+    postgre_config = load_vale["postgresql"]
 
-# logger = get_logger("zxt.tables")
+    if "mysql" in load_vale:
+        mysql_config = load_vale["mysql"]
+    else:
+        mysql_config = None
 
-# # 创建对象的基类:
-# Base = declarative_base()
-#
-# engine_mysql = create_engine(
-#     "mysql+mysqlconnector://root:%s@192.168.28.111:3306/waytous"
-#     % quote("Huituo@123")
-# )
-#
-# engine_postgre = create_engine(
-#     "postgresql://postgres:%s@192.168.28.111:5432/shenbao_2021520"
-#     % quote("Huituo@123")
-# )
-
-# # 创建DBsession_mysql类型:
-# DBsession_mysql = sessionmaker(bind=engine_mysql)
-#
-# DBsession_mysql = scoped_session(DBsession_mysql)
-#
-# DBsession_postgre = sessionmaker(bind=engine_postgre)
-#
-# DBsession_postgre = scoped_session(DBsession_postgre)
-#
-# # 创建session_mysql对象:
-# session_mysql = DBsession_mysql()
-#
-# session_mysql.expire_on_commit = False
-#
-# session_postgre = DBsession_postgre()
-#
-# session_postgre.expire_on_commit = False
+    if "mssql" in load_vale:
+        sqlserver_config = load_vale["mssql"]
+    else:
+        sqlserver_config = None
 
 # 创建对象的基类:
 Base = declarative_base()
 
-sql_str = str("mysql+mysqlconnector://" + mysql_config["user"] + ":%s@" + mysql_config["host"] + \
+if mysql_config is not None:
+
+    sql_str = str("mysql+mysqlconnector://" + mysql_config["user"] + ":%s@" + mysql_config["host"] + \
               ":" + mysql_config["port"] + "/" + mysql_config["database"])
 
-postgre_str = str("postgresql://" + postgre_config["user"] + ":%s@" + postgre_config["host"] + \
-                  ":" + postgre_config["port"] + "/" + postgre_config["database"])
+    engine_mysql = create_engine(sql_str % quote(mysql_config["password"]))
 
-# try:
-engine_mysql = create_engine(
-    # "mysql+mysqlconnector://root:%s@192.168.28.111:3306/waytous"
-    # % quote("Huituo@123")
+else:
 
-    # "mysql+mysqlconnector://" + mysql_config["user"] + ":" + mysql_config["password"] + "@" + mysql_config[
-    # "host"] + ":" + mysql_config["port"] + "/" + mysql_config["database"]
+    sql_str = str("mssql+pymssql://" + sqlserver_config["user"] + ":%s@" + sqlserver_config["host"] + \
+                  ":" + sqlserver_config["port"] + "/" + sqlserver_config["database"])
 
-    sql_str % quote(mysql_config["password"])
-)
+    engine_mysql = create_engine(sql_str % quote(sqlserver_config["password"]))
 
-engine_postgre = create_engine(
-    # "postgresql://postgres:%s@192.168.28.111:5432/shenbao_2021520"
-    # % quote("Huituo@123")
+postgre_str = str("postgresql://" + postgre_config["user"] + ":%s@" + postgre_config["host"] + \
+                  ":" + postgre_config["port"] + "/" + postgre_config["database"])
 
-    # "postgresql://" + postgre_config["user"] + ":" + postgre_config["password"] + "@" + postgre_config[
-    #     "host"] + ":" + postgre_config["port"] + "/" + postgre_config["database"]
+# try:
 
-    postgre_str % quote(postgre_config["password"])
-)
+engine_postgre = create_engine(postgre_str % quote(postgre_config["password"]))
 
 # 创建DBsession_mysql类型:
 DBsession_mysql = sessionmaker(bind=engine_mysql)