《可爱的Python》读书笔记（三）

James

　　先用后学，快速获得体验，然后寻求理论支持。
　　小白根据已有的体验，对PyCDC的软件需求进行了进一步完善。
　　将以往验证想法的代码，变成可以方便使用的功能，让它可以重复在不同应用环境中使用。小白想象着自个儿的PyCDC可以像普通的命令行工具一样来使用。
　　行者提出“可以使用getopt.getopt()优化当前功能函式”
# -*- coding: utf-8 -*-　　
import os
　　
import sys
　　
import getopt
　　

　　

　　
CDROM = 'E:\\'
　　
def cdWalker(cdrom,cdcfile):
　　
    export = ""
　　
    for root, dirs, files in os.walk(cdrom):
　　
        export += "\n %s;%s;%s" % (root,dirs,files)
　　
    open(cdcfile, 'w').write(export)
　　
#运行命令行工具的帮助提示
　　
def usage():
　　
    print('Usage:python %s -e filename' % sys.argv[0])
　　

　　
try:
　　
    opts, args = getopt.getopt(sys.argv[1:], 'e:')
　　
    #内有输入参数将显示帮助提示，并退出程序
　　
    if len(opts) == 0:
　　
        usage()
　　
        sys.exit()
　　
    for opt, arg in opts:
　　
        if opt == '-e':
　　
            cdWalker(CDROM, arg)
　　
            print("记录光盘信息到 %s" % arg)
　　
        else:
　　
            usage()
　　
            sys.exit()
　　
except getopt.GetoptError:
　　
    usage()
　　
    sys.exit()
　　运行效果：
C:\>python pycdc.pyw　　
Usage:python pycdc.pyw -e filename
　　
C:\>python pycdc.pyw -e test.txt
　　
记录光盘信息到 test.txt
　　查看test.txt内容如下：
　　E:\;['EFI', 'images', 'isolinux', '[BOOT]'];['.discinfo', '.treeinfo', 'CentOS_BuildTag', 'EULA', 'GPL', 'RELEASE-NOTES-en-US.html', 'RPM-GPG-KEY-CentOS-6', 'RPM-GPG-KEY-CentOS-Debug-6', 'RPM-GPG-KEY-CentOS-Security-6', 'RPM-GPG-KEY-CentOS-Testing-6', 'TRANS.TBL']
　　E:\EFI;['BOOT'];['TRANS.TBL']
　　E:1\EFI\BOOT;[];['BOOTX64.conf', 'BOOTX64.efi', 'splash.xpm.gz', 'TRANS.TBL']
　　E:\images;['pxeboot'];['efiboot.img', 'efidisk.img', 'install.img', 'TRANS.TBL']
　　E:\images\pxeboot;[];['initrd.img', 'TRANS.TBL', 'vmlinuz']
　　E:\isolinux;[];['boot.cat', 'boot.msg', 'grub.conf', 'initrd.img', 'isolinux.bin', 'isolinux.cfg', 'memtest', 'splash.jpg', 'TRANS.TBL', 'vesamenu.c32', 'vmlinuz']
　　E:\[BOOT];[];['1-Boot-NoEmul.img', '2-Boot-NoEmul.img']
　　小白虽然不是非常明白getopt的使用，也还是依葫芦画瓢写了出来，基本实现了想要的功能。
　　总结：今天学习了使用简单的getopt实现基本的命令行工具，使用了try...except处理异常，使得程序变得更方便使用