change

es_tag.py

@@ -56,8 +56,13 @@ def es_query(doc, body, offset, size, es=None):
         from_=offset,
         size=size)
 
-    number = res["hits"]["total"]
-    return number
+    # number = res["hits"]["total"]
+    total = res["hits"]["hits"]
+    print(total)
+    r = []
+    for i in total:
+        r.append(i['_source']['id'])
+    return r
 
 
 def answer():
@@ -1063,20 +1068,56 @@ def topic():
         tmp.append(es_query(category, video_star5_q, 0, 1))
 
         total_list.append(tmp)
-        # print(i)
-        # print(tmp)
 
     df = pd.DataFrame(total_list)
     df = df.rename(columns={0: "tag", 1: "star_3", 2: "star_4", 3: "star_5",
                             4: "video_star_3", 5: "video_star_4", 6: "video_star_5"})
     df.to_csv("/home/gmuser/topic.csv", index=False, encoding="utf_8_sig")
 
+def test_topic():
+    query_operator = "and"
+    query_type = "cross_fields"
+    category = 'tractate'
+
+    query = "眶隔脂肪释放术"
+    star3_q = {
+        "query": {"filtered": {
+            "filter": {
+                "bool": {
+                    "must": [{
+                        "multi_match": {
+                            "fields": ["content^1", "author^1", "tractate_tag_name^1",
+                                       "tractate_tag_name_content^1"],
+                            "operator": query_operator,
+                            "type": query_type,
+                            "query": query
+                        }
+                    }, {
+                        "term": {
+                            "is_online": True
+                        }
+                    }, {"term": {
+                        "content_level": 3
+                    }
+                    },
+                        {
+                            "term": {
+                                "status": "3"
+                            }
+                        }]
+                }
+
+            }
+        }
+        },"_source":{"include":"id"}
+    }
+    print(es_query(category, star3_q, 0, 10))
+
+
+
 
 if __name__ == "__main__":
-    answer()
-    topic()
-    print("topic")
-    question()
+    test_topic()