AddMonths.py

import datetime
import calendar

def AddMonths(DATE_VAL, 
              ADD_N,
              DATE_FORMAT = "%Y-%m-%d"):

    print("AddMonths: Function initalized")

    FUNCTION_OUTPUT = dict()

    FUNCTION_OUTPUT['ORIGINAL_DATE'] = DATE_VAL
    FUNCTION_OUTPUT["ADD_N"] = ADD_N

    START_DATE = datetime.datetime.strptime(DATE_VAL, DATE_FORMAT)

    DAYS_IN_MONTHS = calendar.monthrange(START_DATE.year, START_DATE.month)[1]

    NEW_DATE = START_DATE + datetime.timedelta(days = DAYS_IN_MONTHS)

    FUNCTION_OUTPUT['NEW_DATE'] = NEW_DATE.strftime(DATE_FORMAT) 

    print("AddMonths: Function run completed")

    return FUNCTION_OUTPUT