CDays–2 完成核心功能 CMD模块 Python基础教程 cmd cli

再过两个CDays我们就完成了所有的功能了，不过是在CMD中运行的。

为了模块化我们的程序，我们先整理一下以前的程序。

# -*- coding: utf-8 -*-
import os
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)

if __name__ == '__main__':      # this way the module can be
    CDROM = 'D:\\CDROM'

这个模块完成了CDROM的遍历，并存进了指定文件中。

虽然这个程序没有GUI，那么让我们给他设计一个CMD运行的界面吧。

根据上一次的日志，我们发现根据命令不同会有很多很多的分支，那么我们看一下书上给的例子pycdc-v0.4.py。

# -*- coding: utf-8 -*-
'''pycdc-v0.4.py
Lovely Python -2 PyDay 
'''
import sys, cmd
class PyCDC(cmd.Cmd):
    def __init__(self):
        cmd.Cmd.__init__(self)                # initialize the base class
        self.CDROM = 'D:\\CDROM'
        self.CDDIR = 'D:\\'

    def help_EOF(self):
        print "退出程序 Quits the program"
    def do_EOF(self, line):
        sys.exit()

    def help_walk(self):
        print "扫描光盘内容 walk cd and export into *.cdc"
    def do_walk(self, filename):
        if filename == "":filename = raw_input("输入cdc文件名:: ")
        print "扫描光盘内容保存到:'%s'" % filename

    def help_dir(self):
        print "指定保存/搜索目录"
    def do_dir(self, pathname):
        if pathname == "": pathname = raw_input("输入指定保存/搜索目录: ")
        print "指定保存/搜索目录:'%s' ;默认是:'%s'" % (pathname,self.CDDIR)

    def help_find(self):
        print "搜索关键词"
    def do_find(self, keyword):
        if keyword == "": keyword = raw_input("输入搜索关键字: ")
        print "搜索关键词:'%s'" % keyword

if __name__ == '__main__':      # this way the module can be
    cdc = PyCDC()            # imported by other programs as well
    cdc.cmdloop()

根据书上所述，我们不知道上面的一些语法是什么，的确，我们确实不知道。

让我们运行一下.

提示输入.

输入非法字符有提示.

输入合法字符, 好吧,程序里没有,我承认我已经看晕了.

让我们看一下书,书上说是用的CMD模块, 查一下cmd是干嘛的.

根据作弊条PCS201

cmd模块为命令行接口cli提供了一个简单的框架.下面给出了一个例子.

# -*- coding: utf-8 -*-
import cmd
import string, sys

class CLI(cmd.Cmd):

    def __init__(self):
        cmd.Cmd.__init__(self)
        self.prompt = '> '    # 定义命令行提示符

    def do_hello(self, arg):   # 定义hello命令所执行的操作
        print "hello again", arg, "!"

    def help_hello(self):        # 定义hello命令的帮助输出
        print "syntax: hello [message]",
        print "-- prints a hello message"

    def do_quit(self, arg):     # 定义quit命令所执行的操作
        sys.exit(1)

    def help_quit(self):        # 定义quit命令的帮助输出
        print "syntax: quit",
        print "-- terminates the application"

    # 定义quit的快捷方式
    do_q = do_quit

# 创建CLI实例并运行
cli = CLI()
cli.cmdloop()

根据这个例子中的注释,我们可以知道这个模块大部分的用法。作为CDays-2的要求，我们知道怎么用就可以了。

我们重新分析pycdc-v0.4.py

没有定义命令提示符

do_命令   —————— 是该命令执行的函数

help_命令  —————— 是该命令的帮助函数

cmdloop( )  的意思是能自动返回cmd输入命令状态。

---

让我们加入自己编写的模块好了。

# -*- coding: utf-8 -*-
'''pycdc-v0.5.py
Lovely Python -2 PyDay 
'''
import sys, cmd
from cdctools import *
class PyCDC(cmd.Cmd):
    def __init__(self):
        cmd.Cmd.__init__(self)                # initialize the base class
        self.CDROM = 'D:\\CDROM'
        self.CDDIR = 'D:\\'
        self.prompt="(PyCDC)>"
        self.intro = '''PyCDC 0.5 使用说明:
    dir 目录名     # 指定保存和搜索目录，默认是 "cdc"
    walk 文件名    # 指定光盘信息文件名，使用 "*.cdc"
    find 关键词    # 遍历搜索目录中所有.cdc文件，输出含有关键词的行
    help          # 查询
    EOF           # 退出系统，也可以使用Crtl+D(Unix)|Ctrl+Z(Dos/Windows)
        '''

    def help_EOF(self):
        print "退出程序 Quits the program"
    def do_EOF(self, line):
        sys.exit()

    def help_walk(self):
        print "扫描光盘内容 walk cd and export into *.cdc"
    def do_walk(self, filename):
        if filename == "":filename = raw_input("输入cdc文件名:: ")
        print "扫描光盘内容保存到:'%s'" % filename
        cdWalker(self.CDROM,self.CDDIR+filename)

    def help_dir(self):
        print "指定保存/搜索目录"
    def do_dir(self, pathname):
        if pathname == "": pathname = raw_input("输入指定保存/搜索目录: ")
        self.CDDIR = pathname
        print "指定保存/搜索目录:'%s' ;默认是:'%s'" % (pathname,self.CDDIR)

    def help_find(self):
        print "搜索关键词"
    def do_find(self, keyword):
        if keyword == "": keyword = raw_input("输入搜索关键字: ")
        print "搜索关键词:'%s'" % keyword
        cdcGrep(self.CDDIR,keyword)

if __name__ == '__main__':      # this way the module can be
    cdc = PyCDC()            # imported by other programs as well
    cdc.cmdloop()

到此为止，我们已经基本完成了CDC的除了查找外的所有功能了。

现在让我们解决查找问题。

根据当时我们保存的内容，我们将每个文件保存为一行，所以我们可以一行一行的寻找。

def cdcGrep(cdcpath,keyword):
    filelist = os.listdir(cdcpath)          # 搜索目录中的文件
    for cdc in filelist:                    # 循环文件列表
        cdcfile = open(cdcpath+cdc)         # 拼合文件路径，并打开文件
        for line in cdcfile.readlines():    # 读取文件每一行，并循环
            if keyword in line:             # 判定是否有关键词在行中
                print line                  # 打印输出

把这段程序放在cdctools.py 中就可以了。