resave

tractate_sentence_similary.py

@@ -73,18 +73,21 @@ def get_tractate_vector(sentence_lst, all_keywords_set, model):
 def save_tractate_vector_to_redis(all_keywords_set, model):
     es_result = get_new_user_tractate_info()
     count = 0
+    res = {}
+    redis_key = "rims:tractate:sentense:vector:new_user"
     for i in es_result:
         count += 1
         source = i["_source"]
         sentences = source["keynote_sentence"]
         id = source["id"]
         vec = get_tractate_vector(sentences, all_keywords_set, model)
-        redis_key = "rims:tractate:sentense:vector:" + str(id)
         if vec:
             print(count, id, len(vec))
-            redis_client5.set(redis_key, json.dumps(vec))
-            # TODO remove
-            redis_client5.expire(redis_key, 60 * 60 * 24 * 3)
+            res[id] = vec
+    if res:
+        redis_client5.set(redis_key, json.dumps(vec))
+        # TODO remove
+        redis_client5.expire(redis_key, 60 * 60 * 24 * 3)
 
 
 def save_tag_vector_to_redis(all_tags_lst, model):