add fields __all__

ahmad88me · Sep 17, 2023 · a222758 · a222758
1 parent a797a3e
commit a222758
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/django-rest-gen/apigen.py b/src/django-rest-gen/apigen.py
@@ -65,7 +65,8 @@ def write_class_serializer(class_name, fpath, write=False):
     """
     content = f"""\nclass {class_name}Serializer(serializers.ModelSerializer):\n
     class Meta:
-        model = {class_name}\n\n"""
+        model = {class_name}
+        fields = '__all__'\n\n"""
     if write:
         with open(fpath, "a") as f:
             f.write(content)