新增pyes测试

zhanghe06 · zhanghe06 · commit 2185dfc45fe9 · 2016-01-05T21:47:30.000+08:00
diff --git a/test/test_pyes.py b/test/test_pyes.py
@@ -0,0 +1,77 @@
+# encoding: utf-8
+__author__ = 'zhanghe'
+
+import pyes
+
+
+conn = pyes.ES(['127.0.0.1:9200'])  # &#36830;&#25509;es
+
+
+def create():
+    conn.create_index('test-index')  # &#26032;&#24314;&#19968;&#20010;&#32034;&#24341;
+
+    # &#23450;&#20041;&#32034;&#24341;&#23384;&#20648;&#32467;&#26500;
+    mapping = {u'id': {'boost': 1.0,
+                       'index': 'not_analyzed',
+                       'store': 'yes',
+                       'type': u'string',
+                       "term_vector": "with_positions_offsets"},
+               u'user_id': {'boost': 1.0,
+                            'index': 'not_analyzed',
+                            'store': 'yes',
+                            'type': u'string',
+                            "term_vector": "with_positions_offsets"},
+               u'nick': {'boost': 1.0,
+                         'index': 'analyzed',
+                         'store': 'yes',
+                         'type': u'string',
+                         "term_vector": "with_positions_offsets"},
+               u'city': {'boost': 1.0,
+                         'index': 'analyzed',
+                         'store': 'yes',
+                         'type': u'string',
+                         "term_vector": "with_positions_offsets"},
+               }
+
+    conn.put_mapping("test-type1", {'properties': mapping}, ["test-index"])  # &#23450;&#20041;test-type
+    conn.put_mapping("test-type2", {"_parent": {"type": "test-type1"}}, ["test-index"])  # &#20174;test-type&#32487;&#25215;
+
+    # &#25554;&#20837;&#32034;&#24341;&#25968;&#25454;
+    # {"id":"1", "user_id":"u1", "nick":u"&#21387;&#21147;&#24456;&#22823;", "city":u"&#25104;&#37117;"}: &#25991;&#26723;&#25968;&#25454;
+    # test-index&#65306;&#32034;&#24341;&#21517;&#31216;
+    # test-type: &#31867;&#22411;
+    # 1: id &#27880;&#65306;id&#21487;&#20197;&#19981;&#32473;&#65292;&#31995;&#32479;&#20250;&#33258;&#21160;&#29983;&#25104;
+    conn.index({"id": "1", "user_id": "u1", "nick": u"&#21387;&#21147;&#24456;&#22823;", "city": u"&#25104;&#37117;"}, "test-index", "test-type1", 1)
+    conn.index({"id": "2", "user_id": "u2", "nick": u"&#21387;&#21147;&#24456;&#23567;", "city": u"&#21271;&#20140;"}, "test-index", "test-type1")
+    conn.index({"id": "3", "user_id": "u3", "nick": u"&#27809;&#26377;&#21387;&#21147;", "city": u"&#25104;&#37117;"}, "test-index", "test-type1")
+
+    conn.default_indices = ["test-index"]  # &#35774;&#32622;&#40664;&#35748;&#30340;&#32034;&#24341;
+    conn.refresh()  # &#21047;&#26032;&#20197;&#33719;&#24471;&#26368;&#26032;&#25554;&#20837;&#30340;&#25991;&#26723;
+
+
+def query():
+    # &#26597;&#35810;nick&#20013;&#21253;&#21547;&#21387;&#21147;&#30340;&#35760;&#24405;
+    q = pyes.StringQuery(u"&#21387;&#21147;", 'nick')
+    results = conn.search(q)
+
+    for r in results:
+        print u"&#26597;&#35810;nick&#20013;&#21253;&#21547;&#21387;&#21147;&#30340;&#35760;&#24405;", r['nick'], r['city']
+
+    # &#26597;&#35810;city&#20013;&#21253;&#21547;&#25104;&#37117;&#30340;&#25968;&#25454;
+    q = pyes.StringQuery(u"&#25104;&#37117;", 'city')
+    results = conn.search(q)
+
+    for r in results:
+        print u"&#26597;&#35810;city&#20013;&#21253;&#21547;&#25104;&#37117;&#30340;&#25968;&#25454;", r['nick'], r['city']
+
+    # &#26597;&#35810;nick&#20013;&#21253;&#21547;&#24456;&#23567;&#25110;&#27809;&#26377;&#30340;&#25968;&#25454;
+    q = pyes.StringQuery(u"&#24456;&#23567; OR &#27809;&#26377;", 'nick')
+    results = conn.search(q)
+
+    for r in results:
+        print u"&#26597;&#35810;nick&#20013;&#21253;&#21547;&#24456;&#23567;&#25110;&#27809;&#26377;&#30340;&#25968;&#25454;", r['nick'], r['city']
+
+
+if __name__ == '__main__':
+    # create()
+    query()
