Firesore Log Handler for Python

handler.py

from logging import Handler

from firebase_admin import firestore


class FirestoreHandler(Handler):
    def __init__(self, operation_id: str):
        super().__init__()
        self.db = firestore.client()
        self.collection = self.db.collection("logs")
        self.document = self.collection.document(operation_id)
        self.document.create({"stream": []})

    def mapLogRecord(self, record):
        """
        Default implementation of mapping the log record into a dict
        that is sent as the CGI data. Overwrite in your class.
        Contributed by Franz Glasner.
        """
        return record.__dict__

    def emit(self, record):
        """
        Emit a record.
        Send the record to firestore
        """
        try:
            data = self.mapLogRecord(record)
            self.document.update({"stream": firestore.ArrayUnion([data])})
        except Exception:
            self.handleError(record)

main.py

import logging

import firebase_admin
from firebase_admin import credentials

from handler import FirestoreHandler


cred = credentials.Certificate("firebase-credentials.json")
firebase_admin.initialize_app(credential=cred)


logger = logging.getLogger(__name__)
logger.addHandler(FirestoreHandler(operation_id="any-id"))


logger.info("Hello Firebase")