Using Loguru with Google Cloud Logging in a Cloud Function

cf_logging.py

import json

from loguru import logger

# From: https://cloud.google.com/run/docs/logging#writing_structured_logs

class CloudLoggingHandler:
    def __init__(self, request):
        project = "" # Add your project name here

        self.global_log_fields = {}

        trace_header = request.headers.get("X-Cloud-Trace-Context")

        if trace_header and project:
            trace = trace_header.split("/")
            self.global_log_fields[
                "logging.googleapis.com/trace"
            ] = f"projects/{project}/traces/{trace[0]}"

    def write(self, message):
        entry = dict(
            severity=message.record["level"].name,
            message=message,
            **self.global_log_fields,
        )

        print(json.dumps(entry))


def handle_http_request(request):
    logger.remove()
    logger.add(CloudLoggingHandler(request), format="{function}:{line} - {message}")

    logger.debug("This is a DEBUG message")
    logger.info("This is an INFO message")
    logger.warning("This is a WARNING message")
    logger.error("This is an ERROR message")

    return "Completed", 200