# -*- coding: utf-8 -*-

import os
import os.path
import numpy as np

rootdir1 = "./stage2/"


parent1 = []
dirname1 = []
filename1 = []



def get_file(rootdir,parent,dirname,filename):
#返回父目录、所有文件夹的名字、所有文件名
    for Parent,Dirnames,Filenames in os.walk(rootdir):
        for Dirname in Dirnames:
                # print "parent is :" + parent
                # parent.append(Parent)
                # print "dirname is :" + dirname
                dirname.append(Dirname)

        for Filename in Filenames:
            # print "parent is :" + parent
            parent.append(Parent)
            # print "filename is :" + filename
            filename.append(Filename)
            # print "the full name of the file is:" + os.path.join(parent,filename)

    return parent,dirname,filename

get_file(rootdir1,parent1,dirname1,filename1)

# print parent1
print dirname1
# print filename1
np.save('./108.npy',dirname1)
# print [file for file in filename1 if file in filename2]
print len(dirname1)