#!/usr/bin/python

#*-*coding:utf8*-*

from optparse import OptionParser

import os

import sys

#使用選項(xiàng)幫助信息可以使用中文

reload(sys)

sys.setdefaultencoding("utf-8")

#定義選項(xiàng)以及命令使用幫助信息

usage = sys.argv[0] + " Directory Options\n\n例："+sys.argv[0] + " /etc --type f --name passwd\n\n注意：選項(xiàng)和目錄益可隨意調(diào)換，可以寫多個(gè)目錄，會從多個(gè)目錄中進(jì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選項(xiàng)是否正確，只能跟 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)