python分析日志脚本

#!/usr/bin/env python
# coding:utf-8

import sys,time

class DisplayFormat(object):

    def format_size(self,size):
        KB = 1024                   # KB -> B  1024
        MB = 1048576                # MB -> B  1024 * 1024
        GB = 1073741824             # GB -> B  1024 * 1024 * 1024
        TB = 1099511627776          # TB -> B  1024 * 1024 * 1024

        if size >= TB:
            size = str(size >> 40) + 'T'
        elif size < KB:
            size = str(size) + 'B'
        elif size >= GB and size < TB:
            size = str(size >> 30) + 'G'
        elif size >= MB and size < GB:
            size = str(size >> 20) + 'M'
        else:
            size = str(size >> 10) + 'K'

        return size

    formatstring = '%-18s %-10s %-12s %8s %10s %10s %10s %10s %10s %10s %10s'

    def echo_line(self):
        '''输出头部横线'''
        print self.formatstring % ('-'*15,'-'*10,'-'*12,'-'*12,'-'*10,'-'*10,'-'*10,'-'*10,'-'*10,'-'*10,'-'*10,)

    def echo_head(self):
        '''输出头部信息'''
        print self.formatstring % ('IP','Traffic','Time','Time%',200,404,403,503,500,302,304)

    def echo_error(self):
        '''输出错误信息'''
        print 'Usage: ' + sys.argv[0] + 'filepath [number]'

    def echo_time(self):
        '''输出脚本执行时间'''
        print 'The script is running %s second' % time.clock()


class HostInfo(object):

    # 定义一个主机ip 的所有状态列表
    host_info = ['200','404','403','503','500','302','304','size','time']

    def __init__(self,host):
        '''初始化一个主机信息字典'''
        self.host = host = {}.fromkeys(self.host_info,0)

    def add_1(self,status_size,is_size):
        '''对访问次数,http返回的状态码，ip流量进行加1操作'''
        if status_size == 'time':
            self.host['time'] += 1
        elif is_size:
            self.host['size'] = self.host['size'] + status_size
        else:
            self.host[status_size] += 1

    def get_value(self,value):
        '''取出字典的值'''
        return self.host[value]


class AnalysisFile(object):

    def __init__(self):
        '''初始化一个空字典'''
        self.empty = {}
        self.total_request_time,self.total_traffic,self.total_200,
        self.total_404,self.total_403,self.total_503,self.total_500,
        self.total_302,self.total_304 = 0,0,0,0,0,0,0,0,0

    def split_line_todict(self,line):
        '''传入文件的每一行取出0、8、9字段 生成字典 并返回这个字典'''
        line_split = line.split()
        line_dict = {'remote_host':line_split[0],'status':line_split[8],'bytes_sent':line_split[9]}
        return line_dict

    def read_log(self,logs):
        for line in logs:
            try:
                dict_line = self.split_line_todict(line)
                host = dict_line['remote_host']
                status = dict_line['status']
            except ValueError:
                continue
            except IndexError:
                continue

            if host not in self.empty:
                host_info_obj = HostInfo(host)
                self.empty[host] = host_info_obj
            else:
                host_info_obj = self.empty[host]

            host_info_obj.add_1('time',False)

            if status in host_info_obj.host_info:
                host_info_obj.add_1(status,False)

            try:
                bytes_sent = int(dict_line['bytes_sent'])
            except ValueError:
                bytes_sent = 0

            host_info_obj.add_1(bytes_sent,True)

        return self.empty

    def return_sorted_list(self,true_dict):
        '''循环读取字典,计算总的流量、总的访问次数以及总的http返回码'''
        for host_key in true_dict:
            host_value = true_dict[host_key]
            time = host_value.get_value('time')
            self.total_request_time = self.total_request_time + time
            size = host_value.get_value('size')
            self.total_traffic = self.total_traffic + size

            # 获取http返回状态码的次数
            v_200 = host_value.get_value('200')
            v_404 = host_value.get_value('404')
            v_403 = host_value.get_value('403')
            v_503 = host_value.get_value('503')
            v_500 = host_value.get_value('500')
            v_302 = host_value.get_value('302')
            v_304 = host_value.get_value('304')

            # 重新规划字典
            true_dict[host_key] = {'200':v_200,'404':v_404,'403':v_403,
                                   '503':v_503,'500':v_500,'302':v_302,
                                   '304':v_304,'size':size,'time':time}


            # 计算http返回状态码的总量
            self.total_200 = self.total_200 + v_200
            self.total_404 = self.total_404 + v_404
            self.total_403 = self.total_403 + v_403
            self.total_503 = self.total_503 + v_503
            self.total_500 = self.total_500 + v_500
            self.total_302 = self.total_302 + v_302
            self.total_304 = self.total_304 + v_304

                # 对总的访问次数和访问流量进行降序排序,并生成一个有序的列表
        sorted_list = sorted(true_dict.items(),key=lambda i:(i[1]['size'],
                                                                 i[1]['time']),reverse=True)

        return sorted_list


class Main(object):

    def main(self):
        '''主调函数'''
        # 初始化DisplayFormat类的实例
        displayformat = DisplayFormat()

        args = len(sys.argv)
        if args == 1:
            displayformat.echo_error()
        elif args == 2 or args == 3:
            log_file = sys.argv[1]
            try:
                files = open(log_file,'r')
                if args == 3:
                    lines = int(sys.argv[2])
                else:
                    lines = 0
            except IOError,e:
                print
                print e
                displayformat.echo_error()
            except VaueError,e:
                print
                print e
                displayformat.echo_error()

        else:
            displayformat.echo_error()


        #AnalysisFile类的实例化
        fileanalysis = AnalysisFile()

        # 调用read_log方法
        news_dict = fileanalysis.read_log(files)

        # 调用return_sorted_list方法
        new_list = fileanalysis.return_sorted_list(news_dict)

        # 计算所有ip的总量
        total_ip = len(new_list)

        if lines:
            new_list = new_list[0:lines]
        files.close()

        # 打印出总的ip数,总访问流量,总的访问次数
        print
        total_request_time = fileanalysis.total_request_time
        total_traffic = displayformat.format_size(fileanalysis.total_traffic)
        print '总IP数量: %s    总的访问流量: %s    总的请求次数: %d' % (total_ip,
                                                                   total_traffic,
                                                                   total_request_time)
        
        # 打印头部信息,和横线                                                                      
        print
        displayformat.echo_head()
        displayformat.echo_line()

        # 循环读取news_list列表取出time项目 计算time百分比 通过displayformat格式化输出主机信息
        for i in new_list:
            time = i[1]['time']
            time_percentage = (float(time) / float(fileanalysis.total_request_time)) * 100
            print displayformat.formatstring % (i[0],
                                                displayformat.format_size(i[1]['size']),
                                                time,str(time_percentage)[0:5],
                                                i[1]['200'],i[1]['404'],i[1]['403'],
                                                i[1]['503'],i[1]['500'],i[1]['302'],i[1]['304'])

        if not lines or total_ip == lines:
            displayformat.echo_line()
            print displayformat.formatstring % (total_ip,total_traffic,total_request_time,'100%',
                                                fileanalysis.total_200,fileanalysis.total_404,
                                                fileanalysis.total_403,fileanalysis.total_503,
                                                fileanalysis.total_500,fileanalysis.total_302,
                                                fileanalysis.total_304)

        # 显示执行脚本的时间
        print
        displayformat.echo_time()

if __name__ == '__main__':
    main = Main()
    main.main()