python 通过输入IP去es中获取相关数据

# -*- coding: utf-8 -*-
from subprocess import Popen, PIPE
import json
import sys
import os
reload(sys)
sys.setdefaultencoding('utf8')

input_ip = raw_input('please input your IP:')
print input_ip

def get_data():
    cmd = 'curl -X POST -H \"Content-Type: application/json\" http://127.0.0.1:9200/ipconnect/_search?pretty -d \'{\"query\": {\"bool\": {\"must\": [{\"term\": {\"src_ip\": \"%s\"}}],\"must_not\": [ ],\"should\": [ ]}},\"from\": 0,\"size\": 10,\"sort\": [ ],\"aggs\": { }}\' '%str(input_ip)
#     print cmd
    p = Popen(cmd, stdout=PIPE,stderr=PIPE,stdin=PIPE,shell=True)
    data = p.stdout.read()
    return data

def get_result_list(es_results):
    try:
        es_results = json.loads(es_results)
        final_result = es_results
#     print final_result
        return final_result
    except Exception as e:
        print e

if __name__ == '__main__':

    # 生成文件路径
    # 停用词路径
    # top10 文件路径
    result_data_path = os.getcwd() + '/IP_count.json'

    # print result_data_path

    data = open(result_data_path, "w")  # 打开当前写入路径

    es_results = get_data()
    final_results = get_result_list(es_results)

    #     print final_results
    data.write(json.dumps(final_results, ensure_ascii=False) + "\n")
    data.close()
    pass