From 3fa4de50038ae42181839f0e3cd3b10761a501c6 Mon Sep 17 00:00:00 2001 From: Alberto Moreta Date: Thu, 29 Aug 2024 13:04:08 +0200 Subject: [PATCH] mes-227: Remove translation fields from API response using get_fields() instead of __init__ (#1) --- api/mixins/ExcludeTranslationFieldsMixin.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/api/mixins/ExcludeTranslationFieldsMixin.py b/api/mixins/ExcludeTranslationFieldsMixin.py index 6d7cecc..2ae143e 100644 --- a/api/mixins/ExcludeTranslationFieldsMixin.py +++ b/api/mixins/ExcludeTranslationFieldsMixin.py @@ -4,8 +4,9 @@ class ExcludeTranslationFieldsMixin: # Exclude the specific language fields from the serializer output - def __init__(self, *args, **kwargs): - super().__init__(args, kwargs) + def get_fields(self): + fields = super().get_fields() for f in self.Meta.model._meta.fields: - if f.name in self.fields and isinstance(f, TranslationField): - self.fields.pop(f.name) \ No newline at end of file + if f.name in fields and isinstance(f, TranslationField): + del fields[f.name] + return fields