write data

src/main_portrait.py

 import datetime
 
 from utils.cache import redis_client2
-from utils.portrait import user_portrait_tag3_get_candidate_unread_list
+from utils.portrait import user_portrait_tag3_get_candidate_unread_list, user_portrait_tag3_write_ctcvr_data
 
 
 def user_portrait_scan_info():
@@ -28,7 +28,7 @@ def user_portrait_scan_info():
 
 
 if __name__ == "__main__":
-    # device_id = "androidid_a25a1129c0b38f7b"
-    # res = user_portrait_tag3_get_candidate_unread_list(device_id, "diary")
-    # print(res)
-    user_portrait_scan_info()
+    device_id = "androidid_a25a1129c0b38f7b"
+    a = [15420031, 15531030, 16049931, 15335637, 16443431, 16269779, 9389022, 16710435, 16258498, 16435592]
+    user_portrait_tag3_write_ctcvr_data(device_id, "diary", a)
+    # user_portrait_scan_info()

src/utils/portrait.py

@@ -69,6 +69,15 @@ def remove_read_with_size(lol, content_type, read_list, size=500):
     return res
 
 
+def _user_portrait_tag3_get_ctcvr_candidate_key(device_id, content_type):
+    """
+    device_id:
+    content_type: diary
+    return:
+    """
+    return "doris:user_portrait:tag3:ctcvr:candidate:{}:device_id:".format(content_type) + str(device_id)
+
+
 def _user_portrait_tag3_get_candidate_key(device_id, content_type, latest=False):
     """
     content_type: diary, tractate, answer
@@ -122,3 +131,17 @@ def user_portrait_tag3_get_candidate_unread_list(device_id, content_type):
         lol = remove_read_with_size(diary_dict.values(), "diary", read_list, 500)
         queue = merge_lists(*lol)
     return queue
+
+
+def user_portrait_tag3_write_ctcvr_data(device_id, content_type, data):
+    """
+    device_id:
+    content_type: diary
+    data: [15420031, 15531030, 16049931, 15335637, 16443431]
+    return:
+    """
+    if data:
+        key = _user_portrait_tag3_get_ctcvr_candidate_key(device_id, content_type)
+        redis_client4.delete(key)
+        redis_client4.rpush(key, *data)
+        redis_client4.expire(key, 60 * 60 * 24 * 30)