新增测试文件

user_data.py

@@ -19,15 +19,14 @@ def get_user():
     sql = "select user_id,phone,name,city_id,channel,auth_type from api_userextra"
     df = pd.DataFrame(list(get_mysql_data(host,port,user,passwd,db,sql)))
     df = df.rename(columns={0: "user_id", 1: "phone",2:"name",3:"city_id",4:"channel",5:"auth_type"})
-    df.to_sql("user", con=yconnect, if_exists='append',chunksize = 1000,index = False)
-    # l = 0
-    # r = 2000
-    # length = len(df)
-    # while (l < length):
-    #
-    #
-    # pd.io.sql.to_sql(df, "user", yconnect, schema='jerry_test', if_exists='append', index=False,
-    #                  chunksize=1)
+    print(df.shape)
+    l = 0
+    r = 2000
+    length = len(df)
+    while l < length:
+        pd.io.sql.to_sql(df[l:r], "user", yconnect, schema='jerry_test', if_exists='append', index=False)
+        l += 2000
+        r += 2000
     print("insert done")
 
 
@@ -41,7 +40,7 @@ if __name__ == "__main__":
     user = "work"
     passwd = "BJQaT9VzDcuPBqkd"
     db = "zhengxing"
-    yconnect = create_engine('mysql+pymysql://root:3SYz54LS9#^9sBvC@172.16.40.158:4000/jerry_test?charset=utf8')
+    yconnect = create_engine('mysql+pymysql://root:3SYz54LS9#^9sBvC@172.16.40.170:4000/jerry_test?charset=utf8')
     get_user()