Added uninstall function to setup

This commit is contained in:
BrianPetkovsek 2019-06-17 12:05:06 -04:00
parent 81af7b1c1b
commit d25857a178

View File

@ -89,6 +89,7 @@ class Video2xSetup:
def _install_python_requirements(self): def _install_python_requirements(self):
""" Read requirements.txt and return its content """ Read requirements.txt and return its content
""" """
pip_uninstall('requirements.txt')
pip_install('requirements.txt') pip_install('requirements.txt')
def _cleanup(self): def _cleanup(self):
@ -203,12 +204,19 @@ def download(url, save_path, chunk_size=4096):
def pip_install(file): def pip_install(file):
""" Install python package via python pip module """ Install python package via python pip module
pip.main() is not available after pip 9.0.1, thus pip.main() is not available after pip 9.0.1, thus
pip module is not used in this case. pip module is not used in this case.
""" """
return subprocess.run([sys.executable, '-m', 'pip', 'install', '-U', '-r', file]).returncode return subprocess.run([sys.executable, '-m', 'pip', 'install', '-U', '-r', file]).returncode
def pip_uninstall(file):
""" Uninstall python package via python pip module
pip.main() is not available after pip 9.0.1, thus
pip module is not used in this case.
"""
return subprocess.run([sys.executable, '-m', 'pip', 'uninstall', '-r', file, '-y']).returncode
if __name__ == '__main__': if __name__ == '__main__':
try: try: