#!/Users/avhirupchakraborty/Programming/anaconda3/bin/python3
import argparse

import os
import sys
import argparse

class Organizer(object):
	"""Class to organize a data science project"""
	folders=['analysis','data','logs','logs/train_logs','logs/tensorboard_logs','model','model_weights','config','pipeline','pipeline/cleaning','pipeline/encoding']

	def __init__(self, project_name,path):
		super(Organizer, self).__init__()
		self.path = path
		self.project_name=project_name


	def make_folders(self):
		"""
		Makes folders for experimentation
		"""
		for name in self.folders:
			os.makedirs(self.path+"/"+name,exist_ok=True)



if __name__ == '__main__':
	parser = argparse.ArgumentParser()
	parser.add_argument("--project_name",help="Project name",type=str,required=True)
	parser.add_argument("--path", help="path to the project ",required=True)
	parser.add_argument('--op',
                      choices=['make_folder','sweep','c'],
                      help='methods that can be done',required=True)

	args = parser.parse_args()
	organizer=Organizer(args.project_name,args.path)
	if args.op=='make_folder':
		organizer.make_folders()

