Coverage for polar/worker/_encoder.py: 85%

1import json 2 ctx1ab

2import uuid 2 ctx1ab

3from typing import Any 2 ctx1ab

4 

5import dramatiq 2 ctx1ab

6 

7 

8def _json_obj_serializer(obj: Any) -> Any: 2 ctx1ab

9 if isinstance(obj, uuid.UUID): 9 ↛ 11line 9 didn't jump to line 11 because the condition on line 9 was always true1 ctx1c

10 return str(obj) 1 ctx1c

11 raise TypeError(f"Object of type {type(obj)} is not JSON serializable") 

12 

13 

14class JSONEncoder(dramatiq.JSONEncoder): 2 ctx1ab

15 def encode(self, data: dict[str, Any]) -> bytes: 2 ctx1ab

16 return json.dumps( 1 ctx1c

17 data, separators=(",", ":"), default=_json_obj_serializer 

18 ).encode("utf-8")