fxreader.online/freelance-project-34-marketing-blog
2
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

[~] Refactor

Browse Source
This commit is contained in:
Siarhei Siniak 2024-07-07 12:14:24 +03:00
parent a1a0a52e4d
commit 4d2e9a3fd7
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
python/tasks/tiktok/tasks.py
View File

@ -383,12 +383,23 @@ def audio_get() -> audio_get_t.res_t:
url=url, url=url,
) )
class process_graph_t:
@dataclasses_json.dataclass_json
@dataclasses.dataclass
class res_t:
ordered_nodes: Iterable[str]=dataclasses.field(default_factory=list)
done_ids: Iterable[str]=dataclasses.field(default_factory=set)
error_ids: Iterable[str]=dataclasses.field(default_factory=set)
task_ids: dict[str, str]=dataclasses.field(default_factory=dict)
pending_ids: Iterable[str]=dataclasses.field(default_factory=set)
done_tasks: Iterable[celery.result.AsyncResult]=dataclasses.field(default_factory=dict)
@shared_task() @shared_task()
def process_graph( def process_graph(
nodes: dict[str, Any], nodes: dict[str, Any],
data_deps: dict[str, Iterable[str]], data_deps: dict[str, Iterable[str]],
execution_deps: dict[str, Iterable[str]], execution_deps: dict[str, Iterable[str]],
) -> Any: ) -> process_graph_t.res_t:
import networkx import networkx
g_data = networkx.DiGraph() g_data = networkx.DiGraph()
@ -466,7 +477,7 @@ def process_graph(
del task del task
node_id += 1 node_id += 1
return dict( return process_graph_t.res_t(
ordered_nodes=ordered_nodes, ordered_nodes=ordered_nodes,
done_ids=done_ids, done_ids=done_ids,
done_tasks={ done_tasks={