async await
Browse files- App/Worker.py +5 -8
- App/__init__.py +1 -1
App/Worker.py
CHANGED
|
@@ -85,19 +85,16 @@ def download_assets(links: List[LinkInfo], temp_dir: str):
|
|
| 85 |
def render_video(directory: str, output_directory: str):
|
| 86 |
os.chdir(directory)
|
| 87 |
os.system(f"npm run build --output {output_directory}")
|
| 88 |
-
bot.start()
|
| 89 |
-
bot.send_file(-1002069945904, file=output_directory, caption="Your video caption")
|
| 90 |
print("complete")
|
| 91 |
|
| 92 |
|
| 93 |
# @celery.task(name="send")
|
| 94 |
-
def cleanup_temp_directory(
|
| 95 |
temp_dir: str, output_dir: str, chat_id: int = -1002069945904
|
| 96 |
):
|
| 97 |
try:
|
| 98 |
print("sending...")
|
| 99 |
-
bot.
|
| 100 |
-
bot.send_file(chat_id, file=output_dir, caption="Your video caption")
|
| 101 |
except Exception as e:
|
| 102 |
print(e)
|
| 103 |
finally:
|
|
@@ -106,7 +103,7 @@ def cleanup_temp_directory(
|
|
| 106 |
|
| 107 |
|
| 108 |
# @celery.task(name="All")
|
| 109 |
-
def celery_task(video_task: EditorRequest):
|
| 110 |
remotion_app_dir = os.path.join("/srv", "Remotion-app")
|
| 111 |
project_id = str(uuid.uuid4())
|
| 112 |
temp_dir = f"/tmp/{project_id}"
|
|
@@ -119,8 +116,8 @@ def celery_task(video_task: EditorRequest):
|
|
| 119 |
create_json_file(video_task.assets, assets_dir),
|
| 120 |
download_assets(video_task.links, temp_dir) if video_task.links else None,
|
| 121 |
render_video(temp_dir, output_dir),
|
| 122 |
-
|
| 123 |
-
cleanup_temp_directory.si(temp_dir, output_dir),
|
| 124 |
|
| 125 |
# chain(
|
| 126 |
# copy_remotion_app.si(remotion_app_dir, temp_dir),
|
|
|
|
| 85 |
def render_video(directory: str, output_directory: str):
|
| 86 |
os.chdir(directory)
|
| 87 |
os.system(f"npm run build --output {output_directory}")
|
|
|
|
|
|
|
| 88 |
print("complete")
|
| 89 |
|
| 90 |
|
| 91 |
# @celery.task(name="send")
|
| 92 |
+
async def cleanup_temp_directory(
|
| 93 |
temp_dir: str, output_dir: str, chat_id: int = -1002069945904
|
| 94 |
):
|
| 95 |
try:
|
| 96 |
print("sending...")
|
| 97 |
+
await bot.send_file(chat_id, file=output_dir, caption="Your video caption")
|
|
|
|
| 98 |
except Exception as e:
|
| 99 |
print(e)
|
| 100 |
finally:
|
|
|
|
| 103 |
|
| 104 |
|
| 105 |
# @celery.task(name="All")
|
| 106 |
+
async def celery_task(video_task: EditorRequest):
|
| 107 |
remotion_app_dir = os.path.join("/srv", "Remotion-app")
|
| 108 |
project_id = str(uuid.uuid4())
|
| 109 |
temp_dir = f"/tmp/{project_id}"
|
|
|
|
| 116 |
create_json_file(video_task.assets, assets_dir),
|
| 117 |
download_assets(video_task.links, temp_dir) if video_task.links else None,
|
| 118 |
render_video(temp_dir, output_dir),
|
| 119 |
+
unsilence(temp_dir),
|
| 120 |
+
await cleanup_temp_directory.si(temp_dir, output_dir),
|
| 121 |
|
| 122 |
# chain(
|
| 123 |
# copy_remotion_app.si(remotion_app_dir, temp_dir),
|
App/__init__.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from telethon
|
| 2 |
from telethon.sessions import StringSession
|
| 3 |
import os
|
| 4 |
|
|
|
|
| 1 |
+
from telethon import TelegramClient
|
| 2 |
from telethon.sessions import StringSession
|
| 3 |
import os
|
| 4 |
|