install_requirements.py

# -*- coding: utf-8 -*-
"""
Created on Fri Apr 24 20:13:05 2020

@author: Rudrajit
"""

# Importing libraries
import subprocess
import sys
import get_pip
import os
import importlib
import contextlib

def install(package):
    '''
    installs a package using pip
    :param package: string
    '''
    subprocess.call([sys.executable, "-m", "pip", "install", package])

required = []
failed = []

# Try opening requirements.txt and read all required packages
try:
    file = open("requirements.txt", "r")
    file_lines = file.readlines()
    required = [line.strip().lower() for line in file_lines]
    file.close()
except:
    print("[ERROR] No requirements.txt file found")

if len(required)>0:
    print("[INPUT] You are about to install", len(required), "packages, would you like to proceed (y/n):", end = " ")
    ans = input()

    if ans.lower() == "y":
        for package in required:
            try:
                print("[LOG] Looking for", package)
                with contextlib.redirect_stdout(None):
                    __import__(package)
                print("[LOG]",package, "is already installed, skipping...")
            except ImportError:
                print("[LOG]", package, "not installed")

                try:
                    print("[LOG] Trying to install", package, "via pip")
                    try:
                        import pip
                    except:
                        print("[EXCEPTION] Pip is not installed")
                        print("[LOG] Trying to install pip")
                        get_pip.main()
                        print("Pip has been installed")
                    
                    print("[LOG] Installing", package)
                    install(package)
                    with contextlib.redirect_stdout(None):
                        __import__(package)
                    print("[LOG]", package, "has been installed")
                except Exception as e:
                    print("[ERROR] Could not install", package, "-", e)
                    failed.append(package)
    
    else:
        print("[STOP] Operation terminated by user")
else:
    print("[LOG] No packages to install")

if len(failed)>0:
    print("[FAILED]", len(failed), "package(s) were not installed. Failed package install(s):", end = " ")
    for x, package in enumerate(failed):
        if x!=len(failed)-1:
            print(package, end = ",")
        else:
            print(package)