# 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 PyPyspark(PythonPackage):
    """Python bindings for Apache Spark"""

    homepage = "https://spark.apache.org"
    pypi = "pyspark/pyspark-3.0.1.tar.gz"

    version("3.3.1", sha256="e99fa7de92be406884bfd831c32b9306a3a99de44cfc39a2eefb6ed07445d5fa")
    version("3.3.0", sha256="7ebe8e9505647b4d124d5a82fca60dfd3891021cf8ad6c5ec88777eeece92cf7")
    version("3.2.1", sha256="0b81359262ec6e9ac78c353344e7de026027d140c6def949ff0d80ab70f89a54")
    version("3.1.3", sha256="39ac641ef5559a3d1286154779fc990316e9934520853615ae4785c1af52d14b")
    version("3.1.2", sha256="5e25ebb18756e9715f4d26848cc7e558035025da74b4fc325a0ebc05ff538e65")
    version("3.0.1", sha256="38b485d3634a86c9a2923c39c8f08f003fdd0e0a3d7f07114b2fb4392ce60479")

    depends_on("py-setuptools", type="build")
    depends_on("py-py4j@0.10.9.5", when="@3.3.0:", type=("build", "run"))
    depends_on("py-py4j@0.10.9.3", when="@3.2.1", type=("build", "run"))
    depends_on("py-py4j@0.10.9", when="@3.0.1:3.1.3", type=("build", "run"))

    def setup_run_environment(self, env):
        env.set("PYSPARK_PYTHON", self.spec["python"].command.path)
        env.set("PYSPARK_DRIVER_PYTHON", self.spec["python"].command.path)