实现类似tail -f file功能

Posted 2020-08-31 dsowasp

python实现类tail -f file功能

python版本py3

tail -f file是打印最后10行，然后跟踪文件追加的内容打印出来。

python3 以为本方式打开的话，不能回退（f.seek(-1,1)）,所有以\'rb\'方式打开文件。

思路是f.seek(-n,2)从文件末尾回退n字节，然后f.readlines()读取文件，如果读到的少于10行，则再往前移动n字节，直到读到11行，然后打印出来，再跟踪打印文件追加的内容，并打印。

知识点：

f.tell()返回当前的文件位置

f.seek(n，m),m=0从文件开头往前或往后移到n字节，负数表示像文件头方向移动，正数表示向文件尾移动，m=1表示从文件指针当前位置移动，m=2表示从文件末尾开始移动，指针超出边界会报错。

f.seek(0,2)移到文件末尾

open(file,\'r\')以文本方式打开，open(file,\'rb\')以二进制方式打开。

由于文档是utf8或gbk编码保存的文件，以二进制方式打开文件读取到的是uft8或gbk编码（字节），所有打印是需要解码。

 

个人觉得在py2中,

"我”是str类型，utf8或gbk编码保存在内存中，只能进行解码成unicode操作：

 

py3中

‘我’是str类型，以unicode编码放在内存中，只能解码成bytes操作：

 

具体代码实现：

 1 #！ -*- coding:utf-8 -*-
 2 # runing on python3
 3 import os,sys
 4 from time import sleep
 5 COUDING=sys.getfilesystemencoding()
 6 
 7 class tail(object):
 8 
 9     def __init__(self,n,filename,callback):
10         self._n = n
11         self.filename= filename
12         self.callback = callback
13         self.track()
14 
15     def track(self):
16         if os.path.exists(self.filename):
17             if os.path.isfile(self.filename):
18                 try:
19                     with open(self.filename,\'rb\') as f:
20                         self._file = f
21                         # 文件内部指针移到末尾
22                         f.seek(0, os.SEEK_END)
23                         self._len = f.tell()
24                         self.showline()
25                         while True:
26                             line = f.readline()
27                             self.callback(line)
28                             sleep(0.5)
29                 except Exception as e:
30                     print(e)
31             else:
32                 sys.stdout.write("tail: %s: cannot follow end of this type of file; giving up on this name"%self.filename)
33         else:
34             sys.stdout.write("tail: cannot open `%s\' for reading: No such file or directory"%self.filename)
35 
36     def showline(self):
37         # 假设每行100个字节
38         line_len = 100
39         n = self._n
40         # 文件字节数小于500，直接从头读文件，然后取后5行
41         if self._len < n * line_len:
42             self._file.seek(0)
43             lines = self._file.readlines()[-self._n:]
44             self.callback(lines)
45         else:
46             # 指针总文件末尾往文件头移动n * line-len个字节
47             while True:
48                 self._file.seek(-n*line_len,os.SEEK_END)
49                 lines = self._file.readlines()
50                 # 读n+1行，因为seek是以字节为单位移到的，如果移到一个中文字符的编码的中间，会出现UnicodeDecodeError: \'utf-8\' codec can\'t decode byte 0x80 in position 0: invalid start byte
51                 if len(lines) > self._n:
52                     self.callback(lines[-self._n:])
53                     return
54                 else:
55                     n += 1
56 
57 def printline(lines):
58     if isinstance(lines,list):
59         for line in lines:
60             sys.stdout.write(line.decode(COUDING))
61     elif isinstance(lines,bytes):
62         sys.stdout.write(lines.decode(COUDING))
63 
64 if __name__ == "__main__":
65     if len(sys.argv) < 2:
66         print("xxx")
67     else:
68         # 可加参数解析模块来获取文件名和显示最后多少行
69         tail(10,sys.argv[1],printline)