李莎 преди 2 години
родител
ревизия
e10767d69d
променени са 1 файла, в които са добавени 23 реда и са изтрити 7 реда
  1. 23 7
      main.py

+ 23 - 7
main.py

@@ -1,5 +1,6 @@
 import datetime
 import datetime
 import json
 import json
+import os
 from MyUtils.ZillionUtil import ZillionUtil
 from MyUtils.ZillionUtil import ZillionUtil
 from MyUtils.MysqlUtils import MysqlUtils
 from MyUtils.MysqlUtils import MysqlUtils
 from dateutil.relativedelta import relativedelta
 from dateutil.relativedelta import relativedelta
@@ -34,13 +35,28 @@ def insert_mysql(sqls,building,my_table):
     print("%s,mysql数据%s,项目%s插入成功,合计%s条..." % (datetime_now(),my_table,building,len(sqls)))
     print("%s,mysql数据%s,项目%s插入成功,合计%s条..." % (datetime_now(),my_table,building,len(sqls)))
 
 
 
 
-with open("config.json", "r") as f:
-    data = json.load(f)
-    hbase_database = data["metadata"]["database"]
-    url = data["metadata"]["url"]
-    building = data["building"]
-    mysql = data["mysql"]
-    my_database = mysql["database"]
+# with open("config.json", "r") as f:
+#     data = json.load(f)
+#     hbase_database = data["metadata"]["database"]
+#     url = data["metadata"]["url"]
+#     building = data["building"]
+#     mysql = data["mysql"]
+#     my_database = mysql["database"]
+
+
+hbase_database = os.getenv("hbase_database")
+url = os.getenv("url")
+building = os.getenv("building")
+mysql = {
+    "host": os.getenv("host"),
+    "port": os.getenv("port"),
+    "user": os.getenv("user"),
+    "passwd": os.getenv("passwd"),
+    "database": os.getenv("database")
+}
+
+
+
 
 
 
 
 now_month = datetime.date.today().strftime("%Y%m")
 now_month = datetime.date.today().strftime("%Y%m")