## TEX ##

import xml.etree.ElementTree as ET
import sys
import enum
import os
import shutil

def main():
	xml = ''

	# parse xml file from argument
	try:
		xml = sys.argv[1]
		tree = ET.parse(xml)
		root = tree.getroot()
	except IndexError:
		print('Error: Please provide an argument to ' + sys.argv[0])
		sys.exit(1)
	except IOError:
		print('Error: \'' + sys.argv[1] + '\' does not exist!')
		sys.exit(2)

	with open('C:\\Documents and Settings\\sveeion\\Desktop\\a.tex', 'wb') as f:

		timtotal = 0

		for tim in root.findall('tim'):
			timtotal += 1
		f.write(timtotal.to_bytes(4, byteorder='little', signed=False))

		for tim in root.findall('tim'):
			size = os.path.getsize(tim.attrib['path']).to_bytes(4, byteorder='little', signed=False)
			f.write(size)

		for tim in root.findall('tim'):
			src = open(tim.attrib['path'], 'rb')
			shutil.copyfileobj(src, f)
			src.close()

		print('TEX file written to desktop!\n')

if __name__ == '__main__':
	main()