Automate folder creation for different processes

create_folders.py

import os
import sys
"""
Author: Huziy
Purpose: Automate folder creation for different processes
"""

def main():
    args = sys.argv

    if len(args) < 2:
        raise Exception("specify number of processes: python {0} {1}".format(args[0], "nprocs"))


    try:
        nprocs = int(args[1])
    except Exception:
        raise ValueError( "The number of processes should be integer")


    for i in range(1, nprocs + 1):
        name = "{0:05d}".format(i)
        if not os.path.isdir(name):
            os.mkdir(name)


if __name__ == "__main__":
    main()