3_pubsub_read.py

# Load data from Pub/Sub infrastructure
def LoadSub(sub_name):
  data = []

  subscriber = pubsub_v1.SubscriberClient(credentials=credentials)
  subscription_path = subscriber.subscription_path(PUBSUB_PROJECT_ID, sub_name)

  def callback(message: pubsub_v1.subscriber.message.Message):
      data.append(message)

  streaming_pull_future = subscriber \
    .subscribe(subscription_path, callback=callback)

  with subscriber:
      try:
          streaming_pull_future.result(timeout=PUBSUB_TIMEOUT)
      except TimeoutError:
          streaming_pull_future.cancel()
          streaming_pull_future.result()

  return data