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

Posted 2020-09-20

实现代码如下：

#!/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)

实现功能如下：

运行结果如下：

本文出自 “蓝色_风暴” 博客，请务必保留此出处http://270142877.blog.51cto.com/12869137/1931011