apache-airflow-providers-google 15.1.0rc1__py3-none-any.whl → 19.1.0rc1__py3-none-any.whl

airflow/providers/google/marketing_platform/hooks/display_video.py

@@ -34,7 +34,7 @@ class GoogleDisplayVideo360Hook(GoogleBaseHook):
 
     def __init__(
         self,
-        api_version: str = "v2",
+        api_version: str = "v4",
         gcp_conn_id: str = "google_cloud_default",
         impersonation_chain: str | Sequence[str] | None = None,
         **kwargs,
@@ -46,18 +46,6 @@ class GoogleDisplayVideo360Hook(GoogleBaseHook):
         )
         self.api_version = api_version
 
-    def get_conn(self) -> Resource:
-        """Retrieve connection to DisplayVideo."""
-        if not self._conn:
-            http_authorized = self._authorize()
-            self._conn = build(
-                "doubleclickbidmanager",
-                self.api_version,
-                http=http_authorized,
-                cache_discovery=False,
-            )
-        return self._conn
-
     def get_conn_to_display_video(self) -> Resource:
         """Retrieve connection to DisplayVideo."""
         if not self._conn:
@@ -89,107 +77,13 @@ class GoogleDisplayVideo360Hook(GoogleBaseHook):
         """
         return [f"gdbm-{partner_id}/entity/{{{{ ds_nodash }}}}.*.{entity_type}.json"]
 
-    def create_query(self, query: dict[str, Any]) -> dict:
-        """
-        Create a query.
-
-        :param query: Query object to be passed to request body.
-        """
-        response = self.get_conn().queries().create(body=query).execute(num_retries=self.num_retries)
-        return response
-
-    def delete_query(self, query_id: str) -> None:
-        """
-        Delete a stored query as well as the associated stored reports.
-
-        :param query_id: Query ID to delete.
-        """
-        self.get_conn().queries().delete(queryId=query_id).execute(num_retries=self.num_retries)
-
-    def get_query(self, query_id: str) -> dict:
-        """
-        Retrieve a stored query.
-
-        :param query_id: Query ID to retrieve.
-        """
-        response = self.get_conn().queries().get(queryId=query_id).execute(num_retries=self.num_retries)
-        return response
-
-    def list_queries(self) -> list[dict]:
-        """Retrieve stored queries."""
-        response = self.get_conn().queries().list().execute(num_retries=self.num_retries)
-        return response.get("queries", [])
-
-    def run_query(self, query_id: str, params: dict[str, Any] | None) -> dict:
-        """
-        Run a stored query to generate a report.
-
-        :param query_id: Query ID to run.
-        :param params: Parameters for the report.
-        """
-        return (
-            self.get_conn().queries().run(queryId=query_id, body=params).execute(num_retries=self.num_retries)
-        )
-
-    def get_report(self, query_id: str, report_id: str) -> dict:
-        """
-        Retrieve a report.
-
-        :param query_id: Query ID for which report was generated.
-        :param report_id: Report ID to retrieve.
-        """
-        return (
-            self.get_conn()
-            .queries()
-            .reports()
-            .get(queryId=query_id, reportId=report_id)
-            .execute(num_retries=self.num_retries)
-        )
-
-    def upload_line_items(self, line_items: Any) -> list[dict[str, Any]]:
-        """
-        Upload line items in CSV format.
-
-        :param line_items: downloaded data from GCS and passed to the body request
-        :return: response body.
-        """
-        request_body = {
-            "lineItems": line_items,
-            "dryRun": False,
-            "format": "CSV",
-        }
-
-        response = (
-            self.get_conn()
-            .lineitems()
-            .uploadlineitems(body=request_body)
-            .execute(num_retries=self.num_retries)
-        )
-        return response
-
-    def download_line_items(self, request_body: dict[str, Any]) -> list[Any]:
-        """
-        Retrieve line items in CSV format.
-
-        :param request_body: dictionary with parameters that should be passed into.
-            More information about it can be found here:
-            https://developers.google.com/bid-manager/v1.1/lineitems/downloadlineitems
-        """
-        response = (
-            self.get_conn()
-            .lineitems()
-            .downloadlineitems(body=request_body)
-            .execute(num_retries=self.num_retries)
-        )
-        return response["lineItems"]
-
     def create_sdf_download_operation(self, body_request: dict[str, Any]) -> dict[str, Any]:
         """
         Create an SDF Download Task and Returns an Operation.
 
         :param body_request: Body request.
 
-        More information about body request n be found here:
+        More information about body request can be found here:
         https://developers.google.com/display-video/api/reference/rest/v1/sdfdownloadtasks/create
         """
         result = (
@@ -219,7 +113,7 @@ class GoogleDisplayVideo360Hook(GoogleBaseHook):
         """
         Download media.
 
-        :param resource_name: of the media that is being downloaded.
+        :param resource_name: The resource name of the media that is being downloaded.
         """
         request = self.get_conn_to_display_video().media().download_media(resourceName=resource_name)
         return request

airflow/providers/google/marketing_platform/links/analytics_admin.py

@@ -18,19 +18,12 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, ClassVar
 
+from airflow.providers.common.compat.sdk import BaseOperatorLink, XCom
+
 if TYPE_CHECKING:
-    from airflow.models import BaseOperator
     from airflow.models.taskinstancekey import TaskInstanceKey
-    from airflow.utils.context import Context
-
-from airflow.providers.google.version_compat import AIRFLOW_V_3_0_PLUS
-
-if AIRFLOW_V_3_0_PLUS:
-    from airflow.sdk import BaseOperatorLink
-    from airflow.sdk.execution_time.xcom import XCom
-else:
-    from airflow.models import XCom  # type: ignore[no-redef]
-    from airflow.models.baseoperatorlink import BaseOperatorLink  # type: ignore[no-redef]
+    from airflow.providers.common.compat.sdk import Context
+    from airflow.providers.google.version_compat import BaseOperator
 
 
 BASE_LINK = "https://analytics.google.com/analytics/web/"
@@ -64,11 +57,9 @@ class GoogleAnalyticsPropertyLink(GoogleAnalyticsBaseLink):
     @staticmethod
     def persist(
         context: Context,
-        task_instance: BaseOperator,
         property_id: str,
     ):
-        task_instance.xcom_push(
-            context,
+        context["task_instance"].xcom_push(
             key=GoogleAnalyticsPropertyLink.key,
             value={"property_id": property_id},
         )

airflow/providers/google/marketing_platform/operators/analytics_admin.py

@@ -39,7 +39,7 @@ if TYPE_CHECKING:
     from google.api_core.retry import Retry
     from google.protobuf.message import Message
 
-    from airflow.utils.context import Context
+    from airflow.providers.common.compat.sdk import Context
 
 
 class GoogleAnalyticsAdminListAccountsOperator(GoogleCloudBaseOperator):
@@ -194,7 +194,6 @@ class GoogleAnalyticsAdminCreatePropertyOperator(GoogleCloudBaseOperator):
         self.log.info("The Google Analytics property %s was created successfully.", prop.name)
         GoogleAnalyticsPropertyLink.persist(
             context=context,
-            task_instance=self,
             property_id=prop.name.lstrip("properties/"),
         )
 

airflow/providers/google/marketing_platform/operators/campaign_manager.py

@@ -28,12 +28,12 @@ from typing import TYPE_CHECKING, Any
 from googleapiclient import http
 
 from airflow.exceptions import AirflowException
-from airflow.models import BaseOperator
 from airflow.providers.google.cloud.hooks.gcs import GCSHook
 from airflow.providers.google.marketing_platform.hooks.campaign_manager import GoogleCampaignManagerHook
+from airflow.providers.google.version_compat import BaseOperator
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
+    from airflow.providers.common.compat.sdk import Context
 
 
 class GoogleCampaignManagerDeleteReportOperator(BaseOperator):
@@ -237,7 +237,7 @@ class GoogleCampaignManagerDownloadReportOperator(BaseOperator):
                 mime_type="text/csv",
             )
 
-        self.xcom_push(context, key="report_name", value=report_name)
+        context["task_instance"].xcom_push(key="report_name", value=report_name)
 
 
 class GoogleCampaignManagerInsertReportOperator(BaseOperator):
@@ -308,7 +308,7 @@ class GoogleCampaignManagerInsertReportOperator(BaseOperator):
         self.log.info("Inserting Campaign Manager report.")
         response = hook.insert_report(profile_id=self.profile_id, report=self.report)
         report_id = response.get("id")
-        self.xcom_push(context, key="report_id", value=report_id)
+        context["task_instance"].xcom_push(key="report_id", value=report_id)
         self.log.info("Report successfully inserted. Report id: %s", report_id)
         return response
 
@@ -381,7 +381,7 @@ class GoogleCampaignManagerRunReportOperator(BaseOperator):
             synchronous=self.synchronous,
         )
         file_id = response.get("id")
-        self.xcom_push(context, key="file_id", value=file_id)
+        context["task_instance"].xcom_push(key="file_id", value=file_id)
         self.log.info("Report file id: %s", file_id)
         return response