Làm thế nào để cài đặt một gói python thiếu từ bên trong kịch bản cần nó?

Question

Giả sử rằng bạn đã cài đặt pip hoặc easy_install trên bản phân phối python của mình, tôi muốn biết cách cài đặt gói yêu cầu trong thư mục người dùng từ bên trong tập lệnh.

Từ những gì tôi biết pip cũng là một mô-đun python vì vậy giải pháp nên hình như:

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

Những gì tôi đang mất tích đang làm "PIP cài đặt zumba --user" từ bên trong python, tôi không muốn làm điều đó bằng cách sử dụng os.system() vì điều này có thể tạo ra các sự cố khác.

tôi giả sử nó có thể ...

Answer 1

Nhờ @Joop tôi đã có thể đi lên với câu trả lời thích hợp.

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

Một lưu ý quan trọng là việc này sẽ hoạt động mà không yêu cầu quyền truy cập root vì nó sẽ cài đặt mô-đun trong thư mục người dùng.

Không chắc chắn nếu nó sẽ làm việc cho mô-đun nhị phân hoặc những người sẽ yêu cầu biên dịch, nhưng nó hoạt động rõ ràng cho các mô-đun tinh khiết-python.

Bây giờ bạn có thể viết các tập lệnh khép kín và không phải lo lắng về các phụ thuộc.

Answer 2

Tôi muốn lưu ý rằng câu trả lời được chấp nhận hiện tại có thể dẫn đến xung đột tên ứng dụng có thể xảy ra. Nhập từ không gian tên ứng dụng không cung cấp cho bạn bức tranh đầy đủ về những gì được cài đặt trên hệ thống.

Cách tốt hơn sẽ là:

import pip 

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

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