Renaming the Campaign Manager Hook for consistency with other GMP hooks.

google · Oct 28, 2019 · 8af2977 · 8af2977
1 parent c8dd166
commit 8af2977
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/orchestra/google/marketing_platform/hooks/campaign_manager.py b/orchestra/google/marketing_platform/hooks/campaign_manager.py
@@ -24,7 +24,7 @@
 from googleapiclient.discovery import build
 
 
-class GoogleCampaignManagerReportingHook(GoogleCloudBaseHook):
+class GoogleCampaignManagerHook(GoogleCloudBaseHook):
   """Hook for connecting to the Doubleclick Campaign Manager DFA Reporting API.
   """
 
@@ -41,7 +41,7 @@ def __init__(self,
       api_name: The name of the DCM API.
       api_version: The version of the DCM API.
     """
-    super(GoogleCampaignManagerReportingHook, self).__init__(
+    super(GoogleCampaignManagerHook, self).__init__(
         gcp_conn_id=gcp_conn_id,
         delegate_to=delegate_to)
     self.api_name = api_name

diff --git a/orchestra/google/marketing_platform/operators/campaign_manager.py b/orchestra/google/marketing_platform/operators/campaign_manager.py
@@ -26,7 +26,7 @@
 
 from airflow.contrib.hooks.gcs_hook import GoogleCloudStorageHook
 from orchestra.google.marketing_platform.hooks.campaign_manager import (
-  GoogleCampaignManagerReportingHook
+  GoogleCampaignManagerHook
 )
 from orchestra.google.marketing_platform.operators.marketing_platform import (
   GoogleMarketingPlatformBaseOperator
@@ -87,7 +87,7 @@ def _run_report(self, profile_id, report_id):
 
   def execute(self, context):
     if self.hook is None:
-      self.hook = GoogleCampaignManagerReportingHook(
+      self.hook = GoogleCampaignManagerHook(
           gcp_conn_id=self.gcp_conn_id,
           delegate_to=self.delegate_to)
 
@@ -137,7 +137,7 @@ def __init__(self,
 
   def execute(self, context):
     if self.hook is None:
-      self.hook = GoogleCampaignManagerReportingHook(
+      self.hook = GoogleCampaignManagerHook(
           gcp_conn_id=self.gcp_conn_id,
           delegate_to=self.delegate_to)
 
@@ -242,7 +242,7 @@ def execute(self, context):
           google_cloud_storage_conn_id=self.gcp_conn_id,
           delegate_to=self.delegate_to)
     if self.cm_hook is None:
-      self.cm_hook = GoogleCampaignManagerReportingHook(
+      self.cm_hook = GoogleCampaignManagerHook(
           gcp_conn_id=self.gcp_conn_id,
           delegate_to=self.delegate_to)
 

diff --git a/orchestra/google/marketing_platform/sensors/campaign_manager.py b/orchestra/google/marketing_platform/sensors/campaign_manager.py
@@ -21,7 +21,7 @@
 
 from airflow.sensors.base_sensor_operator import BaseSensorOperator
 from orchestra.google.marketing_platform.hooks.campaign_manager import (
-  GoogleCampaignManagerReportingHook
+  GoogleCampaignManagerHook
 )
 
 logger = logging.getLogger(__name__)
@@ -72,7 +72,7 @@ def __init__(self,
 
   def poke(self, context):
     if self.hook is None:
-      self.hook = GoogleCampaignManagerReportingHook(
+      self.hook = GoogleCampaignManagerHook(
           gcp_conn_id=self.gcp_conn_id,
           delegate_to=self.delegate_to)
     logger.info(self.gcp_conn_id)