Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update serializers.py #206

Merged
merged 12 commits into from
Dec 26, 2024
43 changes: 7 additions & 36 deletions api_app/analyzers_manager/serializers.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@
# See the file 'LICENSE' for copying permission.
from rest_framework import serializers as rfs

from ..models import PluginConfig, PythonModule
from ..models import PythonModule
from ..serializers.plugin import (
PluginConfigSerializer,
ParameterSerializer,
PythonConfigSerializer,
PythonConfigSerializerForMigration,
)
Expand All @@ -27,9 +27,6 @@ class Meta:


class AnalyzerConfigSerializer(PythonConfigSerializer):
plugin_config = rfs.ListField(
child=rfs.DictField(), write_only=True, required=False
)
python_module = rfs.SlugRelatedField(
queryset=PythonModule.objects.all(), slug_field="module"
)
Expand All @@ -39,37 +36,11 @@ class Meta:
exclude = PythonConfigSerializer.Meta.exclude
list_serializer_class = PythonConfigSerializer.Meta.list_serializer_class

def create(self, validated_data):
plugin_config = validated_data.pop("plugin_config", {})
pc = super().create(validated_data)

# create plugin config
for config in plugin_config:
plugin_config_serializer = PluginConfigSerializer(
data=config, context={"request": self.context["request"]}
)
plugin_config_serializer.is_valid(raise_exception=True)
plugin_config_serializer.save()
return pc

def update(self, instance, validated_data):
plugin_config = validated_data.pop("plugin_config", [])
pc = super().update(instance, validated_data)

# update plugin config
for config in plugin_config:
plugin_config_serializer = PluginConfigSerializer(
data=config, context={"request": self.context["request"]}
)
plugin_config_serializer.is_valid(raise_exception=True)
PluginConfig.objects.filter(
owner=self.context["request"].user,
analyzer_config=plugin_config_serializer.validated_data[
"analyzer_config"
],
parameter=plugin_config_serializer.validated_data["parameter"],
).update_or_create(plugin_config_serializer.validated_data)
return pc
def to_representation(self, instance):
result = super().to_representation(instance)
parameters = ParameterSerializer(instance.parameters, many=True)
result["parameters"] = parameters.data
return result


class AnalyzerConfigSerializerForMigration(PythonConfigSerializerForMigration):
Expand Down
Loading