setup.py

#works for normal installation but not for editable installation unless 
import os
# from setuptools import setup, Extension
from  numpy.distutils.core import setup, Extension
# os.system("f2py -c -m occultquad occultquad.f")
# os.system("f2py -c -m occultnl occultnl.f")

# with open('CONAN/VERSION.dat') as version_file:
#       __version__ = version_file.read().strip()

# # Check if the NO_FORTRAN environment variable is set
no_fortran = os.getenv('NO_FORTRAN', 'False').lower() in ('true', '1', 't')  

if no_fortran:
      print("Skipping compilation of fortran code, python equivalent of transit model will be used")
      ext_modules = []
else:
      print("Compiling fortran code")     
      extOccultnl = Extension('CONAN.occultnl', sources=['CONAN/occultnl.f'])
      extOccultquad = Extension('CONAN.occultquad', sources=['CONAN/occultquad.f'])
      ext_modules = [extOccultnl, extOccultquad]


setup(
      # name='CONAN',
      # version='3.3.2',
      # description='COde for exoplaNet ANalysis',
      # long_description=open("README.md").read(),
      # long_description_content_type='text/markdown',
      # url='https://github.com/mlendl42/CONAN',
      # author='Monika Lendl, Babatunde Akinsanmi',
      # author_email='monika.lendl@unige.ch',
      # license='MIT',
      packages=['CONAN'],
      # python_requires=">=3.10",
      ext_modules=ext_modules,
      # install_requires=['chainconsumer==1.1.2','scipy','pandas','lmfit','dynesty', 'astropy',
      #                   'astroquery','celerite','corner','lightkurve','dill','dace_query',
      #                   'matplotlib','emcee','george','ldtk==1.7.0','tqdm','spleaf','numba'
      #                   ],
      )

#copy command line script to home directory
print("\ncopying command line script to home directory")
os.system("cp conan_cmd_fit.py ~/conan_fit.py")

#add function in .zshrc/.bashrc to call conan_fit.py from anywhere
if os.path.exists(os.path.expanduser("~/.zshrc")):
      if os.system("grep -q 'function conanfit()' ~/.zshrc"):  #check if function already exists
            print("adding conanfit function to ~/.zshrc")
            os.system("echo 'function conanfit() { python ~/conan_fit.py $@; }' >> ~/.zshrc")
            os.system("source ~/.zshrc")
      else:
            print("conanfit function already exists in ~/.zshrc")
      

if os.path.exists(os.path.expanduser("~/.bashrc")):
      if os.system("grep -q 'function conanfit()' ~/.bashrc"): #check if function already exists
            print("adding conanfit function to ~/.bashrc")
            os.system("echo 'function conanfit() { python ~/conan_fit.py $@; }' >> ~/.bashrc")
            os.system("source ~/.bashrc")
      else:
            print("conanfit function already exists in ~/.bashrc")

print("\n")