修改测试文件

user_data.py

@@ -125,23 +125,29 @@ def meigou_tag_to_csv():
     df.to_csv('/tmp/device/meigou_tag.csv', index=False)
 
 
+def hospital_to_csv():
+    sql = "select id,name,city_id,baidu_loc_lng,baidu_loc_lat from api_hospital"
+    df = pd.DataFrame(list(get_mysql_data(host, port, user, passwd, db, sql)))
+    df = df.rename(columns={0: "id", 1: "name",2:"city_id",3:"lng",4:"lat"})
+    df.to_csv('/tmp/device/hospital.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')
-    #
-    # print("end")
-
-    host = "172.16.40.170"
-    port = 4000
-    user = "root"
-    passwd = "3SYz54LS9#^9sBvC"
-    db = "jerry_test"
-
-    meigou_tag_to_csv()
+    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')
+    hospital_to_csv()
+    print("end")
+
+    # host = "172.16.40.170"
+    # port = 4000
+    # user = "root"
+    # passwd = "3SYz54LS9#^9sBvC"
+    # db = "jerry_test"
+