modify

injection/data_sync/tasks.py

@@ -23,7 +23,8 @@ def write_to_es(es_type, pk_list, use_batch_query_set=False):
                 sub_index_name=type_info.name,
                 pk_list=pk_list,
                 use_batch_query_set=use_batch_query_set,
-                es=ESPerform.get_cli()
+                es=ESPerform.get_cli(),
+                es_type=es_type
             )
         else:
             pk_list = list(frozenset(pk_list))

trans2es/type_info.py

@@ -210,13 +210,21 @@ class TypeInfo(object):
             es=es,
         )
 
-    def insert_table_by_pk_list(self, sub_index_name, pk_list, es=None, use_batch_query_set=False):
+    def insert_table_by_pk_list(self, sub_index_name, pk_list, es=None, use_batch_query_set=False,es_type=None):
         if use_batch_query_set:
             qs = self.queryset
         else:
             qs = self.model.objects.all()
 
+        if es_type == "doctor_tips":
+            instance_list = qs.filter(pk__in=pk_list,doctor_type=DOCTOR_TYPE.DOCTOR)
+        elif es_type == "wordrel_tips":
+            instance_list = qs.filter(pk__in=pk_list, category__in=[13, 12, 11, 9, 1])
+        elif es_type == "tag_tips":
+            instance_list = qs.filter(pk__in=pk_list, tag_type__in=[TAG_TYPE.BODY_PART, TAG_TYPE.BODY_PART_SUB_ITEM, TAG_TYPE.ITEM_WIKI])
+        else:
             instance_list = qs.filter(pk__in=pk_list)
+
         data_list = self.bulk_get_data(instance_list)
         self.elasticsearch_bulk_insert_data(
             sub_index_name=sub_index_name,