summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/py-pyarrow/package.py
blob: 0be8e772fc96d401f1e0a85fff87b5348b6290c9 (plain) (blame)
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

from spack.package import *


class PyPyarrow(PythonPackage, CudaPackage):
    """A cross-language development platform for in-memory data.

    This package contains the Python bindings.
    """

    homepage = "https://arrow.apache.org"
    pypi = "pyarrow/pyarrow-0.17.1.tar.gz"
    git = "https://github.com/apache/arrow"

    license("Apache-2.0")

    version("10.0.1", sha256="1a14f57a5f472ce8234f2964cd5184cccaa8df7e04568c64edc33b23eb285dd5")
    version("8.0.0", sha256="4a18a211ed888f1ac0b0ebcb99e2d9a3e913a481120ee9b1fe33d3fedb945d4e")
    version("7.0.0", sha256="da656cad3c23a2ebb6a307ab01d35fce22f7850059cffafcb90d12590f8f4f38")
    version("4.0.1", sha256="11517f0b4f4acbab0c37c674b4d1aad3c3dfea0f6b1bb322e921555258101ab3")
    version("3.0.0", sha256="4bf8cc43e1db1e0517466209ee8e8f459d9b5e1b4074863317f2a965cf59889e")
    version("0.17.1", sha256="278d11800c2e0f9bea6314ef718b2368b4046ba24b6c631c14edad5a1d351e49")
    version("0.15.1", sha256="7ad074690ba38313067bf3bbda1258966d38e2037c035d08b9ffe3cce07747a5")
    version("0.12.1", sha256="10db6e486c918c3af999d0114a22d92770687e3a6607ea3f14e6748854824c2a")
    version("0.11.0", sha256="07a6fd71c5d7440f2c42383dd2c5daa12d7f0a012f1e88288ed08a247032aead")
    version("0.9.0", sha256="7db8ce2f0eff5a00d6da918ce9f9cfec265e13f8a119b4adb1595e5b19fd6242")

    variant("parquet", default=False, description="Build with Parquet support")
    variant("orc", default=False, description="Build with orc support")
    variant("dataset", default=False, description="Build with Dataset support")

    conflicts("~parquet", when="+dataset")

    depends_on("cmake@3.0.0:", type="build")
    depends_on("pkgconfig", type="build")
    depends_on("python@3.5:", type=("build", "run"), when="@0.17:")
    depends_on("python@3.6:", type=("build", "run"), when="@3.0.0:")
    depends_on("python@3.7:", type=("build", "run"), when="@7.0.0:")
    depends_on("py-setuptools", type="build")
    depends_on("py-setuptools@38.6.0:", type="build", when="@7.0.0:")
    depends_on("py-setuptools@40.1.0:", type="build", when="@10.0.1:")
    depends_on("py-setuptools-scm", type="build", when="@0.15.0:")
    depends_on("py-cython", type="build")
    depends_on("py-cython@0.29:", type="build", when="@0.15.0:")
    depends_on("py-cython@0.29.22:", type="build", when="@8.0.0:")
    # in newer pip versions --install-option does not exist
    depends_on("py-pip@:23.0", type="build")

    depends_on("py-numpy@1.14:", type=("build", "run"), when="@0.15.0:")
    depends_on("py-numpy@1.16.6:", type=("build", "run"), when="@3.0.0:")

    arrow_versions = (
        "@0.9.0",
        "@0.11.0",
        "@0.12.1",
        "@0.15.1",
        "@0.17.1",
        "@3.0.0",
        "@4.0.1",
        "@7.0.0",
        "@8.0.0",
        "@10.0.1",
    )
    for v in arrow_versions:
        depends_on("arrow+python" + v, when=v)
        depends_on("arrow+parquet+python" + v, when="+parquet" + v)
        depends_on("arrow+cuda" + v, when="+cuda" + v)
        depends_on("arrow+orc" + v, when="+orc" + v)

    patch("for_aarch64.patch", when="@0 target=aarch64:")

    def setup_build_environment(self, env):
        env.set("PYARROW_WITH_PARQUET", self.spec.satisfies("+parquet"))
        env.set("PYARROW_WITH_CUDA", self.spec.satisfies("+cuda"))
        env.set("PYARROW_WITH_ORC", self.spec.satisfies("+orc"))
        env.set("PYARROW_WITH_DATASET", self.spec.satisfies("+dataset"))

    def install_options(self, spec, prefix):
        args = []
        if spec.satisfies("+parquet"):
            args.append("--with-parquet")
        if spec.satisfies("+cuda"):
            args.append("--with-cuda")
        if spec.satisfies("+orc"):
            args.append("--with-orc")
        if spec.satisfies("+dataset"):
            args.append("--with-dataset")
        return args