try write data

eda/smart_rank/tag3_update_user_portrait_offline.py

@@ -4,17 +4,16 @@ import json
 import operator
 from collections import Counter
 
+import pymysql
 import pytz
 import redis
-
-import pymysql
-from es_tool import es_insert_device_info
 from pyspark import SparkConf
 from pyspark.sql import SparkSession
-from tool import (get_doris_prod, get_jerry_test, get_redis_client,
-    get_tag3_user_log, get_user_portrait_tag3_from_redis,
-    get_user_portrait_tag3_with_score, send_email, write_user_portrait,
-    write_user_portrait_by_event, write_user_portrait_doris)
+
+from es_tool import es_insert_device_info
+from tool import (get_doris_prod, get_jerry_test, get_redis_client, get_tag3_user_log, get_user_portrait_tag3_from_redis,
+                  get_user_portrait_tag3_with_score, send_email, write_user_portrait, write_user_portrait_by_event,
+                  write_user_portrait_doris)
 
 
 # [{'激光': 1.949194898204873}, {'手术': 1.949194898204873}, {'手术': 1.949194898204873}, {'手术': 1.949194898204873, '植发际线': 7.1}]
@@ -268,15 +267,13 @@ def update_tag3_user_portrait(cl_id):
            (len(first_solutions_score.keys()) > 0) or (len(second_solutions_score.keys()) > 0) or \
            (len(first_positions_score.keys()) > 0) or (len(second_positions_score.keys()) > 0) or \
            (len(projects_score.keys()) > 0):
-            pass
-            # TODO
-            # redis_client.set(key, json.dumps(res))
-            # redis_client.expire(key, 60 * 60 * 24 * 180)
+            redis_client.set(key, json.dumps(res))
+            redis_client.expire(key, 60 * 60 * 24 * 180)
 
-            # write_user_portrait(cl_id, ",".join(first_solutions_score.keys()), ",".join(second_solutions_score.keys()),
-            #                     ",".join(first_demands_score.keys()), ",".join(second_demands_score.keys()),
-            #                     ",".join(first_positions_score.keys()), ",".join(second_positions_score.keys()),
-            #                     ",".join(projects_score.keys()))
+            write_user_portrait(cl_id, ",".join(first_solutions_score.keys()), ",".join(second_solutions_score.keys()),
+                                ",".join(first_demands_score.keys()), ",".join(second_demands_score.keys()),
+                                ",".join(first_positions_score.keys()), ",".join(second_positions_score.keys()),
+                                ",".join(projects_score.keys()))
 
             body = {}
             for (k, v) in res.items():
@@ -285,9 +282,7 @@ def update_tag3_user_portrait(cl_id):
             body["device_id"] = cl_id
             body["last_modified"] = datetime.datetime.strftime(datetime.datetime.now(pytz.timezone("Asia/Shanghai")),
                                                                "%Y-%m-%dT%H:%M:%S.%f")[:-7] + "Z"
-
-            print(body)
-            # es_insert_device_info(cl_id, body)
+            es_insert_device_info(cl_id, body)
 
             # # write_user_portrait_doris(cl_id, ",".join(first_solutions_score.keys()), ",".join(second_solutions_score.keys()),
             # #                           ",".join(first_demands_score.keys()), ",".join(second_demands_score.keys()),