import os
def batch_DSSP():
	input_dir = '../data/PDB'
	input_ext = '.pdb'
	command = 'dssp'
	output_dir = '../data/DSSP'
	output_ext = '.dssp'


	# fetch list of all files in input_dir
	files = [ f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir,f))]

	for f in files:
		# only want files that match extension
		if (input_ext not in f):
			continue
		print (f)
		# construct name of input/ output files
		file_parts = f.split('.')
		input_file = os.path.join(input_dir, f)
		output_file = os.path.join(output_dir, file_parts[0] + output_ext)

		# run command
		shell_cmd = command + ' -i ' + input_file + ' -o ' + output_file
		os.system(shell_cmd)


def batch_FEATURE():
	input_dir = '../data/ptf'
	input_ext = '.ptf'
	command = 'featurize'
	output_dir = '../data/ff'
	output_ext = '.ff'
	files = [ f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir,f))]
	for f in files:
		# only want files that match extension
		if (input_ext not in f):
			continue
		print (f)
		# construct name of input/ output files
		file_parts = f.split('.')
		input_file = os.path.join(input_dir, f)
		output_file = os.path.join(output_dir, file_parts[0] + output_ext)
		# run command
		shell_cmd = command + ' -P ' + input_file + ' > ' + output_file
		os.system(shell_cmd)


if __name__ == '__main__':

	batch_DSSP()
	batch_FEATURE()