post_gen_project.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Post gen hook to ensure that the generated project
hase only one package managment, either pipenv or pip."""
import os
import shutil
import sys


def clean_extra_package_managment_files():
    """Removes either requirements files and folderor the Pipfile."""
    use_pipenv = '{{cookiecutter.use_pipenv}}'
    to_delete = []

    if use_pipenv == 'yes':
        to_delete = to_delete + ['requirements.txt', 'requirements']
    else:
        to_delete.append('Pipfile')

    try:
        for file_or_dir in to_delete:
            if os.path.isfile(file_or_dir):
                os.remove(file_or_dir)
            else:
                shutil.rmtree(file_or_dir)
        sys.exit(0)
    except OSError as e:
        sys.stdout.write(
            'While attempting to remove file(s) an error occurred'
        )
        sys.stdout.write('Error: {}'.format(e))


if __name__ == '__main__':
    clean_extra_package_managment_files()