nova-physics-python/setup.py at main · kadir014/nova-physics-python · GitHub

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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import os
import platform
import shutil
import io
import tarfile
import urllib.request
from pathlib import Path
from setuptools import setup


if platform.system() == "Darwin":
    # OSX stuff
    import ssl
    ssl._create_default_https_context = ssl._create_unverified_context


BASE_PATH = Path(os.getcwd())
BUILD_PATH = BASE_PATH / "build"
NOVA_PATH = BUILD_PATH / "nova-physics"


def download_nightly() -> None:
    """ Download the latest Nova Physics commit. """

    print(
        "    \n"
        "    setup.py:\n"
        "    =========================================\n"
        "    Downloading latest commit of Nova Physics\n"
        "    \n"
        "    https://github.com/kadir014/nova-physics\n"
        "    =========================================\n"
        "    \n"
    )

    if os.path.exists(BUILD_PATH / "nova-physics"):
        shutil.rmtree(BUILD_PATH / "nova-physics")

    if os.path.exists(BUILD_PATH / "nova-physics-main"):
        shutil.rmtree(BUILD_PATH / "nova-physics-main")

    response = urllib.request.urlopen(
        f"https://github.com/kadir014/nova-physics/archive/refs/heads/main.tar.gz"
    )

    data = response.read()

    with tarfile.open(mode="r:gz", fileobj=io.BytesIO(data)) as tar:
        tar.extractall(BUILD_PATH)

    os.rename(BUILD_PATH / "nova-physics-main", BUILD_PATH / "nova-physics")

    print(
        "    \n"
        "    setup.py:\n"
        "    =========================================\n"
        "    Downloaded and extracted successfully.\n"
        "    =========================================\n"
        "    \n"
    )


# TODO: Downloads multiple times for one install...
download_nightly()

setup(
    packages=["nova"],
    cffi_modules=["nova/_cffi_gen.py:ffibuilder"],
)