zabbix监控域名带宽

代码地址：https://github.com/Ma-Jing/python/blob/master/ngxv2_traffic_daemon.py

READ.md里有使用说明！

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# a daemon which collecting channel traffic


import subprocess
import multiprocessing
import re
import os
import sys
import time
from os import stat
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
from SocketServer import ThreadingMixIn

__authors__ = ['majing <majing@staff.sina.com.cn>']
__version__ = "1.1"
__date__ = "Aug 14, 2015"
__license__ = "GPL license"

if (hasattr(os, "devnull")):
    NULL_DEVICE = os.devnull
else:
    NULL_DEVICE = "/dev/null"


def _redirectFileDescriptors():
    """
    Redirect stdout and stderr.
    """
    import resource  # POSIX resource information
    maxfd = resource.getrlimit(resource.RLIMIT_NOFILE)[1]
    if maxfd == resource.RLIM_INFINITY:
        maxfd = 1024

    for fd in range(0, maxfd):
        try:
            os.ttyname(fd)
        except:
            continue
        try:
            os.close(fd)
        except OSError:
            pass

    os.open(NULL_DEVICE, os.O_RDWR)
    os.dup2(0, 1)
    os.dup2(0, 2)


def python_daemon():
    """
    Make program run on daemon mode.
    """
    if os.name != 'posix':
        print 'Daemon is only supported on Posix-compliant systems.'
        os._exit(1)

    try:
        if(os.fork() > 0):
            os._exit(0)
    except OSError:
        print "create daemon failed."
        os._exit(1)

    os.chdir('/')
    os.setsid()
    os.umask(0)

    try:
        if(os.fork() > 0):
            os._exit(0)
        _redirectFileDescriptors()
    except OSError:
        print "create daemon failed."
        os._exit(1)


logfile = '/data0/log/sinaedge/esnv2/access.log'
if not os.path.isfile(logfile):
    os._exit(1)

# ensure data is shared between every processes.
manager = multiprocessing.Manager()
channel_traffics = {}
channel_traffics = manager.dict()


# a log generator
def logtailer(logfile):
    ''' custom a generator, when logfile
    rotated, this generator will be closed'''
    with open(logfile) as f:
        last_inode = stat(logfile).st_ino
        f.seek(0, 2)  # seek to eof
        while True:
            line = f.readline()
            if not line:
                if last_inode != stat(logfile).st_ino:
                    raise StopIteration('logfile rotated')
                else:
                    time.sleep(0.05)
                    continue
            yield line


def analysis_and_format_log():
    sourcelines = logtailer(logfile)
    while True:
        try:
            line = sourcelines.next()
            channel, transfer_bytes = line.split()[0:11:10]
            if not transfer_bytes.isdigit():
                continue
            if channel_traffics.has_key(channel):
                channel_traffics[channel] += int(transfer_bytes)
            else:
                channel_traffics[channel] = int(transfer_bytes)
        except StopIteration, e:
            '''
            if log rotated, clear channel_traffics dict;
            then close old generator, start a new lines generator.
            '''
            sourcelines.close()
            channel_traffics.clear()
            sourcelines = logtailer(logfile)
        except Exception, e:
            continue #ignore other error.


class GetChannelBandHandler(BaseHTTPRequestHandler):
    ''' a interface for query a channel current traffic'''
    def do_GET(self):
        self.send_response(200)
        self.end_headers()
        query_channel = self.path.split('/')[-1]
        if query_channel in channel_traffics:
            current_traffic = channel_traffics[query_channel]
            self.wfile.write(current_traffic)
        else:
            self.wfile.write("404: channel not found")
        return


class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
    pass


if __name__ == "__main__":
    python_daemon()
    # start analysis_and_format_log function, run in backgroud.
    d = multiprocessing.Process(name='daemon', target=analysis_and_format_log)
    d.daemon = True
    d.start()
    server = ThreadedHTTPServer(("127.0.0.1", 8888), GetChannelBandHandler)
    print 'Starting server on 8888, use <Ctrl-C> to stop'
    server.serve_forever()