parsons/setup.py at 14f828cac582259e2cdb8ab09b3fcc7ec6223483 · bmos/parsons · 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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
import os
from pathlib import Path

from setuptools import setup


def main():
    limited_deps = os.environ.get("PARSONS_LIMITED_DEPENDENCIES", "")
    install_requires = []
    if limited_deps.strip().upper() in ("1", "YES", "TRUE", "ON"):
        install_requires = [
            "petl",
            "python-dateutil",
            "requests",
            "requests_oauthlib",
            "simplejson",
        ]
        extras_require = {
            "airtable": ["pyairtable"],
            "alchemer": ["surveygizmo"],
            "avro": ["fastavro"],
            "azure": ["azure-storage-blob"],
            "box": ["boxsdk < 10, >=4.1.0", "requests-toolbelt>=1.0.0"],
            "braintree": ["braintree"],
            "catalist": ["paramiko"],
            "civis": ["civis"],
            "dbt-redshift": [
                "dbt-redshift >= 1.5.0",
                "dbt-core >= 1.5.0",
                "lxml;python_version < '3.14'",
                "lxml>=5.5.0;python_version >= '3.14'",
            ],
            "dbt-bigquery": ["dbt-bigquery >= 1.5.0", "dbt-core >= 1.5.0"],
            "dbt-postgres": ["dbt-postgres >= 1.5.0", "dbt-core >= 1.5.0"],
            "dbt-snowflake": ["dbt-snowflake >= 1.5.0", "dbt-core >= 1.5.0"],
            "facebook": ["joblib", "facebook-business"],
            "geocode": [
                "censusgeocode @ git+https://github.com/fitnr/censusgeocode.git@1824f5d558ff6378dc4359b44c9cf535a2ba205f",
                "urllib3 >= 1.26.19;python_version < '3.10'",
                "urllib3 >= 2.6.0;python_version >= '3.10'",
            ],
            "github": ["PyGitHub"],
            "google": [
                "apiclient",
                "google-api-python-client",
                "google-cloud-bigquery",
                "google-cloud-storage",
                "google-cloud-storage-transfer",
                "gspread",
                "httplib2",
                "oauth2client",
                "validate-email",
            ],
            "mysql": ["mysql-connector-python", "sqlalchemy >= 1.4.22"],
            "newmode": ["newmode"],
            "ngpvan": ["suds-py3"],
            "mobilecommons": ["beautifulsoup4"],
            "postgres": [
                "psycopg2-binary <= 2.9.9;python_version < '3.13'",
                "psycopg2-binary >= 2.9.10;python_version >= '3.13'",
                "sqlalchemy >= 1.4.22",
            ],
            "redshift": [
                "boto3",
                "psycopg2-binary <= 2.9.9;python_version < '3.13'",
                "psycopg2-binary >= 2.9.10;python_version >= '3.13'",
                "sqlalchemy >= 1.4.22",
            ],
            "s3": ["boto3"],
            "salesforce": ["simple-salesforce"],
            "scytl": ["defusedxml", "pytz"],
            "sftp": ["paramiko"],
            "slack": ["slack-sdk"],
            "smtp": ["validate-email"],
            "targetsmart": ["xmltodict", "defusedxml"],
            "twilio": ["twilio"],
            "ssh": [
                "sshtunnel",
                "psycopg2-binary <= 2.9.9;python_version < '3.13'",
                "psycopg2-binary >= 2.9.10;python_version >= '3.13'",
                "sqlalchemy >= 1.4.22",
            ],
        }
        extras_require["all"] = sorted({lib for libs in extras_require.values() for lib in libs})
    else:
        pyproject_path = Path(__file__).parent / "requirements.txt"
        install_requires = pyproject_path.read_text().strip().split("\n")
        extras_require = {"all": []}  # No op for forward-compatibility

    setup(
        install_requires=install_requires,
        extras_require=extras_require,
    )


if __name__ == "__main__":
    main()