python/setup.py

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

import os

from setuptools import Extension, find_packages, setup

from sedona import version

with open("README.md") as fh:
    long_description = fh.read()

extension_args = {}

if os.getenv("ENABLE_ASAN"):
    extension_args = {
        "extra_compile_args": ["-fsanitize=address"],
        "extra_link_args": ["-fsanitize=address"],
    }

ext_modules = [
    Extension(
        "sedona.spark.utils.geomserde_speedup",
        sources=[
            "src/geomserde_speedup_module.c",
            "src/geomserde.c",
            "src/geom_buf.c",
            "src/geos_c_dyn.c",
        ],
        **extension_args
    )
]

setup(
    name="apache-sedona",
    version=version,
    description="Apache Sedona is a cluster computing system for processing large-scale spatial data",
    url="https://sedona.apache.org",
    license="Apache License v2.0",
    author="Apache Sedona",
    author_email="dev@sedona.apache.org",
    packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
    ext_modules=ext_modules,
    long_description=long_description,
    long_description_content_type="text/markdown",
    python_requires=">=3.6",
    install_requires=["attrs", "shapely>=1.7.0"],
    extras_require={
        "spark": ["pyspark>=2.3.0"],
        "pydeck-map": ["geopandas", "pydeck==0.8.0"],
        "kepler-map": ["geopandas", "keplergl==0.3.2"],
        "all": [
            "pyspark>=2.3.0",
            "geopandas",
            "pydeck==0.8.0",
            "keplergl==0.3.2",
            "rasterio>=1.2.10",
        ],
    },
    project_urls={
        "Documentation": "https://sedona.apache.org",
        "Source code": "https://github.com/apache/sedona",
        "Bug Reports": "https://issues.apache.org/jira/projects/SEDONA",
    },
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: Apache Software License",
    ],
)