roundGen.py

import os
import sys
import shutil


arg_list = sys.argv

directory = arg_list[1]
# Put the working directory here
par_dir = "/home/mayankk64/Documents/Competivive Programming"
path = os.path.join(par_dir, directory)
try:
    os.mkdir(path)
except OSError as error:
    print("The Folder " + directory + " already exists!!")
    sys.exit(0)


# Put Template path here
template_path = "/home/mayankk64/.vscode/extensions/bam.vscode-file-templates-1.1.0/templates/CPP.cpp"

file_num = int(arg_list[2])

if file_num <= 26:
    created_files = ""

    for idx in range(65, 65+file_num):
        res = str(chr(idx))+str(".cpp")
        created_files = created_files + res+" "
        target_path = os.path.join(path, res)
        shutil.copyfile(template_path, target_path)

    print("Done!!\nCreated : " + created_files)
else:
    print("Too Many Files!!")


if arg_list.__len__() == 4:
    if str(arg_list[3]) == "vscode":
        print("Opening VSCode.....")
        os.system('code "' + path + '"')
    else:
        sys.exit(0)
else:
    sys.exit(0)

# Command format: python3 roundGen.py <round number> <number of files> <vscode>(to open in vscode) or leave blank