Python对ElasticSearch获取数据及操作

#!/usr/bin/env python# -*- coding: utf-8 -*-"""  @Time  : 2018/7/4  @Author : LiuXueWen  @Site  :   @File  : ElasticSearchOperation.py  @Software: PyCharm  @Description: 对elasticsearch数据的操作，包括获取数据，发送数据"""import elasticsearchimport jsonimport Util_Ini_Operationclass elasticsearch_data():  def __init__(self,hosts,username,password,maxsize,is_ssl):    # 初始化ini操作脚本，获取配置文件    try:      # 判断请求方式是否ssl加密      if is_ssl == "true":        # 获取证书地址        cert_pem = Util_Ini_Operation.get_ini("config.ini").get_key_value("certs","certs")        es_ssl = elasticsearch.Elasticsearch(          # 地址          hosts=hosts,          # 用户名密码          http_auth=(username,password),          # 开启ssl          use_ssl=True,          # 确认有加密证书          verify_certs=True,          # 对应的加密证书地址          client_cert=cert_pem        )        self.es = es_ssl      elif is_ssl == "false":        # 创建普通类型的ES客户端        es_ordinary = elasticsearch.Elasticsearch(hosts, http_auth=(username, password), maxsize=int(maxsize))        self.es = es_ordinary    except Exception as e:      print(e)  def query_data(self,keywords_list,date):    gte = "now-"+str(date)    query_data = {      # 查询语句      "query": {        "bool": {          "must": [            {              "query_string": {                "query": keywords_list,                "analyze_wildcard": True              }            },            {              "range": {                "@timestamp": {                  "gte": gte,                  "lte": "now",                  "format": "epoch_millis"                }              }            }          ],          "must_not": []        }      }    }    return query_data  # 从es获取数据  def get_datas_by_query(self,index_name,keywords,param,date):    '''    :param index_name: 索引名称    :param keywords: 关键字词，数组    :param param: 需要数据条件，例如_source    :param date: 过去时间范围,字符串格式，例如过去30分钟内数据，"30m"    :return: all_datas 返回查询到的所有数据（已经过param过滤）    '''    all_datas = []    # 遍历所有的查询条件    for keywords_list in keywords:      # DSL语句      query_data = self.query_data(keywords_list,date)      res = self.es.search(        index=index_name,        body=query_data      )      for hit in res['hits']['hits']:        # 获取指定的内容        response = hit[param]        # 添加所有数据到数据集中        all_datas.append(response)    # 返回所有数据内容    return all_datas  # 当索引不存在创建索引  def create_index(self,index_name):    '''    :param index_name: 索引名称    :return:如果创建成功返回创建结果信息，试过已经存在创建新的index失败返回index的名称    '''    # 获取索引的映射    # index_mapping = IndexMapping.index_mapping    # # 判断索引是否存在    # if self.es.indices.exists(index=index_name) is not True:    #   # 创建索引    #   res = self.es.indices.create(index=index_name,body=index_mapping)    #   # 返回结果    #   return res    # else:    #   # 返回索引名称    #   return index_name    pass  # 插入指定的单条数据内容  def insert_single_data(self,index_name,doc_type,data):    '''    :param index_name: 索引名称    :param doc_type: 文档类型    :param data: 需要插入的数据内容    :return: 执行结果    '''    res = self.es.index(index=index_name,doc_type=doc_type,body=data)    return res  # 向ES中新增数据,批量插入  def insert_datas(self,index_name):    '''    :desc 通过读取指定的文件内容获取需要插入的数据集    :param index_name: 索引名称    :return: 插入成功的数据条数    '''    insert_datas = []    # 判断插入数据的索引是否存在    self.createIndex(index_name=index_name)    # 获取插入数据的文件地址    data_file_path = self.ini.get_key_value("datafile","datafilepath")    # 获取需要插入的数据集    with open(data_file_path,"r+") as data_file:      # 获取文件所有数据      data_lines = data_file.readlines()      for data_line in data_lines:        # string to json        data_line = json.loads(data_line)        insert_datas.append(data_line)    # 批量处理    res = self.es.bulk(index=index_name,body=insert_datas,raise_on_error=True)    return res  # 从ES中在指定的索引中删除指定数据（根据id判断）  def delete_data_by_id(self,index_name,doc_type,id):    '''    :param index_name: 索引名称    :param index_type: 文档类型    :param id: 唯一标识id    :return: 删除结果信息    '''    res = self.es.delete(index=index_name,doc_type=doc_type,id=id)    return res  # 根据条件删除数据  def delete_data_by_query(self,index_name,doc_type,param,gt_time,lt_time):    '''    :param index_name:索引名称，为空查询所有索引    :param doc_type:文档类型，为空查询所有文档类型    :param param:过滤条件值    :param gt_time:时间范围，大于该时间    :param lt_time:时间范围，小于该时间    :return:执行条件删除后的结果信息    '''    # DSL语句    query_data = {      # 查询语句      "query": {        "bool": {          "must": [            {              "query_string": {                "query": param,                "analyze_wildcard": True              }            },            {              "range": {                "@timestamp": {                  "gte": gt_time,                  "lte": lt_time,                  "format": "epoch_millis"                }              }            }          ],          "must_not": []        }      }    }    res = self.es.delete_by_query(index=index_name,doc_type=doc_type,body=query_data,_source=True)    return res  # 指定index中删除指定时间段内的全部数据  def delete_all_datas(self,index_name,doc_type,gt_time,lt_time):    '''    :param index_name:索引名称，为空查询所有索引    :param doc_type:文档类型，为空查询所有文档类型    :param gt_time:时间范围，大于该时间    :param lt_time:时间范围，小于该时间    :return:执行条件删除后的结果信息    '''    # DSL语句    query_data = {      # 查询语句      "query": {        "bool": {          "must": [            {              "match_all": {}            },            {              "range": {                "@timestamp": {                  "gte": gt_time,                  "lte": lt_time,                  "format": "epoch_millis"                }              }            }          ],          "must_not": []        }      }    }    res = self.es.delete_by_query(index=index_name, doc_type=doc_type, body=query_data, _source=True)    return res  # 修改ES中指定的数据  def update_data_by_id(self,index_name,doc_type,id,data):    '''    :param index_name: 索引名称    :param doc_type: 文档类型，为空表示所有类型    :param id: 文档唯一标识编号    :param data: 更新的数据    :return: 更新结果信息    '''    res = self.es.update(index=index_name,doc_type=doc_type,id=id,body=data)    return res