使用Python实现简单Linux的find命令
代码如下:
#!/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)
运行结果如下:
Copyright© 2013-2020
All Rights Reserved 京ICP备2023019179号-8