def cleanup_temp()

in src/helper.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def cleanup_temp(temp_location):
    """This will cleanup a temporary location

    Args:
        temp_location (str): Directory location to remove files/directories

    Returns:
        N/A
    """
    for f in os.listdir(temp_location):
        logger.info('Removing {} from {}'.format(f, temp_location))
        if os.path.isdir(f"{temp_location}/{f}"):
            shutil.rmtree(f"{temp_location}/{f}")

        else:
            os.remove(os.path.join(temp_location, f))