开发学院软件开发Python 探索 Pexpect,第 2 部分:Pexpect 的实例分析 阅读

探索 Pexpect,第 2 部分:Pexpect 的实例分析

 2009-08-29 00:00:00 来源:WEB开发网   
核心提示: 注:运行后,输出结果为: Congratulations!ftplogincorrect!gettingafile...successfullyreceivedthefile本例 expect 函数中的 pattern 使用了 List,探索 Pexpect,第 2 部分:Pexpect 的实例

注:

运行后,输出结果为:

Congratulations! ftp login correct! 
getting a file... 
successfully received the file

本例 expect 函数中的 pattern 使用了 List,并包含了 pexpect.EOF和pexpect.TIMEOUT,这样出现了超时或者 EOF,不会抛出 expection 。(关于 expect() 函数的具体使用,请参阅参考资料)

如果程序运行中间出现了错误,如用户名密码错误,超时或者 EOF,远程 server 连接不上,都会使用 c hild.close(force=True) 关掉 ftp 子程序。调用 close 可以用来关闭与子程序的 connection 连接,如果你不仅想关闭与子程序的连接,还想确保子程序是真的被 terminate 终止了,设置参数 force=True,其最终会调用 c hild.kill(signal.SIGKILL) 来杀掉子程序。

例 2:记录 log

本例实现了如下功能:运行一个命令,并将该命令的运行输出结果记录到 log 文件中 ./command.py [-a] [-c command] {logfilename} -c 后接的是要运行的命令的名字,默认是“ls -l”; logfilename 是记录命令运行结果的 log 文件名,默认是“command.log”;指定 -a 表示命令的输出结果会附加在 logfilename 后,如果 logfilename 之前已经存在的话。

清单 2. 记录 log 的例子代码

#!/usr/bin/env python 
""" 
This run a user specified command and log its result. 
 
./command.py [-a] [-c command] {logfilename} 
 
logfilename : This is the name of the log file. Default is command.log. 
-a : Append to log file. Default is to overwrite log file. 
-c : spawn command. Default is the command 'ls -l'. 
 
Example: 
 
This will execute the command 'pwd' and append to the log named my_session.log: 
 
./command.py -a -c 'pwd' my_session.log 
 
""" 
import os, sys, getopt 
import traceback 
import pexpect 
 
# 如果程序中间出错,打印提示信息后退出 
def exit_with_usage(): 
  print globals()['__doc__'] 
  os._exit(1) 
 
def main(): 
  ###################################################################### 
  # Parse the options, arguments, get ready, etc. 
  ###################################################################### 
  try: 
    optlist, args = getopt.getopt(sys.argv[1:], 'h?ac:', ['help','h','?']) 
  # 如果指定的参数不是’ -a ’ , ‘ -h ’ , ‘ -c ’ , ‘ -? ’ , ‘ --help ’ , 
  #‘ --h ’或’ --? ’时,会抛出 exception, 
  # 这里 catch 住,然后打印出 exception 的信息,并输出 usage 提示信息. 
  except Exception, e: 
    print str(e) 
    exit_with_usage() 
  options = dict(optlist) 
  # 最多只能指定一个 logfile,否则出错. 
  if len(args) > 1: 
    exit_with_usage() 
  # 如果指定的是 '-h','--h','-?','--?' 或 '--help',只输出 usage 提示信息. 
  if [elem for elem in options if elem in ['-h','--h','-?','--?','--help']]: 
    print "Help:" 
    exit_with_usage() 
  # 获取 logfile 的名字. 
  if len(args) == 1: 
    script_filename = args[0] 
  else: 
  # 如果用户没指定,默认 logfile 的名字是 command.log 
    script_filename = "command.log" 
  # 如果用户指定了参数 -a,如果之前该 logfile 存在,那么接下来的内容会附加在原先内容之后, 
  # 如果之前没有该 logfile,新建一个文件,并且接下来将内容写入到该文件中. 
  if '-a' in options: 
    fout = open (script_filename, "ab") 
  else: 
  # 如果用户没指定参数 -a,默认按照用户指定 logfile 文件名新建一个文件,然后将接下来将内容写入到该文件中. 
    fout = open (script_filename, "wb") 
  # 如果用户指定了 -c 参数,那么运行用户指定的命令. 
  if '-c' in options: 
    command = options['-c'] 
  # 如果用户没有指定 -c 参数,那么默认运行命令'ls – l' 
  else: 
    command = "ls -l" 
 
  # logfile 文件的 title 
  fout.write ('==========Log Tile: IBM developerWorks China==========\n') 
 
  # 为接下来的运行命令生成一个 pexpect 的 spawn 类子程序的对象. 
  p = pexpect.spawn(command) 
  # 将之前 open 的 file 对象指定为 spawn 类子程序对象的 log 文件. 
  p.logfile = fout 
  # 命令运行完后,expect EOF 出现,这时会将 spawn 类子程序对象的输出写入到 log 文件. 
  p.expect(pexpect.EOF) 
  #open 完文件,使用完毕后,需关闭该文件. 
  fout.close() 
  return 0 
 
if __name__ == "__main__": 
  try: 
    main() 
  except SystemExit, e: 
    raise e 
  except Exception, e: 
    print "ERROR" 
    print str(e) 
    traceback.print_exc() 
    os._exit(1)

上一页  1 2 3 4 5 6 7 8  下一页

Tags:探索 Pexpect 部分

编辑录入:爽爽 [复制链接] [打 印]
[]
  • 好
  • 好的评价 如果觉得好,就请您
      0%(0)
  • 差
  • 差的评价 如果觉得差,就请您
      0%(0)
赞助商链接