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

最新推荐文章于 2024-12-07 17:32:05 发布

原创最新推荐文章于 2024-12-07 17:32:05 发布 · 265 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #elasticsearch

python 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种通过用户输入的IP地址查询Elasticsearch中相关信息的方法，利用Python的subprocess模块执行curl命令，实现从指定ES索引中搜索特定IP的连接记录，并将结果保存为JSON文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- 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