added names
Browse files- App/Worker.py +11 -9
App/Worker.py
CHANGED
@@ -20,10 +20,12 @@ celery.conf.update(
|
|
20 |
|
21 |
@worker_process_init.connect
|
22 |
def worker_process_init_handler(**kwargs):
|
23 |
-
|
|
|
|
|
24 |
|
25 |
|
26 |
-
@celery.task
|
27 |
def create_json_file(assets: List[Assets], asset_dir: str):
|
28 |
for asset in assets:
|
29 |
filename = f"{asset.type.capitalize()}Sequences.json"
|
@@ -53,7 +55,7 @@ def download_with_wget(link, download_dir, filename):
|
|
53 |
subprocess.run(["aria2c", link, "-d", download_dir, "-o", filename])
|
54 |
|
55 |
|
56 |
-
@celery.task
|
57 |
def copy_remotion_app(src: str, dest: str):
|
58 |
shutil.copytree(src, dest)
|
59 |
|
@@ -62,7 +64,7 @@ def copy_remotion_app(src: str, dest: str):
|
|
62 |
# create_symlink(source_dir, target_dir=dest, symlink_name="node_module")
|
63 |
|
64 |
|
65 |
-
@celery.task
|
66 |
def unsilence(directory: str):
|
67 |
output_dir = os.path.join(directory, "out/video.mp4")
|
68 |
shortered_dir = os.path.join(directory, "out/temp.mp4")
|
@@ -71,13 +73,13 @@ def unsilence(directory: str):
|
|
71 |
os.rename(shortered_dir, output_dir)
|
72 |
|
73 |
|
74 |
-
@celery.task
|
75 |
def install_dependencies(directory: str):
|
76 |
os.chdir(directory)
|
77 |
os.system("npm install")
|
78 |
|
79 |
|
80 |
-
@celery.task
|
81 |
def download_assets(links: List[LinkInfo], temp_dir: str):
|
82 |
public_dir = f"{temp_dir}/public"
|
83 |
for link in links:
|
@@ -86,14 +88,14 @@ def download_assets(links: List[LinkInfo], temp_dir: str):
|
|
86 |
download_with_wget(file_link, public_dir, file_name)
|
87 |
|
88 |
|
89 |
-
@celery.task
|
90 |
def render_video(directory: str, output_directory: str):
|
91 |
os.chdir(directory)
|
92 |
os.system(f"npm run build --output {output_directory}")
|
93 |
print("complete")
|
94 |
|
95 |
|
96 |
-
@celery.task
|
97 |
def cleanup_temp_directory(
|
98 |
temp_dir: str, output_dir: str, chat_id: int = -1002069945904
|
99 |
):
|
@@ -107,7 +109,7 @@ def cleanup_temp_directory(
|
|
107 |
shutil.rmtree(temp_dir, ignore_errors=True)
|
108 |
|
109 |
|
110 |
-
@celery.task
|
111 |
def celery_task(video_task: EditorRequest):
|
112 |
remotion_app_dir = os.path.join("/srv", "Remotion-app")
|
113 |
project_id = str(uuid.uuid4())
|
|
|
20 |
|
21 |
@worker_process_init.connect
|
22 |
def worker_process_init_handler(**kwargs):
|
23 |
+
name = kwargs.get("name")
|
24 |
+
if name == "send":
|
25 |
+
bot.start()
|
26 |
|
27 |
|
28 |
+
@celery.task(name="CreateFile")
|
29 |
def create_json_file(assets: List[Assets], asset_dir: str):
|
30 |
for asset in assets:
|
31 |
filename = f"{asset.type.capitalize()}Sequences.json"
|
|
|
55 |
subprocess.run(["aria2c", link, "-d", download_dir, "-o", filename])
|
56 |
|
57 |
|
58 |
+
@celery.task(name="CopyRemotion")
|
59 |
def copy_remotion_app(src: str, dest: str):
|
60 |
shutil.copytree(src, dest)
|
61 |
|
|
|
64 |
# create_symlink(source_dir, target_dir=dest, symlink_name="node_module")
|
65 |
|
66 |
|
67 |
+
@celery.task(name="Unsilence")
|
68 |
def unsilence(directory: str):
|
69 |
output_dir = os.path.join(directory, "out/video.mp4")
|
70 |
shortered_dir = os.path.join(directory, "out/temp.mp4")
|
|
|
73 |
os.rename(shortered_dir, output_dir)
|
74 |
|
75 |
|
76 |
+
@celery.task(name="InstallDependency")
|
77 |
def install_dependencies(directory: str):
|
78 |
os.chdir(directory)
|
79 |
os.system("npm install")
|
80 |
|
81 |
|
82 |
+
@celery.task(name="DownloadAssets")
|
83 |
def download_assets(links: List[LinkInfo], temp_dir: str):
|
84 |
public_dir = f"{temp_dir}/public"
|
85 |
for link in links:
|
|
|
88 |
download_with_wget(file_link, public_dir, file_name)
|
89 |
|
90 |
|
91 |
+
@celery.task(name="RenderFile")
|
92 |
def render_video(directory: str, output_directory: str):
|
93 |
os.chdir(directory)
|
94 |
os.system(f"npm run build --output {output_directory}")
|
95 |
print("complete")
|
96 |
|
97 |
|
98 |
+
@celery.task(name="send")
|
99 |
def cleanup_temp_directory(
|
100 |
temp_dir: str, output_dir: str, chat_id: int = -1002069945904
|
101 |
):
|
|
|
109 |
shutil.rmtree(temp_dir, ignore_errors=True)
|
110 |
|
111 |
|
112 |
+
@celery.task(name="All")
|
113 |
def celery_task(video_task: EditorRequest):
|
114 |
remotion_app_dir = os.path.join("/srv", "Remotion-app")
|
115 |
project_id = str(uuid.uuid4())
|