update field

eda/smart_rank/tag3_update_user_portrait_offline.py

@@ -236,11 +236,10 @@ def update_tag3_user_portrait(cl_id):
             if k not in forbidden_set:
                 projects_score[k] = v
 
-
         tmp_res = {}
-        if (len(first_demands_score) == 0) and (len(first_solutions_score) == 0) and (len(first_positions_score) == 0):
+        if (len(first_demands_score) == 0) and (len(first_solutions_score) == 0) and (len(first_positions_score) == 0) and (
+                len(second_demands_score) == 0) and (len(second_solutions_score) == 0) and (len(second_positions_score) == 0):
             tmp_res = _get_all_by_projects(list(projects_score.keys()))
-            print("!!!!!!!")
 
         res = {
             "first_demands": first_demands_score,
@@ -251,27 +250,23 @@ def update_tag3_user_portrait(cl_id):
             "second_positions": second_positions_score,
             "projects": projects_score
         }
-        print(res)
         if tmp_res:
             res.update(tmp_res)
-        print("==================")
-        print(res)
 
         key = "doris:user_portrait:tag3:device_id:" + str(cl_id)
         redis_client = redis.StrictRedis.from_url("redis://:ReDis!GmTx*0aN9@172.16.40.173:6379")
 
-        # TODO
-        # if (len(first_demands_score.keys()) > 0) or (len(second_demands_score.keys()) > 0) or \
-        #    (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):
-        #     redis_client.set(key, json.dumps(res))
-        #     redis_client.expire(key, 60 * 60 * 24 * 30)
+        if (len(first_demands_score.keys()) > 0) or (len(second_demands_score.keys()) > 0) or \
+           (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):
+            redis_client.set(key, json.dumps(res))
+            redis_client.expire(key, 60 * 60 * 24 * 30)
 
-        #     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()))
 
         return cl_id
 
@@ -316,11 +311,12 @@ def consume_kafka():
 
 if __name__ == "__main__":
     start = datetime.datetime.now()
+
     # update_tag3_user_portrait("androidid_a25a1129c0b38f7b")
-    # TODO
-    device_id = "862460044588666"
-    update_tag3_user_portrait(device_id)
-    # consume_kafka()
+    # device_id = "862460044588666"
+    # update_tag3_user_portrait(device_id)
+
+    consume_kafka()
     end = datetime.datetime.now()
     print(end - start)
     print("done")