diff --git a/Dockerfile b/Dockerfile index d1fc953c8..e6102ecb5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -7,6 +7,6 @@ WORKDIR $USR_APP_PATH RUN chmod 777 . COPY . /app -RUN pip install --no-cache-dir "/app[postgres, snowflake, bigquery, redshift, databricks]" +RUN pip install --no-cache-dir "/app[all]" ENTRYPOINT ["edr"] diff --git a/setup.py b/setup.py index d39ca3843..73a192ced 100644 --- a/setup.py +++ b/setup.py @@ -8,6 +8,16 @@ # The text of the README file. README = (HERE / "README.md").read_text() +ADAPTER_EXTRA_REQUIREMENTS = { + "snowflake": ["dbt-snowflake>=0.20,<2.0.0"], + "bigquery": ["dbt-bigquery>=0.20,<2.0.0"], + "redshift": ["dbt-redshift>=0.20,<2.0.0"], + "postgres": ["dbt-postgres>=0.20,<2.0.0"], + "databricks": ["dbt-databricks>=0.20,<2.0.0"], + "spark": ["dbt-spark>=0.20,<2.0.0", "dbt-spark[PyHive]>=0.20,<2.0.0"], +} + + setup( name="elementary-data", description="Data monitoring and lineage", @@ -42,12 +52,8 @@ "packaging>=20.9,<22.0", ], extras_require={ - "snowflake": ["dbt-snowflake>=0.20,<2.0.0"], - "bigquery": ["dbt-bigquery>=0.20,<2.0.0"], - "redshift": ["dbt-redshift>=0.20,<2.0.0"], - "postgres": ["dbt-postgres>=0.20,<2.0.0"], - "databricks": ["dbt-databricks>=0.20,<2.0.0"], - "spark": ["dbt-spark>=0.20,<2.0.0", "dbt-spark[PyHive]>=0.20,<2.0.0"], + **ADAPTER_EXTRA_REQUIREMENTS, + "all": sum(ADAPTER_EXTRA_REQUIREMENTS.values(), []), }, long_description_content_type="text/markdown", license="",