Use full import to get client

andrewsg · andrewsg · commit aa6c226c185f · 2017-09-25T10:48:08.000-07:00
diff --git a/appengine/flexible/tasks/create_app_engine_queue_task.py b/appengine/flexible/tasks/create_app_engine_queue_task.py
@@ -29,11 +29,10 @@ def seconds_from_now_to_rfc3339_datetime(seconds):
 def create_task(project, queue, location, payload=None, in_seconds=None):
     """Create a task for a given queue with an arbitrary payload."""
 
-    from googleapiclient import discovery
+    import googleapiclient.discovery
 
     # Create a client.
-    client = discovery.build(
-        'cloudtasks', 'v2beta2')
+    client = googleapiclient.discovery.build('cloudtasks', 'v2beta2')
 
     url = '/log_payload'
     body = {
diff --git a/tasks/pull_queue_snippets.py b/tasks/pull_queue_snippets.py
@@ -27,10 +27,10 @@
 def create_task(project, queue, location):
     """Create a task for a given queue with an arbitrary payload."""
 
+    import googleapiclient.discovery
+
     # Create a client.
-    from googleapiclient import discovery
-    client = discovery.build(
-        'cloudtasks', 'v2beta2')
+    client = googleapiclient.discovery.build('cloudtasks', 'v2beta2')
 
     payload = 'a message for the recipient'
     task = {
@@ -54,10 +54,10 @@ def create_task(project, queue, location):
 def pull_task(project, queue, location):
     """Pull a single task from a given queue and lease it for 10 minutes."""
 
+    import googleapiclient.discovery
+
     # Create a client.
-    from googleapiclient import discovery
-    client = discovery.build(
-        'cloudtasks', 'v2beta2')
+    client = googleapiclient.discovery.build('cloudtasks', 'v2beta2')
 
     duration_seconds = '600s'
     pull_options = {
@@ -79,10 +79,10 @@ def pull_task(project, queue, location):
 def acknowledge_task(task):
     """Acknowledge a given task."""
 
+    import googleapiclient.discovery
+
     # Create a client.
-    from googleapiclient import discovery
-    client = discovery.build(
-        'cloudtasks', 'v2beta2')
+    client = googleapiclient.discovery.build('cloudtasks', 'v2beta2')
 
     body = {'scheduleTime': task['scheduleTime']}
     client.projects().locations().queues().tasks().acknowledge(
