__main__.py

import argparse
import sys
from os.path import isdir, isfile

from file.PythonFiles import PythonFiles
from project.Project import Project


def get_user_inputs():
    """
    Function to get user inputs
    """
    # Input variable definition
    parser = argparse.ArgumentParser(formatter_class=argparse.RawTextHelpFormatter,
                                     description='Script which initiate the docstring documentation in a given python '
                                                 'file if it is not already existing.')
    parser.add_argument('input_file', help='path of the python file', type=str)
    return parser.parse_args()


def main(argv):
    user_inputs = get_user_inputs()
    if isfile(user_inputs.input_file):
        p1 = PythonFiles(user_inputs.input_file)

        first_level_function = p1.get_first_level_function_in_file()
        class_dict = p1.get_class_in_file()

        try:
            p1.write_first_level_function_docstring()
        except KeyError:
            print('No first level function')
            pass

        try:
            p1.write_class_docstring()
        except KeyError:
            print('No class')
            pass

    elif isdir(user_inputs.input_file):
        p1 = Project(user_inputs.input_file)
        p1.document_project()


if __name__ == "__main__":
    main(sys.argv[1:])