install.py

import os 
import yaml
import sys 
 


required  = ['pip','matplotlib','numpy','scipy','pandas','seaborn','cython','pathos'] 


#print(sys.argv)
env_name = sys.argv[1]
print("The environment name you have chosen is :", env_name)
print("=======================================================================")
print("What packaged would you like to install for multiprocessing?")
print("The default mode (none) is single threaded mode, which may not be optimal for heavy simulations \n")

try:
    multi_pck = input("There are two options: pathos and ray. The CBGTPy has been modified to run on both packages. Pathos is installed by default. Do you want to install ray? (n/y):  ")
except NameError:
    multi_pck = raw_input("There are two options: pathos and ray. The CBGTPy has been modified to run on both packages. Pathos is installed by default. Do you want to install ray? (n/y):  ")

print(multi_pck)
if multi_pck == "y":
    required.insert(0,"ray")

os.system("conda env export --name "+env_name+" --file environment.yml")

# Add the rest of the packages
with open('environment.yml') as f:
    doc = yaml.safe_load(f)

# Make a new dictionary, pip key is a dictionary in order to include the pip list
doc_new = dict()
for k in doc.keys():
    if k == "dependencies":
        doc_new[k] = dict()
    else:
        doc_new[k] = []

for k in doc.keys():
    doc_new[k] = doc[k]

if 'dependencies' not in doc_new:
    doc_new['dependencies'] = []

doc_new['dependencies'].append({'pip':[]})
doc_new['dependencies'][-1]['pip'].append('notebook')
for r in required:
    if r == "ray":
        #doc_new['dependencies'].append({'pip':[]})
        doc_new['dependencies'][-1]['pip'].append('ray')
    else:
        doc_new['dependencies'].append(r)

doc_new['dependencies'].append('ipykernel')

with open('environment.yml', 'w') as file:
        yaml.dump(doc_new,file,sort_keys=False)


prefix = doc_new['prefix']
os.system("conda env update --file environment.yml --name "+env_name+" --prune")