修改测试文件

user_data.py

@@ -118,22 +118,30 @@ def location_to_csv():
     df = df.rename(columns={0: "device_id", 1: "lat", 2: "lng", 3: "partition_date"})
     df.to_csv('/tmp/device/location.csv', index=False)
 
+def meigou_tag_to_csv():
+    sql = "select service_id,tag_id from meigou_tag"
+    df = pd.DataFrame(list(get_mysql_data(host, port, user, passwd, db, sql)))
+    df = df.rename(columns={0: "service_id", 1: "tag_id"})
+    df.to_csv('/tmp/device/meigou_tag.csv', index=False)
+
 
 if __name__ == "__main__":
-    host = "172.16.30.141"
-    port = 3306
-    user = "work"
-    passwd = "BJQaT9VzDcuPBqkd"
-    db = "zhengxing"
-    yconnect = create_engine('mysql+pymysql://root:3SYz54LS9#^9sBvC@172.16.40.170:4000/jerry_test?charset=utf8')
-    get_meigou_tag()
-    print("end")
-
-    # host = "172.16.40.170"
-    # port = 4000
-    # user = "root"
-    # passwd = "3SYz54LS9#^9sBvC"
-    # db = "jerry_test"
+    # host = "172.16.30.141"
+    # port = 3306
+    # user = "work"
+    # passwd = "BJQaT9VzDcuPBqkd"
+    # db = "zhengxing"
+    # yconnect = create_engine('mysql+pymysql://root:3SYz54LS9#^9sBvC@172.16.40.170:4000/jerry_test?charset=utf8')
+    #
+    # print("end")
+
+    host = "172.16.40.170"
+    port = 4000
+    user = "root"
+    passwd = "3SYz54LS9#^9sBvC"
+    db = "jerry_test"
+
+    meigou_tag_to_csv()