liqian 2 年之前
父节点
当前提交
87b9945312
共有 2 个文件被更改,包括 8 次插入6 次删除
  1. 2 2
      config.py
  2. 6 4
      user_group_update.py

+ 2 - 2
config.py

@@ -926,8 +926,8 @@ class ProductionConfig(BaseConfig):
 
 def set_config():
     # 获取环境变量 ROV_OFFLINE_ENV
-    env = os.environ.get('ROV_OFFLINE_ENV')
-    # env = 'dev'
+    # env = os.environ.get('ROV_OFFLINE_ENV')
+    env = 'dev'
     if env is None:
         # log_.error('ENV ERROR: is None!')
         return

+ 6 - 4
user_group_update.py

@@ -8,7 +8,7 @@ config_, _ = set_config()
 log_ = Log()
 redis_helper = RedisHelper()
 
-mid_group_list = [
+features = [
     'apptype',
     'return1mids',
     'return2_3mids',
@@ -23,12 +23,14 @@ mid_group_list = [
 def update_user_group_to_redis(project, table, dt, app_type):
     """更新mid对应分组到redis中"""
     # 获取用户分组数据
-    feature_df = get_feature_data(project=project, table=table, features=mid_group_list, dt=dt)
+    feature_df = get_feature_data(project=project, table=table, features=features, dt=dt)
     feature_df['apptype'] = feature_df['apptype'].astype(int)
     feature_df = feature_df[feature_df['apptype'] == app_type]
-    for group in mid_group_list:
+    print(feature_df)
+    for group in features[1:]:
         log_.info(f"group = {group} update redis start ...")
         mid_list = feature_df[group].tolist()
+        print(mid_list)
         for mid in mid_list:
             key_name = f"{config_.KEY_NAME_PREFIX_MID_GROUP}{mid}"
             redis_helper.set_data_to_redis(key_name=key_name, value=group, expire_time=25 * 3600)
@@ -40,7 +42,7 @@ def timer_check():
         app_type = config_.APP_TYPE['VLOG']
         project = config_.ad_model_data['user_group'].get('project')
         table = config_.ad_model_data['user_group'].get('table')
-        now_date = datetime.datetime.today()
+        now_date = datetime.datetime.today() - datetime.timedelta(days=2)
         dt = datetime.datetime.strftime(now_date, '%Y%m%d')
         log_.info(f"now_date: {dt}")
         now_min = datetime.datetime.now().minute