• Python Elasticsearch api


    一、介绍

    ElasticSearch是一个基于Lucene的搜索服务器。它提供了一个分布式多用户能力的全文搜索引擎,基于RESTful web接口。下面介绍了利用Python API接口进行数据查询,方便其他系统的调用。

    安装API

    pip3 install elasticsearch

    建立es连接

    无用户名密码状态

    from elasticsearch import Elasticsearch
    es = Elasticsearch([{'host':'10.10.13.12','port':9200}])

    默认的超时时间是10秒,如果数据量很大,时间设置更长一些。如果端口是9200,直接写IP即可。代码如下:

    es = Elasticsearch(['10.10.13.12'], timeout=3600)

    用户名密码状态

    如果Elasticsearch开启了验证,需要用户名和密码

    es = Elasticsearch(['10.10.13.12'], http_auth=('xiao', '123456'), timeout=3600)

    数据检索功能

    es.search(index='logstash-2015.08.20', q='http_status_code:5* AND server_name:"web1"', from_='124119')
    常用参数
    • index - 索引名
    • q - 查询指定匹配 使用Lucene查询语法
    • from_ - 查询起始点  默认0
    • doc_type - 文档类型
    • size - 指定查询条数 默认10
    • field - 指定字段 逗号分隔
    • sort - 排序  字段:asc/desc
    • body - 使用Query DSL
    • scroll - 滚动查询

    统计查询功能

    语法同search大致一样,但只输出统计值

    es.count(index='logstash-2015.08.21', q='http_status_code:500')

    输出:

    {'_shards':{'failed':0, 'successful':5, 'total':5}, 'count':17042}

    17042 就是统计值!

    知识扩展

    滚动demo

    # Initialize the scroll
    page = es.search(
        index ='yourIndex',
        doc_type ='yourType',
        scroll ='2m',
        search_type ='scan',
        size =1000,
        body ={
        # Your query's body
    })
     
    sid = page['_scroll_id']
    scroll_size = page['hits']['total']
     
    # Start scrolling
    while(scroll_size >0):
        print "Scrolling..."
        page = es.scroll(scroll_id = sid, scroll ='2m')
        # Update the scroll ID
        sid = page['_scroll_id']
        # Get the number of results that we returned in the last scroll
        scroll_size = len(page['hits']['hits'])
        print "scroll size: "+ str(scroll_size)
        # Do something with the obtained page
    View Code

    以上demo实现了一次取若干数据,数据取完之后结束,不会获取到最新更新的数据。我们滚动完之后想获取最新数据怎么办?滚动的时候会有一个统计值,如total: 5。跳出循环之后,我们可以用_from参数定位到5开始滚动之后的数据。

    但是我用的不是这个,用的是以下方法,链接如下:

    https://www.cnblogs.com/blue163/p/8126156.html

    在下面的内容中,我会详细介绍此代码如何使用!

    二、Query DSL

    range过滤器查询范围

    gt: > 大于
    lt: < 小于
    gte: >= 大于或等于
    lte: <= 小于或等于

    示例代码1

    "range":{
        "money":{
            "gt":20,
            "lt":40
        }
    }

    时间范围

    最近时间段

    比如我要查询最近1分钟的

    "range": {
        '@timestamp': {'gt': 'now-1m'}
    }

    最新1小时

    "range": {
        '@timestamp': {'gt': 'now-1h'}
    }

    最新1天的

    "range": {
        '@timestamp': {'gt': 'now-1d'}
    }

    指定时间段

    那么问题来了,它是根据当前时间来计算最近的时间。但是有些情况下,我需要制定时间范围,精确到分钟

    假设需要查询早上8点到9点的数据,可以这样

    "range": {
        '@timestamp': {
            "gt" : "{}T{}:00:00".format("2018-12-17","08"),
            "lt": "{}T{}:59:59".format("2018-12-17","09"),
            "time_zone": "Asia/Shanghai"
        }
    }

    注意:日期和小时之间,有一个字母T来间隔。不能用空格!

    time_zone 表示时区,如果默认的时区不会,可能会影响查询结果!

    bool组合过滤器

    must:所有分句都必须匹配,与 AND 相同。
    must_not:所有分句都必须不匹配,与 NOT 相同。
    should:至少有一个分句匹配,与 OR 相同。

    示例代码

    {
        "bool":{
          "must":[],
          "should":[],
          "must_not":[],
        }
    }

    term过滤器

    term单过滤

    {
        "terms":{
          "money":20
        }
    }

    表示money包含20的记录

    terms复数版本

    允许多个匹配条件

    {
        "terms":{
          "money": [20,30]
        }
    }

    表示money包含20或者30的记录

    结合bool+term来举一个实际的例子:

    查询path字段中包含applogs最近1分钟的记录

    "bool": {
        "must": [
            {
                "terms": {
                    "path": [
                        "applogs",
                    ]
                }
            },
            {
                "range": {
                    '@timestamp': {'gt': 'now-1m'}
                }
            }
        ]
    }
    View Code

    这里使用了terms复数版本,可以随时添加多个条件!

    正则查询 

    {
        "regexp": {
            "http_status_code": "5.*"
        }
    }

    match查询

    match 精确匹配

    {
        "match":{
          "email":"123456@qq.com"
        }
    }

    multi_match 多字段搜索

    {
        "multi_match":{
          "query":"11",
          "fields":["Tr","Tq"]
        }
    }

    demo

    获取最近一小时的数据

    {'query':
        {'filtered':
            {'filter':
                {'range':
                    {'@timestamp':{'gt':'now-1h'}}
                }
            }
        }
    }
    View Code


    条件过滤查询

    {
        "query":{
            "filtered":{
                "query":{"match":{"http_status_code":500}},
                "filter":{"term":{"server_name":"vip03"}}
            }
        }
    }
    View Code

    Terms Facet 单字段统计

    {'facets':
        {'stat':
            {'terms':
                {'field':'http_status_code',
                  'order':'count',
            'size':50}
            }
        },
        'size':0
    }
    View Code

    一次统计多个字段

    {'facets':
        {'cip':
            {'terms':
                {'fields':['client_ip']}},
                  'status_facets':{'terms':{'fields':['http_status_code'],
                  'order':'term',
                  'size':50}}},
            'query':{'query_string':{'query':'*'}},
        'size':0
    }
    View Code

    多个字段一起统计

    {'facets':
        {'tag':
            {'terms':
                {'fields':['http_status_code','client_ip'],
                  'size':10
               }
            }
        },
        'query':
            {'match_all':{}},
        'size':0
    }
    View Code

    数据组装

    以下是kibana首页的demo,用来统计一段时间内的日志数量

    {
      "facets": {
        "0": {
          "date_histogram": {
            "field": "@timestamp",
            "interval": "5m"
          },
          "facet_filter": {
            "fquery": {
              "query": {
                "filtered": {
                  "query": {
                    "query_string": {
                      "query": "*"
                    }
                  },
                  "filter": {
                    "bool": {
                      "must": [
                        {
                          "range": {
                            "@timestamp": {
                              'gt': 'now-1h'
                            }
                          }
                        },
                        {
                          "exists": {
                            "field": "http_status_code.raw"
                          }
                        },
                        # --------------- -------
                        # 此处加匹配条件
                      ]
                    }
                  }
                }
              }
            }
          }
        }
      },
      "size": 0
    }
    View Code

    如果想添加匹配条件,在以上代码标识部分加上过滤条件,按照以下代码格式即可

    {
    "query": {
        "query_string": {"query": "backend_name:baidu.com"}
        }
    },
    View Code

    先介绍到这里,后续会有Query DSL API介绍。

    三、需求分析

    需求

    下面是kibana展示的日志

    需要统计某一天的日志,统计每一个小时用户数,要求用户id不能重复。一个用户id就是一个用户,也称之为一个PV。

    看一段message字段信息

    2018-12-17 12:00:00,533 l=INFO [r=9538381535][s=2] [t=http-xxx-543] [APP=user] [Class=o.s.u.c.AccountController:1189] [Method=findCustomerByLoid]- Operation=find customer by loid,Params=loid:001,Content=start

    其中有一个[r=9538381535],这个9538381535就是用户id。那么用户登录手机APP操作,都会带着这个id,产生一条日志。

    比如user项目,那么最终要的数据格式如下:

    "user":{
        "00":1,
        "01":0,
        ...
        "22":3245,
        "23":765
    }

    这里使用24小时制来表示每一个时间段,有多个个用户访问了。注意:已经去重了用户id,统计用户数!

    四、相关技术点

    在放出最终代码之前,先来介绍相关技术点,便于理解代码。按照代码从上到下原则,分别来介绍!

    项目列表

    project_list = ['user',...]

    实际的项目是user,但是存储到elasticsearch中,是userlogs,加了一个logs后缀。这个是java后端代码定义的,便于识别!

    判断日期是否合法

    def isVaildDate(self, date):
        try:
            if ":" in date:
                time.strptime(date, "%Y-%m-%d %H:%M:%S")
            else:
                time.strptime(date, "%Y-%m-%d")
            return True
        except:
            return False
    View Code

    因为需要统计一周的数据,所以脚本执行时,需要传一个日期参数。那么日期参数,传给程序是否合法呢?需要有一个函数来判断!

    记录日志

        def write_log(self, content):
            """
            写入日志文件
            :param path:
            :param content:
            :return:
            """
            path = "print.log"
            with open(path, mode='a+', encoding='utf-8') as f:
                content = time.strftime('%Y-%m-%d %H:%M:%S') + ' ' + content + "
    "
                print(content)
                f.write(content)
    View Code

    为啥不用Python的日志模块呢?因为测试发现,它写入一些,我不想要的信息,太占用磁盘空间了。所以,我单独写了一个记录日志方法。

    获取elasticsearch数据

    def Get_Data_By_Body(self, project, fixed_date, hour):
        """
        获取数据
        :param project: 项目名
        :param fixed_date: 指定日期
        :param hour: 24小时制中的某一个小时
        :return: object
        """
        # 查询条件,查询项目最近1小时的数据。
        doc = {
            "query": {
                "bool": {
                    "must": [
                        {
                            "terms": {
                                "path": [
                                    project + "logs",
                                ]
                            }
                        },
                        {
                            "range": {
                                '@timestamp': {
                                    "gt": "{}T{}:00:00".format(fixed_date, hour),
                                    "lt": "{}T{}:59:59".format(fixed_date, hour),
                                    "time_zone": "Asia/Shanghai"
                                }
                            }
                        }
                    ]
                }
            }
        }
    View Code

    由于线上数据量过大,因此直接查询一天的数据,会卡死。所以是切分为每一个小时查询!

    上面的query表示查询语句,大概就是查询指定项目(项目名+logs),1小时范围内的数据

    scroll获取数据

    由于1小时内的数据量,也很大。不能直接返回!默认不指定size,是返回10条数据!

    size = 1000  # 指定返回1000条
    queryData = self.es.search(index=self.index_name, body=doc, size=size, scroll='1m', )

    参数解释:

    size 指定返回的条数,默认返回10条

    index 指定索引名

    body 查询语句

    scroll 告诉 Elasticsearch 把搜索上下文再保持一分钟。1m表示1分钟

    返回结果

    mdata = queryData.get("hits").get("hits")  # 返回数据,它是一个列表类型
    if not mdata:
        self.write_log('%s mdata is empty!' % project)

    queryData 返回一个字典,那么真正的查询结果在queryData['hits']['hits']中,如果这个值没有,表示没有查询到数据!

    注意:它并不是返回所有的结果,而是一页的数据,是一个列表类型。因为我们使用了scroll获取数据,只返回一页!

    分页数据

    上面只是返回了1页,我要所有数据,怎么办?需要使用分页,先来看一下分页公式

    divmod(总条数, 每页大小)

    注意:divmod返回一个元祖,第一个元素,就是要分页数

    总条数,使用

    total = queryData['hits']['total']  # 返回数据的总条数

    每页大小,就是上面指定的size

    size = 1000  # 指定返回1000条

    那么遍历每一页数据,需要这样

    scroll_id = queryData['_scroll_id']  # 获取scrollID
    total = queryData['hits']['total']  # 返回数据的总条数
    
    # 使用divmod设置分页查询
    # divmod(total,1000)[0]+1 表示总条数除以1000,结果取整数加1
    for i in range(divmod(total, size)[0] + 1):
        res = self.es.scroll(scroll_id=scroll_id, scroll='1m')  # scroll参数必须指定否则会报错
        mdata += res["hits"]["hits"]  # 扩展列表

    scroll_id给es.scroll获取数据使用,这个参数必须要有。

    由于Python中的range是顾头不顾尾,所以需要加1。使用for循环,就可以遍历每一个分页数

    es.scroll(scroll_id=scroll_id, scroll='1m') 才是真正查询每一页的数据,必须要指定这2个参数。它的返回结果,就是查询结果!返回一个列表

    上面的mdata是一个列表,res也是列表。因此使用+=就可以扩展列表,得到所有数据!

    创建年月日目录

    def create_folder(self, fixed_date):
        """
        创建年/月/日 文件夹
        :return: path
        """
    
        # 系统当前时间年份
        # year = time.strftime('%Y', time.localtime(time.time()))
        # # 月份
        # month = time.strftime('%m', time.localtime(time.time()))
        # # 日期
        # day = time.strftime('%d', time.localtime(time.time()))
    
        # 年月日
        year, month, day = fixed_date.split("-")
    
        # 具体时间 小时分钟毫秒
        # mdhms = time.strftime('%m%d%H%M%S', time.localtime(time.time()))
    
        # 判断基础目录是否存在
        if not os.path.exists(os.path.join(self.BASE_DIR, 'data_files')):
            os.mkdir(os.path.join(self.BASE_DIR, 'data_files'))
    
        # 年月日
        fileYear = os.path.join(self.BASE_DIR, 'data_files', year)
        fileMonth = os.path.join(fileYear, month)
        fileDay = os.path.join(fileMonth, day)
    
        # 判断目录是否存在,否则创建
        try:
            if not os.path.exists(fileYear):
                os.mkdir(fileYear)
                os.mkdir(fileMonth)
                os.mkdir(fileDay)
            else:
                if not os.path.exists(fileMonth):
                    os.mkdir(fileMonth)
                    os.mkdir(fileDay)
                else:
                    if not os.path.exists(fileDay):
                        os.mkdir(fileDay)
    
            return fileDay
        except Exception as e:
            print(e)
            return False
    View Code

    统计结果是最终写入到一个txt里面,那么如何存储呢?使用年月日目录在区分,可以知道这个txt文件,是属于哪一天的。到了一定时间后,可以定期清理,非常方便!

    这里使用的传参方式,传入一个日期。所以使用"-"就可以切割出年月日

    # 年月日
    year, month, day = fixed_date.split("-")

    输出24小时

    使用以下代码就可以实现

    hour_list = ['{num:02d}'.format(num=i) for i in range(24)]

    输出:

    ['00', '01', '02', '03', '04', '05', '06', '07', '08', '09', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23']

    项目统计字典

    需要统计每一个项目的每一个小时的用户id,用户id必须去重。既然要去重,我们首先会想到用集合。

    但是还有一个办法,使用字典,也可以去重。因为字典的key是唯一的。

    构造24小时字典

    先来构造项目user的数据,格式如下:

    "basebusiness": {
        "00": {},
        "01": {},
        "02": {},
        "03": {},
        "04": {},
        "05": {},
        "06": {},
        "07": {},
        "08": {},
        "09": {},
        "10": {},
        "11": {},
        "12": {},
        "13": {},
        "14": {},
        "15": {},
        "16": {},
        "17": {},
        "18": {},
        "19": {},
        "20": {},
        "21": {},
        "22": {},
        "23": {},
    }
    View Code

    这只是一个项目,实际有很多项目。所以每一个字典,都有这样的24小时数据。相关代码如下:

    project_dic = {}  # 项目统计字典
    # 24小时
    hour_list = ['{num:02d}'.format(num=i) for i in range(24)]
    
    for hour in hour_list:  # 遍历24小时
        # print("查询{}点的数据###############".format(hour))
        self.write_log("查询{}点的数据###############".format(hour))
        for project in project_list:  # 遍历项目列表
            if not project_dic.get(project):
                project_dic[project] = {}  # 初始化项目字典
    
            if not project_dic[project].get(hour):
                project_dic[project][hour] = {}  # 初始化项目小时字典
    View Code

    这里的每一个小时,都是空字典。还没有添加数据,需要添加用户id,下面会讲到!

    正则匹配用户id

    看这一点字符串

    2018-12-17 12:00:00,533 l=INFO [r=9538381535][s=2] [t=http-xxx-543] [APP=user]

    需要提取出9538381535,思路就是:匹配中括号内容-->提取以r=开头的内容-->使用等号切割,获取用户id

    匹配中括号内容

    p1 = re.compile(r'[[](.*?)[]]', re.S)  # 最小匹配,匹配中括号的内容

    注意:这里要使用最小匹配,不能使用贪婪匹配。这一段正则,我是用网上找的,测试ok

    提取和切割,就比较简单了。使用startswith和split方法,就可以了!

    使用字典去重

    接下来,需要将用户id写入到字典中,需要去重,否则字典添加时,会报错!

    那么如何使用字典去重呢?只需要遵循一个原则即可! 有则忽略,无则添加

    # 判断字典中rid不存在时,避免字典键值重复
    if not project_dic[project][hour].get(rid):
        project_dic[project][hour][rid] = True  # 添加值

    生成器

    这里主要在2个方法中,使用了生成器。生成器的优点,就是节省内容。

    一处在是Get_Data_By_Body方法中,它需要返回所有查询的数据。数据量非常大,因此必须要生成器,否则服务器内容就溢出!

    还有一处,就main方法。它是返回项目的统计结果。注意,它不是最终结果。它里面保存了每一个项目,每一个小时中的用户id,是已经去重的用户id。

    数据量也是比较大,当然,没有Get_Data_By_Body方法返回的结果大。

    统计每一个小时用户数

    main方法,返回的字典格式如下:

    "user":{
        "00":{
            "242412":True,
        }
        "01":{
            "":True,
        },
        ...
        "22":{
            "457577":True,
            "546583":True,
        },
        "23":{
            "457577":True,
            "546583":True,
            "765743":True,
        }
    }

    我需要知道,每一个小时的用户数。怎么统计呢?用2个方法

    1. 遍历字典的每一个小时,使用计数器

    2. 使用len方法(推荐)

    最简单的方法,就是使用len方法,就可以知道每一个小时有多少个key

    for i in dic:  # 遍历数据
        if not final_dic.get(i):
            final_dic[i] = {}  # 初始化字典
    
        for h in sorted(dic[i]):  # 遍历项目的每一个小时
            # 统计字典的长度
            final_dic[i][h] = len(dic[i][h])

    有序字典

    看下面的数据

    可以发现,24小时,排序是乱的。这样给领导看时,不太美观。所以需要对24小时进行排序!

    在Python 3.6之前,字典的key是无序的。因此,需要定义一个有序字典,在写入之前,要对字典的key做一次排序。

    这样顺序写入到有序字典之后,之后再次调用,依然是有序的!

    order_dic = OrderedDict()  # 实例化一个有序字典
        final_dic = {}  # 最终统计结果
        for dic in data:  # 遍历生成器
            for i in dic:  # 遍历数据
                if not final_dic.get(i):
                    final_dic[i] = order_dic  # 初始化字典
    
                # 有序字典必须先对普通字典key做排序
                for h in sorted(dic[i]):  # 遍历项目的每一个小时
                    # 统计字典的长度
                    final_dic[i][h] = len(dic[i][h])

    完整代码

    #!/usr/bin/env python3
    # coding: utf-8
    
    
    import re
    import os
    import sys
    import json
    import time
    from collections import OrderedDict
    from elasticsearch import Elasticsearch
    
    # 项目列表
    project_list = ['usercenter', ['login']]
    
    
    # yesterday = (datetime.datetime.now() + datetime.timedelta(days=-1)).strftime("%Y-%m-%d")
    # today = datetime.datetime.now().strftime("%Y-%m-%d")
    
    
    class ElasticObj:
        def __init__(self, index_name, ip, fixed_date, timeout=3600):
            '''
            :param index_name: 索引名称
            :param ip: elasticsearch地址
            :param timeout: 设置超时间,默认是10秒的,如果数据量很大,时间要设置更长一些
            '''
            self.index_name = index_name
            self.ip = ip
            self.timeout = timeout
            # 无用户名密码状态
            # self.es = Elasticsearch([self.ip], timeout=self.timeout)
            # 用户名密码状态
            # self.es = Elasticsearch([ip],http_auth=('elastic', 'password'),port=9200)
            self.es = Elasticsearch([self.ip], http_auth=('elastic', '123456'), timeout=self.timeout)
    
            self.fixed_date = fixed_date  # 指定日期
    
            # 当前py文件所在的文件夹
            self.BASE_DIR = os.path.dirname(os.path.abspath(__file__))
            self.fileDay = self.create_folder()  # 创建日志和数据目录
    
        @staticmethod
        def isVaildDate(date):
            """
            判断日期是否合法
            :param date: 日期,比如: 2018-03-30
            :return: bool
            """
            try:
                if ":" in date:
                    time.strptime(date, "%Y-%m-%d %H:%M:%S")
                else:
                    time.strptime(date, "%Y-%m-%d")
                return True
            except:
                return False
    
        def write_log(self, content):
            """
            写入日志文件
            :param content: 写入内容
            :return:
            """
            path = os.path.join(self.fileDay,"output_%s.log" %self.fixed_date)
            # path = "output_{}.log".format(self.fixed_date)
            with open(path, mode='a+', encoding='utf-8') as f:
                content = time.strftime('%Y-%m-%d %H:%M:%S') + ' ' + content + "
    "
                print(content)
                f.write(content)
    
        def Get_Data_By_Body(self, project, hour):
            """
            获取数据
            :param project: 项目名
            :param hour: 24小时制中的某一个小时
            :return: 生成器
            """
            # doc = {'query': {'match_all': {}}}
            # 查询条件,查询项目最近1小时的数据。now-1h表示最近1小时
            # print(type(fixed_date))
            # print("{date}T00:00:00".format(date=fixed_date))
            # 24小时
    
            doc = {
                "query": {
                    "bool": {
                        "must": [
                            {
                                "terms": {
                                    "path": [
                                        project + "logs",
                                    ]
                                }
                            },
                            {
                                # "range": {
                                #     '@timestamp': {'gt': 'now-1m'}
                                # }
                                "range": {
                                    '@timestamp': {
                                        "gt": "{}T{}:00:00".format(self.fixed_date, hour),
                                        "lt": "{}T{}:59:59".format(self.fixed_date, hour),
                                        "time_zone": "Asia/Shanghai"
                                    }
                                }
                            }
                        ]
                    }
                }
            }
            # queryData = self.es.search(index=self.index_name, body=doc)
            # scroll 参数告诉 Elasticsearch 把搜索上下文再保持一分钟,1m表示1分钟
            # size 参数允许我们配置没匹配结果返回的最大命中数。每次调用 scroll API 都会返回下一批结果,直到不再有可以返回的结果,即命中数组为空。
            size = 1000  # 指定返回1000条
            queryData = self.es.search(index=self.index_name, body=doc, size=size, scroll='1m', )
            # print(queryData['hits']['total'])
    
            mdata = queryData.get("hits").get("hits")  # 返回查询的数据,不是所有数据,而是一页的数据,它是一个列表类型
            if not mdata:
                self.write_log('%s mdata is empty!' % project)
    
            # scroll_id 的值就是上一个请求中返回的 _scroll_id 的值
            scroll_id = queryData['_scroll_id']  # 获取scrollID
            total = queryData['hits']['total']  # 返回数据的总条数
            # print("查询项目{} {}点的数据,总共有{}条".format(project,hour,total))
            self.write_log("查询项目{} {}点的数据,总共有{}条".format(project, hour, total))
    
            # 使用divmod设置分页查询
            # divmod(total,1000)[0]+1 表示总条数除以1000,结果取整数加1
            for i in range(divmod(total, size)[0] + 1):
                res = self.es.scroll(scroll_id=scroll_id, scroll='1m')  # scroll参数必须指定否则会报错
                mdata += res["hits"]["hits"]  # 扩展列表
                # yield mdata
    
            # print(mdata)
            # return mdata
            yield mdata
    
        def create_folder(self):
            """
            创建年/月/日 文件夹
            :return: path
            """
    
            # 系统当前时间年份
            # year = time.strftime('%Y', time.localtime(time.time()))
            # # 月份
            # month = time.strftime('%m', time.localtime(time.time()))
            # # 日期
            # day = time.strftime('%d', time.localtime(time.time()))
    
            # 年月日
            year, month, day = self.fixed_date.split("-")
    
            # 具体时间 小时分钟毫秒
            # mdhms = time.strftime('%m%d%H%M%S', time.localtime(time.time()))
    
            # 判断基础目录是否存在
            if not os.path.exists(os.path.join(self.BASE_DIR, 'data_files')):
                os.mkdir(os.path.join(self.BASE_DIR, 'data_files'))
    
            # 年月日
            fileYear = os.path.join(self.BASE_DIR, 'data_files', year)
            fileMonth = os.path.join(fileYear, month)
            fileDay = os.path.join(fileMonth, day)
    
            # 判断目录是否存在,否则创建
            try:
                if not os.path.exists(fileYear):
                    os.mkdir(fileYear)
                    os.mkdir(fileMonth)
                    os.mkdir(fileDay)
                else:
                    if not os.path.exists(fileMonth):
                        os.mkdir(fileMonth)
                        os.mkdir(fileDay)
                    else:
                        if not os.path.exists(fileDay):
                            os.mkdir(fileDay)
    
                return fileDay
            except Exception as e:
                print(e)
                return False
    
        def main(self):
            """
            主要处理逻辑
            :return: 生成器
            """
            project_dic = {}  # 项目统计字典
            # fixed_date = datetime.datetime.strptime(fixed_date, "%Y-%m-%d")
            # strftime("%Y-%m-%d")
            # conv_date = fixed_date.strftime("%Y-%m-%d")
            # print(conv_date, type(conv_date))
            # exit()
            # now_hour = fixed_date.strftime('%H')  # 当前时间的小时
            # print(now_hour)
            # 24小时
            hour_list = ['{num:02d}'.format(num=i) for i in range(24)]
            # hour_list = ['{num:02d}'.format(num=i) for i in range(2)]
    
            # project="usercenter"
            # project_dic[project] = {now_hour: {}}  # 初始化字典
            for hour in hour_list:  # 遍历24小时
                # print("查询{}点的数据###############".format(hour))
                self.write_log("查询{}点的数据###############".format(hour))
                for project in project_list:  # 遍历项目列表
                    if not project_dic.get(project):
                        project_dic[project] = {}  # 初始化项目字典
    
                    if not project_dic[project].get(hour):
                        project_dic[project][hour] = {}  # 初始化项目小时字典
    
                    mdata = self.Get_Data_By_Body(project, hour)  # 获取数据
                    for item in mdata:  # 遍历生成器
                        for hit in item:  # 遍历返回数据
                            # hit是一个字典
                            str1 = hit['_source']['message']  # 查询指定字典
                            p1 = re.compile(r'[[](.*?)[]]', re.S)  # 最小匹配,匹配中括号的内容
                            for i in re.findall(p1, str1):  # 遍历结果
                                if i.startswith('r='):  # 判断以r=开头的
                                    rid = i.split("=")[1]  # 获取rid
                                    # print("rid",rid)
                                    # 判断字典中rid不存在时,避免字典键值重复
                                    if not project_dic[project][hour].get(rid):
                                        project_dic[project][hour][rid] = True  # 添加值
    
                time.sleep(1)  # 休眠1秒钟
    
            # return project_dic
            yield project_dic
    
    
    if __name__ == '__main__':
        # fixed_date = "2018-12-16"
        fixed_date = sys.argv[1]  # 日期参数
        if not ElasticObj.isVaildDate(fixed_date):
            print("日期不合法!")
            exit()
    
        startime = time.time()  # 开始时间
    
        index_name = "common-*"
        es_server = "192.168.92.131"
    
        obj = ElasticObj(index_name, es_server, fixed_date)  # 连接elasticsearch
        print("正在查询日期%s这一天的数据" % fixed_date)
        obj.write_log("###########################################")
        obj.write_log("正在查询日期%s这一天的数据" % fixed_date)
    
        data = obj.main()
        # print("初步结果",data)
    
        # fileDay = obj.create_folder()  # 创建目录
        # if not fileDay:
        #     # print("创建目录失败!")
        #     obj.write_log("创建目录失败!")
        #     exit()
    
        order_dic = OrderedDict()  # 实例化一个有序字典
        final_dic = {}  # 最终统计结果
        for dic in data:  # 遍历生成器
            for i in dic:  # 遍历数据
                if not final_dic.get(i):
                    final_dic[i] = order_dic  # 初始化字典
    
                # 有序字典必须先对普通字典key做排序
                for h in sorted(dic[i]):  # 遍历项目的每一个小时
                    # 统计字典的长度
                    final_dic[i][h] = len(dic[i][h])
    
        # print("最终结果",final_dic)  # 统计结果
        obj.write_log("最终结果执行完毕!")
    
        # 写入文件
        with open(os.path.join(obj.fileDay, 'access_data.txt'), encoding='utf-8', mode='a') as f:
            f.write(json.dumps(final_dic) + "
    ")
    
        endtime = time.time()
        take_time = endtime - startime
    
        if take_time < 1:  # 判断不足1秒时
            take_time = 1  # 设置为1秒
        # 计算花费时间
        m, s = divmod(take_time, 60)
        h, m = divmod(m, 60)
    
        # print("本次花费时间 %02d:%02d:%02d" % (h, m, s))
        obj.write_log("统计日期%s这一天的数据完成!请查阅data_files目录的日志和数据文件" % fixed_date)
        obj.write_log("本次花费时间 %02d:%02d:%02d" % (h, m, s))
    View Code

    日志文件和数据文件,都在年月日目录里面!

    本文参考链接:

    http://www.cnblogs.com/letong/p/4749234.html

    http://www.linuxyw.com/790.html

    https://www.cnblogs.com/blue163/p/8126156.html

  • 相关阅读:
    如何稳定地使用 Google 搜索https://encrypted.google.com/
    widows 2008 同步时间命令
    MySql 初始化权限脚本
    [转] windows下Svn服务器之必须提交修改注释篇
    给编译好的DLL增加签名
    Anychart 破解备注
    Javascript 日期时间格式正则
    微服务项目规范(二)
    微服务项目规范(一)
    mac系统安装、启动与关闭redis
  • 原文地址:https://www.cnblogs.com/xiao987334176/p/10130712.html
Copyright © 2020-2023  润新知