#!/usr/bin/python
# -*- coding:utf-8 -*-
 
import os
# 这步在创建文件夹xim,这个文件夹下面还有images文件夹和itst文件夹
def mkdir(pathx):
 
	p = os.path.exists(pathx)
 
	if not p:                   #判断是否存在文件夹如果不存在则创建为文件夹
		os.makedirs(pathx)            #makedirs 创建文件时如果路径不存在会创建这个路径
		print("---  new p...  ---")
		print("---  OK  ---")
 
	else:
		print("---  There is this p!  ---")
		
x = "/Users/sunchanglan/Desktop/ml/test/xim"
        #调用函数

def generate(dir,label):
	files = os.listdir(dir)#返回path指定的文件夹包含的文件或文件夹的名字的列表。
	files.sort()
	print('****************')
	print('input :',dir)
	print('start...')
	listText = open('allof.txt','a')
	for file in files:
		fileType = os.path.split(file)
		if fileType[1] == '.txt':
			continue
		name = file + ' ' + str(int(label)) +'\n'
		listText.write(name)
	listText.close()
	print('down!')
	print('****************')


outer_path = '/Users/sunchanglan/Desktop/ml/test/xim'   #这里是你的图片的目录


if __name__ == '__main__':
	mkdir(x)     
	i = 0
	folderlist = os.listdir(outer_path)          #列举文件夹
	print(folderlist)
	for folder in folderlist:
		generate(os.path.join(outer_path,folder),i)
		# print(folder)
		i += 1