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