diff --git a/dbt/common/ui.py b/dbt/common/ui.py index 80eeb6c6..1056df2e 100644 --- a/dbt/common/ui.py +++ b/dbt/common/ui.py @@ -1,3 +1,4 @@ +import os import sys import textwrap from typing import Dict diff --git a/pyproject.toml b/pyproject.toml index f641b275..74e760a0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -21,14 +21,21 @@ classifiers = [ ] dependencies = [ "agate~=1.7.0", - "colorama>=0.3.9,<0.5", # TODO: major version 0 - should we use it? + "colorama>=0.3.9,<0.5", + "isodate>=0.6,<0.7", "jsonschema~=4.0", "Jinja2~=3.0", "mashumaro[msgpack]~=3.9", + "pathspec>=0.9,<0.12", # TODO: I'm not sure this is needed. check search.py? "protobuf>=4.0.0", "python-dateutil~=2.0", "requests<3.0.0", "typing-extensions~=4.4", + # Match snowflake-connector-python, to ensure compatibility in dbt-snowflake + "cffi>=1.9,<2.0.0", + "idna>=2.5,<4", + "requests<3.0.0", + "urllib3~=1.0", # we pin urllib3 ~= 1.0 because of openssl requirement for requests ] [build-system] diff --git a/third-party-stubs/colorama/__init__.py b/third-party-stubs/colorama/__init__.py new file mode 100644 index 00000000..4502880e --- /dev/null +++ b/third-party-stubs/colorama/__init__.py @@ -0,0 +1,16 @@ +from typing import Optional, Any + +class Fore: + RED: str = ... + GREEN: str = ... + YELLOW: str = ... + +class Style: + RESET_ALL: str = ... + +def init( + autoreset: bool = ..., + convert: Optional[Any] = ..., + strip: Optional[Any] = ..., + wrap: bool = ..., +) -> None: ...