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