Track deprecated stawberry graphql fields with elasticsearch

extension.py

from strawberry.extensions import Extension
from strawberry.types import Info

class TrackDeprecatedFieldsExtension(Extension):

    def __init__(self, *args, callback: Callable[[Dict[str, int]], None], single_count=True, **kwargs):
        """
         :param callback: Callback which is called with the deprecated fields dictionary
         :param single_count: If true only count once per request, otherwise count once per field call
        """
        super().__init__(*args, **kwargs)
        self.deprecated_fields = dict()
        self.callback = callback
        self.single_count = single_count

    def on_request_end(self):
        if self.deprecated_fields:
            self.callback(self.deprecated_fields)

    def resolve(self, _next, root, info: Info, *args, **kwargs):
        field_name = info.field_name
        try:
            is_deprecated = info.parent_type.fields[field_name].is_deprecated
            if is_deprecated:
                key = f'{info.parent_type.name}.{field_name}'
                if self.single_count:
                    self.deprecated_fields[key] = 1
                else:
                    self.deprecated_fields[key] = self.deprecated_fields.get(key, 0) + 1
        except KeyError:
            pass
        return _next(root, info, *args, **kwargs)

schema.py

from elasticsearch import Elasticsearch
import strawberry


# Just an example, for performance reason you should send this to an async queue/task, e.g.:
# from django_q.tasks import async_task
# send_to_elasticsearch = lambda payload: async_task('schedule.index_es', {'index': 'deprecated_graphql_fields.v1', doc: {**payload, "schema": "myschemaname"}})
def send_to_elasticsearch(payload):
    es = Elasticsearch(['https://myelasticsearchendpoint.com'], http_auth=('username', 'password'))
    doc = {
        **payload,
        "schema": "myschemaname"
    }
    es.index('deprecated_graphql_fields.v1', body=doc)
    es.close()


schema = strawberry.Schema(query=Query, mutation=Mutation, extensions=[lambda *args, **kwargs: TrackDeprecatedFieldsExtension(*args, callback=send_to_elasticsearch, **kwargs)])