【Python】linux系统more基本命令python源码分享

此python源码是linux系统more基本命令的实现。
实现linux中more的基本功能，当more后加一个文件名参数时候，分屏显示按空格换页，按回车换行',在左下角显示百分比；
以处理管道参数的输入，处理选项+num:从指定行开始显示，+/string :查找字符串，从指定字符串之后开始显示

运行环境：安装有PYTHON的linux系统

调用示例：
more.py [+num ] [+/pattern] filename
command|./more.py [+num ] [+/pattern]
more.p --help 输出帮助信息
num 是 要从第几行开始显示，pattern是要在文件中查找的字符串

学习python过程中有不懂的可以加入我的python零基础系统学习交流秋秋qun：前面是784，中间758，后面是214，与你分享Python企业当下人才需求及怎么从零基础学习Python，和学习什么内容。相关学习视频资料、开发工具都有分享！

`#!/usr/bin/env python`
`# -*- coding:utf-8-*-`
`#文件名字:more.py`
`import` `os`
`import` `sys`
`import` `curses` `#用于获取终端的尺寸`
`import` `re` `#用于字符匹配`
`import` `signal` `#用于处理ctrl+c中断`
`import` `fcntl` `# 处理显示过程中屏幕的变化`
`import` `termios` `#获取终端信息`
`import` `struct`
`page_len` `=` `0`   `#满屏时可以显示的最大行数`
`line_len` `=` `0`   `#满屏时每行可以显示的最大字节数`
`sig_up` `=` `0`    `#中断信号标志`
`winsz_chg` `=` `0`   `#窗口大小改变标志`
`def` `win_sz_chg(signum, frame):`
`'''  函数功能：本函数是屏幕变化信号的处理函数'''`
`global` `page_len, line_len, winsz_chg`
`winsz_chg` `=` `1`
`signal.signal(signal.SIGWINCH, signal.SIG_IGN)`
`s` `=` `struct.pack(``"HHHH"``,` `0``,` `0``,` `0``,` `0``) `
`a` `=` `struct.unpack(``'hhhh'``, fcntl.ioctl(sys.stdout.fileno(), termios.TIOCGWINSZ , s))`
`#获取当前窗口的大小`
`page_len` `=` `int``(a[``0``])` `-` `1`  `#留一行显示进度`
`line_len` `=` `int``(a[``1``])`
`signal.signal(signal.SIGWINCH, win_sz_chg)` `#不调用会导致只能检测一次屏幕变化`
`signal.signal(signal.SIGWINCH, win_sz_chg)` `#接收处理窗口改变信号`
`def` `term_do_exit(signum, frame):`
`'''  函数功能：键盘中断信号的响应函数'''`
`global` `sig_up`
`sig_up` `=` `1`           `#将键盘中断标志置1`
`os.system(``"stty -F /dev/tty echo"``)` `#恢复终端输出回车有效状态`
`os.system(``"stty -F /dev/tty -cbreak"``)``#重新设着屏幕为输入字符回显状态`
`return`
`signal.signal(signal.SIGINT, term_do_exit)` `#接收并处理键盘中断信号`
`def` `usage():`
`'''显示脚本的各参数的含义和调用格式'''`
`print` `"-----------------usage-----------------"`
`print` `"1./more.py [+num] [+/pattern] filename"`
`print` `"2 command | ./more.py"`
`print` `"num: Start at line number num. "`
`print` `"pattern:Start at line number num."`
`print` `"space: next page"`
`print` `"q :do_exit the program"`
`print` `"enter:next line"`
`print``"----------------------------------------"`
`sys.exit()`
`def` `do_exit():`
`'''用于系统退出 '''`
`os.system(``"stty -F /dev/tty echo"``)` `#恢复终端输出回车有效状态`
`os.system(``"stty -F /dev/tty -cbreak"``)``#重新设着屏幕为输入字符回显状态`
`sys.exit()`
`def` `is_input():`
`''' 检测是否有管道数据输入 '''`
`try``:`
`f` `=` `sys.stdin.fileno()` `#判断时候有管道输入`
`init_tty` `=` `termios.tcgetattr(f)` `#当没有管道输入，也没有参数时候，显示提示`
`return` `0`
`except``:`
`return` `1`
`def` `get_line_num(args):`
`''' 从命令行参数中获取开始显示的指定行`
`参数：args:从命令行获取的参数返回值：要开始显示的指定行  '''`
`line_num` `=` `1`
`for` `i` `in` `args:` `#匹配要从第几行开始的行数`
`ln` `=` `re.search(r``'\+\d+'``,` `str``(i))`
`if` `ln:`
`line_num` `=` `int``(ln.group().lstrip(``'+'``))``#采用正则表达式处理去掉‘+’，得到开始显示的行号`
`break`
`return` `line_num`
`def` `get_patstr(args):`
`'''从命令行中获取要查找的字符串`
`参数：args:从命令行中获取的参数为返回值：要查找的字符串:  '''`
`patstr` `=` `""`
`for` `i` `in` `args:` `#获取要匹配的字符串`
`pa` `=` `re.search(r``'(\+\/\w*[^\s])'``,` `str``(i))`
`if` `pa:`
`break`
`if` `pa:`
`patstr` `=` `str``(pa.group().lstrip(``'+/'``))`
`return` `patstr`
`def` `get_args():`
`'''用于从命令行获取参数，解析各参数`
`返回值：(line_num,patstr,fp):要开始显示的指定行，要查找的字符串，要操作的文件对象 '''`
`line_num` `=` `1`
`patstr` `=` `""`
`args` `=` `sys.argv[``1``:]`
`if` `not` `args:`
`if` `is_input():` `#在没有参数时候，判断是否为管道命令输入，不是提示正确输入参数`
`fp` `=` `sys.stdin`
`return` `(line_num, patstr, fp)`
`else``:`
`usage()`
`else``:`
`if` `args[``-``1``]` `=``=` `"--help"``:`
`usage()`
`line_num` `=` `get_l`
`2000`
`ine_num(args)`
`patstr` `=` `get_patstr(args)`
`if` `'+'` `not` `in` `args[``-``1``]:`
`filename` `=` `args[``-``1``]`
`if` `not` `os.path.exists(filename):`
`print` `" 没有那个文件或目录"`
`do_exit()`
`else``:`
`fp` `=` `open``(filename)`
`else``:`
`if` `not` `is_input():`
`usage()`
`else``:`
`fp` `=` `sys.stdin`
`return` `(line_num, patstr, fp)`
`def` `get_screen_size():`
`''' 用于获取文件显示终端的尺寸   '''`
`global` `page_len, line_len      `
`screen` `=` `curses.initscr()      `
`page_len, line_len` `=` `screen.getmaxyx()``#获取屏幕显示尺寸`
`page_len` `=` `page_len` `-` `2`   `#去掉输入命令那行，和最后要显示more的那一行`
`curses.endwin()` `#此处不结束会导致后面显示的乱码`
`def` `show_more(pre_pg_len):`
`''' 等待键盘输入命令 ，进行相应的处理。`
`:param pre_pg_len:屏幕改变以前保存的可显示的最大行数'''`
`global`  `sig_up, winsz_chg, page_len`
`ft` `=` `open``(``'/dev/tty'``)` `#打开标准终端输入`
`sys.stdout.flush()` `#刷新缓存输出，否则显示会出现问题`
`c` `=` `''`
`while` `True``:`
`try``:`
`c` `=` `ft.read(``1``)``#读取一个字符`
`except` `IOError:`
`if` `sig_up:`
`do_exit()` `#键盘中断退出程序`
`if` `c` `=``=` `" "``:`
`print` `"\033[20D\033[K"` `#控制光标回到more--反白字体的行首，删除此行以达到more不随文字滚动效果`
`if` `winsz_chg:` `#如果此时屏幕大小变化，第一次返回之前屏幕满屏行数`
`winsz_chg` `=` `0`
`return` `pre_pg_len`
`else``:`
`return` `page_len` `#当输入是空格时候，分屏显示，显示下一屏`
`elif` `c` `=``=` `"q"``:`
`print` `"\033[20D\033[K"`
`return` `0`          `#当输入是"q"时，退出显示`
`elif` `c` `=``=` `'\n'``:`
`print` `"\033[20D\033[K"``,`
`return` `1`           `#当输入是换行符时候，多显示一行`
`def` `skip_ln(fp, line_num):`
`''' 读取文件到指定开始显示的行  '''`
`n` `=` `line_num` `-` `1`
`while` `n:`
`fp.readline()`
`if` `not` `fp:`
`return`
`n` `=` `n` `-` `1`
`def` `search(fp, patstr):`
`''' 在文件中寻找要查找的字符串。`
`:param fp:要显示的文件对象   `
`:param patstr:要查找的检索词  '''`
`global`  `sig_up`
`text` `=` `' '`
`while` `True``:`
`try``:`
`s` `=` `fp.read(``1``)        `
`if` `not` `s:`
`print` `"can not find the string in the file"`
`do_exit()`
`text` `=` `text` `+` `s`
`if` `patstr` `in` `text:`
`return`
`except` `IOError:`
`if` `sig_up:`
`do_exit()`
`def` `show_prog(read_size, total_size):        `
`'''  在显示屏幕的左下角显示反显的"More"`
`当要显示的是一个文件时，同时显示已经显示文件的百分比`
`当显示的是一个管道输入时，只显示“More”`
`:param read_size: 已经显示的文件`
`:param total_size:要显示的文件的总大小  '''`
`if` `total_size:`
`prog` `=` `int``(read_size` `*` `100` `/` `float``(total_size))`
`print`  `"\033[7m --More--"` `+` `str``(prog)` `+` `'%'` `+` `"\033[0m"``,` `#输出反白的文字“more”和显示百分数`
`else``:`
`print`  `"\033[7m --More--"` `+` `"\033[0m"``,` `#输出反白的文字“more”`
`return`
`def` `do_more(fp , line_num , patstr):`
`'''分屏显示文件内容`
`:param fp:要显示的文件对象`
`:param page_len: 可显示的最大行数`
`:param line_len:可每行可显示的最大字节数   '''`
`global` `page_len, line_len`
`read_size` `=` `0`
`total_size` `=` `0`
`os.system(``"stty -F /dev/tty cbreak"``)` `#调用linux命令设置屏幕为不等回车`
`os.system(``"stty -F /dev/tty -echo "``)``#设置屏幕为输入字符不回显`
`if` `fp !``=` `sys.stdin:`
`fp.seek(``0``,` `2``)` `#获取文件的总字节数，以便后来显示输出的百分比`
`total_size` `=` `fp.tell()`
`fp.seek(``0``,` `0``)`
`if` `line_num !``=` `1``:`
`skip_ln(fp, line_num)`
`if` `patstr:`
`search(fp, patstr)`
`try``:`
`line` `=` `fp.readline(line_len)``#按行读取文件，可以设置最大读取字数，当遇到“\n”时候，将"\n"读入结束。`
`read_size` `=` `len``(line)`
`num_lns` `=` `0`
`while` `line:`
`if` `num_lns` `=``=` `page_len:` `#每次输出满屏后，等待指令`
`pre_pg_len` `=` `page_len`
`show_prog(read_size, total_size)`
`reply` `=` `show_more(pre_pg_len)          `
`if` `reply` `=``=` `0``:`
`break`
`num_lns` `-``=` `reply`
`print` `line.strip(``'\n'``)` `#用,来消除print 输出的换行符`
`sys.stdout.flush()` `#刷新缓存，否则会出现文件显示迟缓的问题`
`read_size` `=` `read_size` `+` `len``(line)`
`num_lns` `+``=` `1`
`line` `=` `fp.readline(line_len)`
`fp.close()`
`except` `IOError:`
`if` `sig_up:`
`do_exit()`
`if` `__name__` `=``=` `'__main__'``:`
`get_screen_size()` `#获取显示终端的尺寸`
`(line_num, patstr, fp)` `=` `get_args()`
`do_more(fp, line_num, patstr)`
`do_exit()`




来自 “ ITPUB博客 ” ，链接：http://blog.itpub.net/69913713/viewspace-2648826/，如需转载，请注明出处，否则将追究法律责任。

转载于:http://blog.itpub.net/69913713/viewspace-2648826/