File tree Expand file tree Collapse file tree 1 file changed +2
-23
lines changed
services/storage/src/simcore_service_storage/modules/celery Expand file tree Collapse file tree 1 file changed +2
-23
lines changed Original file line number Diff line number Diff line change 1- from functools import partial
21from pathlib import Path
3- from typing import Any
42
53from kombu .utils .json import register_type # type: ignore[import-untyped]
6- from pydantic import BaseModel
74
85
96def _path_encoder (obj ):
@@ -19,32 +16,14 @@ def _path_decoder(obj):
1916 return obj
2017
2118
22- def _class_full_name (clz : type [ BaseModel ] ) -> str :
19+ def _class_full_name (clz : type ) -> str :
2320 return "." .join ([clz .__module__ , clz .__qualname__ ])
2421
2522
26- def _encoder (obj : BaseModel , * args , ** kwargs ) -> dict [str , Any ]:
27- return obj .model_dump (* args , ** kwargs )
28-
29-
30- def _decoder (clz : type [BaseModel ], data : dict [str , Any ]) -> BaseModel :
31- return clz (** data )
32-
33-
34- def _register_pydantic_types (* models : type [BaseModel ]) -> None :
35- for model in models :
36- register_type (
37- model ,
38- _class_full_name (model ),
39- encoder = _encoder ,
40- decoder = partial (_decoder , model ),
41- )
42-
43-
4423def register_celery_types () -> None :
4524 register_type (
4625 Path ,
47- "." . join ([ Path . __module__ , Path . __qualname__ ] ),
26+ _class_full_name ( Path ),
4827 _path_encoder ,
4928 _path_decoder ,
5029 )
You can’t perform that action at this time.
0 commit comments