-
Notifications
You must be signed in to change notification settings - Fork 0
/
package-installer.py
55 lines (50 loc) · 2.03 KB
/
package-installer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from googlesearch import search
from bs4 import BeautifulSoup
from subprocess import run
from colorama import Fore, Style, init
import sys, getopt, requests, os
def parse_inputs(opts):
for opt, arg in opts:
if opt == '-h':
print(Fore.YELLOW + 'python package-installer.py -e <conda environment> -p <package name>' + Fore.RESET)
sys.exit()
elif opt in ("-e"):
env = arg
elif opt in ("-p"):
query = "How to install {package_name} conda".format(package_name=arg)
return env, query
def get_commands(query):
commands = []
for url in search(query, tld="com", num=10, stop=10, pause=2):
if "anaconda.org" in url :
print("using commands from: " + url + '\n')
html_text = requests.get(url).text
soup = BeautifulSoup(html_text, "html.parser")
for code in soup.find_all('code'):
commands.append(code.get_text().lstrip().rstrip())
break
return commands
def main():
try:
init(convert=True)
os.system("mode 360")
opts, _ = getopt.getopt(sys.argv[1:],"e:p:")
env, query = parse_inputs(opts)
commands = get_commands(query)
if len(commands) == 0:
print(Style.BRIGHT + Fore.RED + "\nPackage not found, type a more accurate name!" + Style.NORMAL + Fore.RESET)
else:
for command in commands:
cmd = command + " -y -n " + env
print(cmd)
userinput = input(Style.BRIGHT + Fore.RED + "Use this command? (y/n): " + Style.NORMAL + Fore.RESET)
if(userinput == "y"):
proc = run(cmd,
text=True, capture_output=True)
return print(proc.stdout)
print(Fore.LIGHTYELLOW_EX + "\nNo packages installed!\n" + Fore.RESET)
except getopt.GetoptError:
print(Fore.YELLOW + 'python package-installer.py -e <conda environment> -p <package name>' + Fore.RESET)
sys.exit(2)
if __name__ == "__main__":
main()