parent
26a30c6f01
commit
6ecc22bd3d
@ -0,0 +1,35 @@ |
||||
#!/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() |
Loading…
Reference in new issue