ログ監視スクリプト

linuxには、tail -fという便利なコマンドがあります。os.statというものがあるというのを今日知ったのでちょこっとソースを書いてみた。引数から渡されたファイルを監視するようにしています。ファイルの末尾への移動、現在のファイルポインタの位置の取得方法、ファイルからのデータの読み出し方法だけで実現しています。
※久しぶりにpython触りました。

sample.py

#-*- coding: utf-8 -*-
import time
import os
import sys
from stat import *

def usage():
    print "Usage: # python %s filename" % argv[0]
    quit()


def init(filename):
    file = open(filename,'r')

    #ファイル末尾へ移動
    st_results = os.stat(filename)
    st_size = st_results[ST_SIZE]
    file.seek(st_size)

    return file

def tail_f(file, usec):
    msec = usec / 1000
    while 1:
        fpos = file.tell()
        line = file.readline()
        if not line:
            time.sleep(msec)
            file.seek(fpos)
        else:
            print line,
    # 未到達
    file.close()
    pass

if __name__ == '__main__':
    argv  = sys.argv
    argc  = len(argv)
    if( argc != 2 ):
        usage()

    filename = argv[1]
    file = init(filename)
    tail_f(file, 500)

ちなみにlinuxでしか使わないのであれば、tail -fのラッパースクリプトにしちゃうという手もありですね。

sample.py

import sys
import os

if __name__ == '__main__':
    argv = sys.argv
    argc = len(argv)

    if argc != 2:
        print "test",
        quit()

    filename = argv[1]
    os.system('tail -f ' + filename )