remove http client, use py package

uzhastik · Mar 16, 2024 · 072e625 · 072e625
1 parent daa9680
commit 072e625
Show file tree

Hide file tree

Showing 11 changed files with 6 additions and 1,108 deletions.
diff --git a/airflow/providers/yandex/hooks/yq.py b/airflow/providers/yandex/hooks/yq.py
@@ -27,7 +27,7 @@
 from airflow.exceptions import AirflowException
 from airflow.providers.yandex.hooks.yandex import YandexCloudBaseHook
 from airflow.providers.yandex.utils.user_agent import provider_user_agent
-from airflow.providers.yandex.yq_client.http_client import YQHttpClient, YQHttpClientConfig
+from yandex_query_client import YQHttpClient, YQHttpClientConfig
 
 
 class YQHook(YandexCloudBaseHook):

diff --git a/airflow/providers/yandex/provider.yaml b/airflow/providers/yandex/provider.yaml
@@ -49,6 +49,7 @@ versions:
 dependencies:
   - apache-airflow>=2.6.0
   - yandexcloud>=0.228.0
+  - yandex-query-client>=0.1.2
   - python-dateutil>=2.8.0
   # Requests 3 if it will be released, will be heavily breaking.
   - requests>=2.27.0,<3

diff --git a/airflow/providers/yandex/yq_client/__init__.py b/airflow/providers/yandex/yq_client/__init__.py
diff --git a/airflow/providers/yandex/yq_client/http_client.py b/airflow/providers/yandex/yq_client/http_client.py