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 中就可以了。