setup.py

from setuptools import setup
from distutils.extension import Extension
import os
import subprocess
from bs4 import BeautifulSoup
import urllib.request


cpu_extensions=dict(
    libraries=["gsl", "gslcblas"],
    language="c++",
    runtime_library_dirs=[],
    extra_compile_args=["-O3", "-shared", "-std=c++11", "-fPIC"],
    include_dirs=["/usr/lib64",
                    "/mnt/Data_Volume/Documents/software/Anaconda/envs/few_env/include/python3.7m",
                    "/home/shaunf/Documents/Computer/Code/projects/Massive_Vector_Field_Dynamical_Friction/ProcaAroundKerr/GWGenerator/include",
                    "include"
                    ],
    library_dirs=None
)


frequency_ext = Extension(
            "pyKerrFreqs",
            sources=["src/Utility.cc", "src/UtilityFuncs.pyx"],
            **cpu_extensions
)

flux_ext = Extension(
            "pyAnalyticFluxes",
            sources=["src/dIdt8H_5PNe10.cc", "src/dIdt.pyx"],
            **cpu_extensions
)

extensions = [flux_ext, frequency_ext]


##Verify proca data directory exists. If not, download from Zenodo
ProcaDataPath = os.path.abspath(os.path.dirname(__file__))+"/GWGen/ProcaData"
if not os.path.exists(ProcaDataPath):
    print("Download data from Zenodo repository. This will take some time...")
    os.mkdir(ProcaDataPath)
    ZenodoURL = "https://zenodo.org/record/7590391"
    page = urllib.request.urlopen(ZenodoURL)
    soup = BeautifulSoup(page, "html.parser")
    hreflinks = [link.get("href") for link in soup.findAll("link")]
    datasetUrls = [str for str in hreflinks if "Mode_1_Overtone_0" in str] #take only m=1, n=0 datasets
    basefilenames = [os.path.basename(i) for i in datasetUrls]
    datasetTargetFilenames = [ProcaDataPath+"/"+i for i in basefilenames]

    for i in range(len(datasetUrls)):
        subprocess.run(["wget", "--quiet", "--no-check-certificate", "--output-document="+datasetTargetFilenames[i], datasetUrls[i]])


setup(
    name="GWGen",
    version="0.1",
    description="Calculating modified waveform due to precense of Proca cloud around spinning black hole",
    author="Shaun Fell",
    packages=["GWGen",
                "GWGen.WFGenerator",
                "GWGen.Utils",
                "GWGen.UndressedFluxes",
                "GWGen.DressedFluxes",
                "GWGen.ProcaData"
                ],
    package_data={"GWGen": ['ProcaData/*.npz']},
    ext_modules=extensions,

)