Create generic viewset and serializer for models in django rest framework

Question

I am making a lot of views and serializers for my models that takes all fields. They all look like this:

class MyModelViewSet(viewset.ModelViewSet):
    queryset = MyModel.objects.all()
    serializer_class = MyModelSerializer


class MyModelSerializer(serializers.ModelSerializer):
    class Meta:
        model = MyModel
        fields = '__all__'

I would like to make a generic Viewset and Serializer for that can simply take inn my model. I found this post to be helpful, but I would rather not have the model in the url. If possible, I would send the model with the router.register:

router.register('my_url', MyView(MyModel), basename='my-basename')

but this gives me an error:

AttributeError: This method is available only on the class, not on instances.

Any idea of how I can solve this?

You must use the viewset class, ie `MyModelViewSet` with routers, not an instance of the viewset, ie `MyModelViewSet(...)` — JPG, Jan 26 '21 at 14:41
You can adjust/arrange the router class the way you want by creating a custom router by inheriting the existing one — JPG, Jan 26 '21 at 14:42
@JPG I understand I have to use the class, but how do I then send inn the model I need for that url? — Kasper Skjeggestad, Jan 26 '21 at 15:03
Thats what I have already mentioned *in other words*, which is, "*it is not possible with built-in setup*" — JPG, Jan 26 '21 at 15:05

rokdd · Answer 1 · 2021-09-08T19:36:57.640

0

I came across when looking for the same problem. Note that in your GeneralViewSet you have to get the model from self.model. So the correct code for looping all models:

classes={}
for model in django.apps.apps.get_models():
    name = model.__name__
    classes[name]=type(name, (views.GeneralViewSet,), {})
    classes[name].model=model
    router.register(name ,classes[name], basename=name)

edited Sep 08 '21 at 19:36

answered Sep 08 '21 at 18:06

rokdd

541
4
14

Create generic viewset and serializer for models in django rest framework

1 Answers1