from FileList import FileList

import yaml
import pprint  # for better print

# Set Path
input_folder_prefix = os.path.join(os.path.dirname(__file__), "txt")
input_foldername_prefix = "dayTraining_"

output_folder_prefix = os.path.join(os.path.dirname(__file__), "yaml")

for i in range(1, 3):
	input_folder = os.path.join(input_folder_prefix, input_foldername_prefix) + str(i)
	flist = FileList(input_folder)

	for item in flist.getFilesPathList():
		# for every .txt
		# print(item)

		# Read content
		item_list = []
		with open(item, "r") as rf:
			for row in rf:
				item_list.append(row)

		output_yaml = {'annotation': {'filename': '000001.jpg',
									  'folder': 'VOC2007',
									  'object': [],
									  'owner': {'flickrid': 'Fried Camels',
												'name': 'Jinky the Fruit Bat'},
									  'segmented': '0',