Python实现Linux的find命令实例分享

Python  /  管理员 发布于 7年前   432

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

代码如下：

#!/usr/bin/python#*-*coding:utf8*-*from optparse import OptionParserimport osimport 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 或者 dif 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)

运行结果如下：