From aa9a2709806ded36faa60cdb5d7b8bca1c80ecf9 Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@gmail.com>
Date: Sun, 7 Jul 2024 13:33:02 +0300
Subject: [PATCH] [~] Refactor

---
 docker/tiktok/docker-compose.yml |  2 +-
 python/tasks/tiktok/utils.py     | 10 ++++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/docker/tiktok/docker-compose.yml b/docker/tiktok/docker-compose.yml
index 156e02f..f6aee7b 100644
--- a/docker/tiktok/docker-compose.yml
+++ b/docker/tiktok/docker-compose.yml
@@ -25,4 +25,4 @@ services:
       - ./../../tmp/cache/tiktok:/app/tmp/cache/tiktok:rw
       - ./../../python/tasks/tiktok:/app/python/tasks/tiktok:ro
     command:
-      - celery -A python.tasks.tiktok.celery worker -c 1
+      - celery -A python.tasks.tiktok.celery worker -c 2
diff --git a/python/tasks/tiktok/utils.py b/python/tasks/tiktok/utils.py
index c27e38d..3597fa4 100644
--- a/python/tasks/tiktok/utils.py
+++ b/python/tasks/tiktok/utils.py
@@ -199,7 +199,10 @@ class Task(celery.app.task.Task):
             for current_name in class_names:
                 c = getattr(c, current_name)
 
-            return c(**data['data'])
+            return c.from_dict({
+                k : cls._loads(data=v)
+                for k, v in data['data'].items()
+            })
         else:
             if isinstance(data, list):
                 return [
@@ -229,7 +232,10 @@ class Task(celery.app.task.Task):
                 type='dataclass_json',
                 module=data.__class__.__module__,
                 _class=data.__class__.__qualname__,
-                data=data.to_dict(),
+                data={
+                    k : cls._dumps(v, need_native=True,)
+                    for k, v in data.__dict__.items()
+                },
             )
         else:
             if isinstance(data, (list, tuple)):