Skip to content

Commit 164fabd

Browse files
Fix test
1 parent 55d18c9 commit 164fabd

File tree

1 file changed

+20
-13
lines changed
  • src/openklant/components/klantinteracties/api/serializers

1 file changed

+20
-13
lines changed

src/openklant/components/klantinteracties/api/serializers/partijen.py

+20-13
Original file line numberDiff line numberDiff line change
@@ -248,30 +248,37 @@ class Meta:
248248

249249
@transaction.atomic
250250
def update(self, instance, validated_data):
251-
if partij := validated_data.pop("partij", None):
252-
validated_data["partij"] = Partij.objects.get(uuid=str(partij.get("uuid")))
253251

254-
if categorie := validated_data.pop("categorie", None):
252+
if "partij" in validated_data:
253+
validated_data["partij"] = Partij.objects.get(
254+
uuid=str(validated_data["partij"].get("uuid"))
255+
)
256+
257+
if "categorie" in validated_data:
255258
validated_data["categorie"] = Categorie.objects.get(
256-
uuid=str(categorie.get("uuid"))
259+
uuid=str(validated_data["categorie"].get("uuid"))
257260
)
258261

259262
return super().update(instance, validated_data)
260263

261264
@transaction.atomic
262265
def create(self, validated_data):
263-
begin_datum = validated_data.get(
264-
"begin_datum", datetime.datetime.today().strftime("%Y-%m-%d")
265-
)
266-
267-
validated_data["begin_datum"] = begin_datum
266+
if "begin_datum" in validated_data:
267+
begin_datum = validated_data["begin_datum"]
268+
validated_data["begin_datum"] = (
269+
begin_datum
270+
if begin_datum
271+
else datetime.datetime.today().strftime("%Y-%m-%d")
272+
)
268273

269-
if partij := validated_data.pop("partij", None):
270-
validated_data["partij"] = Partij.objects.get(uuid=str(partij.get("uuid")))
274+
if "partij" in validated_data:
275+
validated_data["partij"] = Partij.objects.get(
276+
uuid=str(validated_data["partij"].get("uuid"))
277+
)
271278

272-
if categorie := validated_data.pop("categorie", None):
279+
if "categorie" in validated_data:
273280
validated_data["categorie"] = Categorie.objects.get(
274-
uuid=str(categorie.get("uuid"))
281+
uuid=str(validated_data["categorie"].get("uuid"))
275282
)
276283

277284
return super().create(validated_data)

0 commit comments

Comments
 (0)