Elasticsearch上手——Python API的簡單使用

Python夠直接，從它開始是個不錯的選擇。

Elasticsearch客戶端列表：https://www.elastic.co/guide/en/elasticsearch/client/index.html
Python API：https://www.elastic.co/guide/en/elasticsearch/client/python-api/current/index.html
參考文檔：http://elasticsearch-py.readthedocs.io/en/master/index.html

安裝

我在CentOS 7上安裝了Python3.6，安裝時使用下面的命令：

pip3 install elasticsearch

安裝時需要root權限

牛刀小試

由於Elasticsearch索引的文檔是JSON形式，而MongoDB存儲也是以JSON形式，因此這裏選擇通過MongoDB導出數據添加到Elasticsearch中。

使用MongoDB的Python API時，需要先安裝pymongo，命令：pip3 install pymongo

import traceback
from pymongo import MongoClient
from elasticsearch import Elasticsearch

# 建立到MongoDB的連接
_db = MongoClient('mongodb://127.0.0.1:27017')['blog']

# 建立到Elasticsearch的連接
_es = Elasticsearch()

# 初始化索引的Mappings設置
_index_mappings = {
  "mappings": {
    "user": { 
      "properties": { 
        "title":    { "type": "text"  }, 
        "name":     { "type": "text"  }, 
        "age":      { "type": "integer" }  
      }
    },
    "blogpost": { 
      "properties": { 
        "title":    { "type": "text"  }, 
        "body":     { "type": "text"  }, 
        "user_id":  {
          "type":   "keyword" 
        },
        "created":  {
          "type":   "date"
        }
      }
    }
  }
}

# 如果索引不存在，則創建索引
if _es.indices.exists(index='blog_index') is not True:
  _es.indices.create(index='blog_index', body=_index_mappings) 

# 從MongoDB中查詢數據，由於在Elasticsearch使用自動生成_id，因此從MongoDB查詢
# 返回的結果中將_id去掉。
user_cursor = db.user.find({}, projection={'_id':False})
user_docs = [x for x in user_cursor]

# 記錄處理的文檔數
processed = 0
# 將查詢出的文檔添加到Elasticsearch中
for _doc in user_docs:
  try:
    # 將refresh設爲true，使得添加的文檔可以立即搜索到；
    # 默認爲false，可能會導致下面的search沒有結果
    _es.index(index='blog_index', doc_type='user', refresh=True, body=_doc)
    processed += 1
    print('Processed: ' + str(processed), flush=True)
  except:
    traceback.print_exc()

# 查詢所有記錄結果
print('Search all...',  flush=True)
_query_all = {
  'query': {
    'match_all': {}
  }
}
_searched = _es.search(index='blog_index', doc_type='user', body=_query_all)
print(_searched, flush=True)

# 輸出查詢到的結果
for hit in _searched['hits']['hits']:
  print(hit['_source'], flush=True)

# 查詢姓名中包含jerry的記錄
print('Search name contains jerry.', flush=True)
_query_name_contains = {
  'query': {
    'match': {
      'name': 'jerry'
    }
  }
}
_searched = _es.search(index='blog_index', doc_type='user', body=_query_name_contains)
print(_searched, flush=True)

運行上面的文件(elasticsearch_trial.py)：

python3 elasticsearch_tria.py

可以得到下面的輸出結果：

Processed: 1
Processed: 2
Processed: 3
Search all...
{'took': 1, 'timed_out': False, '_shards': {'total': 5, 'successful': 5, 'failed': 0}, 'hits': {'total': 3, 'max_score': 1.0, 'hits': [{'_index': 'blog_index', '_type': 'user', '_id': 'AVn4TrrVXvwnWPWhxu5q', '_score': 1.0, '_source': {'title': 'Manager', 'name': 'Trump Heat', 'age': 67}}, {'_index': 'blog_index', '_type': 'user', '_id': 'AVn4TrscXvwnWPWhxu5s', '_score': 1.0, '_source': {'title': 'Engineer', 'name': 'Tommy Hsu', 'age': 32}}, {'_index': 'blog_index', '_type': 'user', '_id': 'AVn4Trr2XvwnWPWhxu5r', '_score': 1.0, '_source': {'title': 'President', 'name': 'Jerry Jim', 'age': 21}}]}}
{'title': 'Manager', 'name': 'Trump Heat', 'age': 67}
{'title': 'Engineer', 'name': 'Tommy Hsu', 'age': 32}
{'title': 'President', 'name': 'Jerry Jim', 'age': 21}
Search name contains jerry.
{'took': 3, 'timed_out': False, '_shards': {'total': 5, 'successful': 5, 'failed': 0}, 'hits': {'total': 1, 'max_score': 0.25811607, 'hits': [{'_index': 'blog_index', '_type': 'user', '_id': 'AVn4Trr2XvwnWPWhxu5r', '_score': 0.25811607, '_source': {'title': 'President', 'name': 'Jerry Jim', 'age': 21}}]}}