generate label for mysql

process_data.py

@@ -229,7 +229,7 @@ class UserProcess(object):
                 except:
                     pass
 
-    def generate_user_data(self):
+    def generate_user_data(self, flag, dt_time=None):
         """
         生成user训练数据
         :return:
@@ -272,6 +272,14 @@ if __name__ == "__main__":
             S.spider_data_produce(flag=mode, dt_time=dt)
         case "user":
             U = UserProcess()
+            if mode == "generate":
+                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("train")
+            else:
+                print("Error")
         case "Data":
             D = DataProcessor()
 

test_script.py

@@ -0,0 +1,9 @@
+from read_data_from_odps import VideoDataGenerator
+
+
+P = VideoDataGenerator()
+
+r = P.get_daily_user_info(dt="20240228")
+print(r)
+for i in r[:10]:
+    print(i)