# -*- coding: utf-8 -*-
import os

class GetImageTool(object):

    def __init__(self):
        self.imagearray = [];
        self.imageDic = {};

    # 遍历指定目录，显示目录下的所有文件名
    def eachFile(self,filepath):
        if os.path.isdir(filepath):
            # print("it's a directory")
            pathDir = os.listdir(filepath)
            for allDir in pathDir:
                if allDir == ".DS_Store":
                    continue
                child = os.path.join(filepath, allDir)
                # print(child)  # .decode('gbk')是解决中文显示乱码问题
                self.eachFile(child)
        elif os.path.isfile(filepath):
            lastPath = os.path.split(filepath)[1]
            suffix = os.path.splitext(lastPath)[1]
            lastPathName = os.path.splitext(lastPath)[0]
            if suffix ==".png"or suffix==".jpg":
                self.imagearray.append(lastPathName)
                self.imageDic[lastPathName]=filepath;
        else:
            print("it's a special file (socket, FIFO, device file)")
            print(filepath)


    def getImageArrayWithFilePath(self,path):
        self.eachFile(path)
        return self.imagearray