使用Python实现简单Linux之find命令

shenzhang

#!/usr/bin/python　　
#*-*coding:utf8*-*
　　

　　
from optparse import OptionParser
　　
import os
　　
import sys
　　

　　
#使用选项帮助信息可以使用中文
　　
reload(sys)
　　
sys.setdefaultencoding("utf-8")
　　

　　
#定义选项以及命令使用帮助信息
　　
usage = sys.argv[0] + " Directory Options\n\n例："+sys.argv[0] + " /etc --type f --name passwd\n\n注意：选项和目录益可随意调换，可以写多个目录，会从多个目录中进行查找"
　　
parser = OptionParser(usage)
　　
parser.add_option("--type",
　　
                  dest="filetype",
　　
                  action="store",
　　
                  default=False,
　　
                  help="指定查找对象的类型,文件类型可以是 d:代表目录 f:代表文件")
　　
parser.add_option("--name",
　　
                  dest="name",
　　
                  action="store",
　　
                  default=False,
　　
                  help="指定查找对象的名称,文件或目录全名")
　　
options, args = parser.parse_args()
　　

　　
def find(dir):
　　
    directory = os.walk(dir)
　　
    for x, y, z in directory:
　　
        if options.filetype == "f":
　　
            for name in z:
　　
                if name == options.name:
　　
                    path = os.path.join(x,name)
　　
                    print(path)
　　
        if options.filetype == "d":
　　
            for name in y:
　　
                if name == options.name:
　　
                    path = os.path.join(x,name)
　　
                    print(path)
　　

　　
#判断目录是否存在，并且是否为目录
　　
for dir in args:
　　
    if os.path.exists(dir) == False:
　　
        sys.stderr.write(dir+" is not found\n")
　　
        exit(100)
　　

　　
    if os.path.isfile(dir):
　　
        sys.stderr.write(dir+" is a file\n")
　　
        exit(101)
　　
#判断--type选项是否正确，只能跟 f 或者 d
　　
if not (options.filetype == "f" or options.filetype == "d"):
　　
    sys.stderr.write("--type only support d or f\n")
　　
    exit(102)
　　

　　
if __name__ == "__main__":
　　
    for dir in args:
　　
        find(dir)