Для DRF есть генератор документации: rest_framework_swagger. В нем есть похожий функционал. Можете попробовать посмотреть, тем более у вас как раз DRF используется.
В классе DocumentationGenerator метод get_models
https://github.com/marcgibbons/django-rest-swagger...