def main():
if len(sys.argv) < 2:
help()
else:
countArg = 0
for arg in sys.argv:
countArg += 1
if arg == "-i":
if sys.argv[countArg] == '-n':
fileName = sys.argv[countArg + 1]
template = template_agg()
init_single_file(f'{fileName}.cpp', template)
print(colored.yellow(f'Created {fileName}.cpp'))
break
elif sys.argv[countArg] == '-b':
if sys.argv[countArg + 1] == '-ne':
contestName = sys.argv[countArg + 2]
fileNames = get_fn_beginner()
init_noerror(contestName, fileNames)
elif sys.argv[countArg + 1] == '-t2':
contestName = sys.argv[countArg + 2]
fileNames = get_fn_beginner()
init_noerror_agg(contestName, fileNames)
else:
contestName = sys.argv[countArg + 1]
fileNames = get_fn_beginner()
init(contestName, fileNames)
elif sys.argv[countArg] == '-ne':
contestName = sys.argv[countArg + 1]
fileNames = get_filename()
init_noerror(contestName, fileNames)
elif sys.argv[countArg] == '-t2':
contestName = sys.argv[countArg + 1]
fileNames = get_filename()
init_noerror_agg(contestName, fileNames)
else:
contestName = sys.argv[countArg]
fileNames = get_filename()
init(contestName, fileNames)
elif arg == "-f":
contestName = sys.argv[countArg]
fileNames = get_filename()
create_only_files(contestName, fileNames)
elif arg == "-v":
print(
colored.magenta(
'CODEIt v0.0.3 (c)2020 - 2021, alphaX86 under Arch-1'))
elif arg == "-h":
help()
def create_only_files(contestName, fileNames):
try:
print(colored.blue('Making files for ' + colored.magenta(contestName)))
os.makedirs(os.path.join(os.getcwd(), contestName))
except OSError:
print(
colored.red(
"Failed! This directory cannot be created as it exists."))
else:
print(colored.yellow('Directory is made'))
# create files for contest (should be 6 cpp files)
for files in range(len(fileNames)):
write_to_file(fileNames[files], template_agg(), contestName)
print(colored.green('Files have been created. Happy Coding!'))
def init_noerror_agg(contestName, fileNames):
try:
print(colored.blue('Making files and folders for ' + colored.magenta(contestName)))
os.makedirs(os.path.join(os.getcwd(), contestName))
except OSError:
print(colored.red("Failed! This directory cannot be created as it exists."))
else:
print(colored.yellow('Directory is made'))
# create files for contest (should be 6 cpp files)
for files in range(len(fileNames)):
write_to_file(fileNames[files], template_agg(), contestName)
# create input file
write_to_file('input.txt', '', contestName)
write_to_file('output.txt', '', contestName)
print(colored.green('Files have been created without error TXT file. Happy Coding!'))