test

eda/smart_rank/dist_update_user_portrait_service.py

@@ -78,36 +78,36 @@ def get_user_service_portrait(cl_id, all_word_tags, all_tag_tag_type, all_3tag_2
             gmkv_tag_score2_sum = tag_score_sum[["tag2", "tag_score"]][:size].to_dict('record')
             gmkv_tag_score2_sum_dict = {i["tag2"]: i["tag_score"] for i in gmkv_tag_score2_sum}
             # 写gmkv
-            # gm_kv_cli = redis.Redis(host="172.16.40.135", port=5379, db=2, socket_timeout=2000)
-            # cl_id_portrait_key = "user:service_portrait_tags:cl_id:" + str(cl_id)
-            # tag_id_list_json = json.dumps(gmkv_tag_score_sum)
-            # gm_kv_cli.set(cl_id_portrait_key, tag_id_list_json)
-            # gm_kv_cli.expire(cl_id_portrait_key, time=30 * 24 * 60 * 60)
-
-            # 写redis
-            # redis_client = redis.StrictRedis.from_url('redis://:ReDis!GmTx*0aN9@172.16.40.173:6379')
-            # cl_id_portrait_key2 = "user:service_portrait_tags2:cl_id:" + str(cl_id)
-            # # 如果画像的tag个数小于5，则补充热搜词
-            # if len(gmkv_tag_score2_sum_dict) < 5:
-            #     hot_search_wordskey2 = "user:service_coldstart_tags2"
-            #     hot_search_words_score = redis_client.hgetall(hot_search_wordskey2)
-            #     for tag_id in hot_search_words_score:
-            #         if int(tag_id) in gmkv_tag_score2_sum_dict:
-            #             continue
-            #         else:
-            #             gmkv_tag_score2_sum_dict.update({tag_id: hot_search_words[tag_id]})
-            #         if len(gmkv_tag_score2_sum_dict) > 4:
-            #             break
-            # redis_client.delete(cl_id_portrait_key2)
-            # redis_client.hmset(cl_id_portrait_key2, gmkv_tag_score2_sum_dict)
-            # redis_client.expire(cl_id_portrait_key2, time=30 * 24 * 60 * 60)
-            #
-            # # 标签name写redis
-            # cl_id_portrait_key3 = "user:service_portrait_tags3:cl_id:" + str(cl_id)
-            # gmkv_tag_score3_sum_dict = {all_tags_name[i]: gmkv_tag_score2_sum_dict[i] for i in gmkv_tag_score2_sum_dict}
-            # redis_client.delete(cl_id_portrait_key3)
-            # redis_client.hmset(cl_id_portrait_key3, gmkv_tag_score3_sum_dict)
-            # redis_client.expire(cl_id_portrait_key3, time=30 * 24 * 60 * 60)
+            gm_kv_cli = redis.Redis(host="172.16.40.135", port=5379, db=2, socket_timeout=2000)
+            cl_id_portrait_key = "user:service_portrait_tags:cl_id:" + str(cl_id)
+            tag_id_list_json = json.dumps(gmkv_tag_score_sum)
+            gm_kv_cli.set(cl_id_portrait_key, tag_id_list_json)
+            gm_kv_cli.expire(cl_id_portrait_key, time=30 * 24 * 60 * 60)
+
+            写redis
+            redis_client = redis.StrictRedis.from_url('redis://:ReDis!GmTx*0aN9@172.16.40.173:6379')
+            cl_id_portrait_key2 = "user:service_portrait_tags2:cl_id:" + str(cl_id)
+            # 如果画像的tag个数小于5，则补充热搜词
+            if len(gmkv_tag_score2_sum_dict) < 5:
+                hot_search_wordskey2 = "user:service_coldstart_tags2"
+                hot_search_words_score = redis_client.hgetall(hot_search_wordskey2)
+                for tag_id in hot_search_words_score:
+                    if int(tag_id) in gmkv_tag_score2_sum_dict:
+                        continue
+                    else:
+                        gmkv_tag_score2_sum_dict.update({tag_id: hot_search_words[tag_id]})
+                    if len(gmkv_tag_score2_sum_dict) > 4:
+                        break
+            redis_client.delete(cl_id_portrait_key2)
+            redis_client.hmset(cl_id_portrait_key2, gmkv_tag_score2_sum_dict)
+            redis_client.expire(cl_id_portrait_key2, time=30 * 24 * 60 * 60)
+
+            # 标签name写redis
+            cl_id_portrait_key3 = "user:service_portrait_tags3:cl_id:" + str(cl_id)
+            gmkv_tag_score3_sum_dict = {all_tags_name[i]: gmkv_tag_score2_sum_dict[i] for i in gmkv_tag_score2_sum_dict}
+            redis_client.delete(cl_id_portrait_key3)
+            redis_client.hmset(cl_id_portrait_key3, gmkv_tag_score3_sum_dict)
+            redis_client.expire(cl_id_portrait_key3, time=30 * 24 * 60 * 60)
 
             # 写tidb，redis同步
             stat_date = datetime.datetime.today().strftime('%Y-%m-%d')