generate label for mysql

process_data.py

@@ -290,10 +290,10 @@ if __name__ == "__main__":
                 sd = str(input("输入开始日期，格式为 YYYYmmdd"))
                 ed = str(input("输入结束日期，格式为 YYYYmmdd"))
                 U.userinfo_to_mysql(start_date=sd, end_date=ed)
-            elif mode == "train":
-                U.generate_user_data(flag=mode, dt_time=dt)
             else:
-                print("Error")
+                U.generate_user_data(flag=mode, dt_time=dt)
+            # else:
+            #     print("Error")
         case "Data":
             D = DataProcessor()
             D.producer()