added some async code stabilize pyrogram
Browse files- App/Worker.py +8 -2
- requirements.txt +1 -0
App/Worker.py
CHANGED
@@ -6,7 +6,7 @@ from App import celery_config, bot
|
|
6 |
from typing import List
|
7 |
from App.Editor.Schema import EditorRequest, LinkInfo
|
8 |
from celery.signals import worker_process_init
|
9 |
-
|
10 |
|
11 |
celery = Celery()
|
12 |
celery.config_from_object(celery_config)
|
@@ -70,8 +70,14 @@ def render_video(directory: str, output_directory: str):
|
|
70 |
def cleanup_temp_directory(
|
71 |
temp_dir: str, output_dir: str, chat_id: int = -1002069945904
|
72 |
):
|
|
|
|
|
|
|
|
|
73 |
try:
|
74 |
-
bot.send_video(
|
|
|
|
|
75 |
finally:
|
76 |
# Cleanup: Remove the temporary directory
|
77 |
shutil.rmtree(temp_dir, ignore_errors=True)
|
|
|
6 |
from typing import List
|
7 |
from App.Editor.Schema import EditorRequest, LinkInfo
|
8 |
from celery.signals import worker_process_init
|
9 |
+
from asgiref.sync import async_to_sync
|
10 |
|
11 |
celery = Celery()
|
12 |
celery.config_from_object(celery_config)
|
|
|
70 |
def cleanup_temp_directory(
|
71 |
temp_dir: str, output_dir: str, chat_id: int = -1002069945904
|
72 |
):
|
73 |
+
async_to_sync(SendVideo(temp_dir, output_dir, chat_id=-1002069945904))
|
74 |
+
|
75 |
+
|
76 |
+
async def SendVideo(temp_dir: str, output_dir: str, chat_id: int = -1002069945904):
|
77 |
try:
|
78 |
+
await bot.send_video(
|
79 |
+
-1002069945904, video=output_dir, caption="Your video caption"
|
80 |
+
)
|
81 |
finally:
|
82 |
# Cleanup: Remove the temporary directory
|
83 |
shutil.rmtree(temp_dir, ignore_errors=True)
|
requirements.txt
CHANGED
@@ -23,3 +23,4 @@ python-multipart
|
|
23 |
git+https://github.com/redis/redis-py.git
|
24 |
TgCrypto
|
25 |
pyrogram==2.0.100
|
|
|
|
23 |
git+https://github.com/redis/redis-py.git
|
24 |
TgCrypto
|
25 |
pyrogram==2.0.100
|
26 |
+
asgiref
|