全国绿色计算大赛（第一阶段） 模拟赛Python 第3关：图片查看器

图片查看器

在第二题的基础上选择后缀为图片类型的即可

在第二题解的基础上添加了一个判断

import os, sys

class Task:
    def showDirTree(self,path):
########## BEGIN ##########
        def list_search(path_1,num):

            tup=['jpg','png','bmp']
            if os.path.isdir(path_1):
                print(' '*num+'+--'+path_1.split('/')[-1])

                for i in os.listdir(path_1):
                    list_search(path_1+'/'+i,num+2)
            else:
                if tup.count(path_1.split('.')[-1])==1:#新加的判断条件
                    print(' '*num+'--'+path_1.split('/')[-1])
        list_search(path,0)


########## END ##########