diff --git a/requirements.txt b/requirements.txt index 9efd3d4a..a888b017 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,4 +12,4 @@ PyYAML requests[socks]<3 setuptools>=65.5.1 shtab -SQLAlchemy[postgresql_psycopg2binary]~=2.0 +SQLAlchemy~=2.0 diff --git a/requirements.win.txt b/requirements.win.txt index e3f8d8fa..074cd925 100644 --- a/requirements.win.txt +++ b/requirements.win.txt @@ -12,4 +12,4 @@ PyYAML requests<3 setuptools>=65.5.1 shtab -SQLAlchemy[postgresql_psycopg2binary]~=2.0 +SQLAlchemy~=2.0 diff --git a/setup.py b/setup.py index 12defeff..9d02e689 100644 --- a/setup.py +++ b/setup.py @@ -141,6 +141,9 @@ def get_requirements(*, mode="standard"): "docs": ["requirements.docs.txt"], "test": ["requirements.test.txt"], "psql": ["requirements.txt"], + "postgresql": ["requirements.txt"], + "psql-binary": ["requirements.txt"], + "postgresql-binary": ["requirements.txt"], } if is_win_platform(): @@ -177,8 +180,10 @@ def get_requirements(*, mode="standard"): result.add(line) - if mode == "psql": + if mode in ("psql", "postgresql"): result.add("psycopg2") + elif mode in ("psql-binary", "postgresql-binary"): + result.add("psycopg2-binary") return list(result) @@ -259,6 +264,9 @@ def get_console_scripts(): # pragma: no cover "dev": get_requirements(mode="dev"), "test": get_requirements(mode="test"), "psql": get_requirements(mode="psql"), + "psql-binary": get_requirements(mode="psql-binary"), + "postgresql": get_requirements(mode="postgresql"), + "postgresql-binary": get_requirements(mode="postgresql-binary"), "full": get_requirements(mode="full"), }, description="The tool to check the availability or syntax of domain, IP or URL.",