Jak zainstalować brakujący pakiet Pythona z poziomu skryptu, który tego potrzebuje?

Question

Zakładając, że masz już zainstalowaną pip lub easy_install na swojej dystrybucji Pythona, chciałbym wiedzieć, w jaki sposób mogę zainstalować wymagany pakiet w katalogu użytkownika z samego skryptu.

Z tego co wiem, pip jest również moduł Pythona, więc rozwiązanie powinno wyglądać następująco:

try: 
    import zumba 
except ImportError: 
    import pip 
    # ... do "pip install --user zumba" or throw exception <-- how? 
    import zumba 

Co mi brakuje robi „pip zainstalować --user zumba” od wewnątrz pytona, nie wiem chcesz to zrobić, używając os.system(), ponieważ może to spowodować inne problemy.

Zakładam, że jest to możliwe ...

Answer 1

Dzięki @Joop udało mi się przyjść się z właściwą odpowiedzią.

try: 
    import zumba 
except ImportError: 
    import pip 
    pip.main(['install', '--user', 'zumba']) 
    import zumba 

Jedna ważna uwaga jest taka, że ​​będzie działać bez wymagania uprawnień administratora, ponieważ zainstaluje moduł w katalogu użytkownika.

Nie wiem, czy zadziała w przypadku modułów binarnych lub takich, które wymagałyby kompilacji, ale wyraźnie sprawdza się w przypadku modułów czysto-pythonowych.

Teraz możesz pisać samodzielne skrypty i nie martwić się o zależności.

Answer 2

Pragnę zauważyć, że bieżąca zaakceptowana odpowiedź może spowodować kolizję nazwy aplikacji. Importowanie z przestrzeni nazw aplikacji nie daje pełnego obrazu tego, co jest zainstalowane w systemie.

Lepszym sposobem byłoby:

import pip 

packages = [package.project_name for package in pip.get_installed_distributions()] 

if 'package' not in packages: 
    pip.main(['install', 'package'])