免费注册 查看新帖 |

Chinaunix

  平台 论坛 博客 文库
最近访问板块 发新帖
查看: 3827 | 回复: 0

Python中的cmd模块2--cmdqueue [复制链接]

论坛徽章:
0
发表于 2010-01-11 16:58 |显示全部楼层

最近应用 cmd 模块做了一个小工具,实现了一些小功能:
  • 自动将本地的文件同步到远程主机上(通过 ftplib实现)
  • 到远程主机上执行一下命令(通过 telnetlib 实现)
  • 调用系统中的程序 (通过 os.system("cmd") 实现)
  • 向手机发信息(用了 cocobear 的 pyfetion :) )
  • 自动生成一些代码

期间总是想实现: 一次输入多个命令, cmd 逐个执行的功能.
发了下代码发现: cmd模块的类Cmd中有个成员 cmdqueue.  利用这个成员可以完成我的需求.
cmdqueue 在 __init__ 中初始化了一次,之后在 cmdloop 中引用
                while not stop:
                if self.cmdqueue:
                    line = self.cmdqueue.pop(0)
                else:
                    if self.use_rawinput:
                        try:
                            line = raw_input(self.prompt)
                        except EOFError:
                            line = 'EOF'
                    else:
                        self.stdout.write(self.prompt)
                        self.stdout.flush()
                        line = self.stdin.readline()
                        if not len(line):
                            line = 'EOF'
                        else:
                            line = line[:-1] # chop \n
                line = self.precmd(line)
                stop = self.onecmd(line)
                stop = self.postcmd(stop, line)
当程序执行完一个用户命令后,会判断 cmdqueue 中是否有待执行的命令, 如果有, 则执行.
否则,等待用户输入下一个命令.
BTW:
self.preloop 循环前执行, 你可以在这里加载配置文件、修改 prompt、...
self.precmd 只是对用户的输入做了下预处理.
self.onecmd 负责执行命令.
self.postcmd 做执行命令后的一些操作.
self.postloop 循环结束时,执行的操作. 你可以在这里关闭日志、关闭数据库、say goodbye、...
于是我在自己的类中将 self.precmd 覆盖为:
    def precmd(self, line):
        cmdlist = line.split('&&')
        line = cmdlist[0] if len(cmdlist) >0 else ''
        self.cmdqueue += cmdlist[1:]
        return line
OK. 
效果如下:
$echo 1st && sleep 1 && echo 2nd
1st
2nd
1st
2nd
$
cmd 模块真的很棒 :)


本文来自ChinaUnix博客,如果查看原文请点:http://blog.chinaunix.net/u/10686/showart_2145873.html
您需要登录后才可以回帖 登录 | 注册

本版积分规则 发表回复

  

北京盛拓优讯信息技术有限公司. 版权所有 京ICP备16024965号-6 北京市公安局海淀分局网监中心备案编号:11010802020122 niuxiaotong@pcpop.com 17352615567
未成年举报专区
中国互联网协会会员  联系我们:huangweiwei@itpub.net
感谢所有关心和支持过ChinaUnix的朋友们 转载本站内容请注明原作者名及出处

清除 Cookies - ChinaUnix - Archiver - WAP - TOP