tropicoo/yt-dlp-bot
1
1
Fork 0
mirror of https://github.com/tropicoo/yt-dlp-bot.git synced 2025-03-01 01:03:07 +08:00
Code Issues Projects Releases Packages Wiki Activity

Maintenance

Browse source
This commit is contained in:
Taras Terletskyi 2023-04-17 18:51:26 +03:00
parent a1b916949a
commit c0d708eb1a
3 changed files with 8 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
README.md
View file

@ -104,8 +104,7 @@ details
- **API**: default port is `1984` and no auth. Port can be changed - **API**: default port is `1984` and no auth. Port can be changed
in `docker-compose.yml` in `docker-compose.yml`
- **RabbitMQ**: default credentials are located in `envs/.env_common` - **RabbitMQ**: default credentials are located in `envs/.env_common`
- **PostgreSQL**: default credentials are located in `envs/.env_common`. Same credentials are stored - **PostgreSQL**: default credentials are located in `envs/.env_common`.
for Alembic in `app_worker/alembic.ini` on 53rd line.
## API ## API

14
app_bot/bot/core/handlers/success.py
View file

@ -29,8 +29,8 @@ class SuccessHandler(AbstractHandler):
self._rmq_publisher = RmqPublisher() self._rmq_publisher = RmqPublisher()
async def handle(self) -> None: async def handle(self) -> None:
try:
coro_tasks = [] coro_tasks = []
try:
for media_object in self._body.media.get_media_objects(): for media_object in self._body.media.get_media_objects():
coro_tasks.append(self._handle(media_object)) coro_tasks.append(self._handle(media_object))
await asyncio.gather(*coro_tasks) await asyncio.gather(*coro_tasks)
@ -126,22 +126,20 @@ class SuccessHandler(AbstractHandler):
user = self._bot.allowed_users[self._get_sender_id()] user = self._bot.allowed_users[self._get_sender_id()]
return user.upload.upload_video_file return user.upload.upload_video_file
def _validate_file_size_for_upload(self, media_object: BaseMedia) -> None: def _validate_file_size_for_upload(self, media_obj: BaseMedia) -> None:
if self._body.context.source is TaskSource.API: if self._body.context.source is TaskSource.API:
max_file_size = self._bot.conf.telegram.api.upload_video_max_file_size max_file_size = self._bot.conf.telegram.api.upload_video_max_file_size
else: else:
user = self._bot.allowed_users[self._get_sender_id()] user = self._bot.allowed_users[self._get_sender_id()]
max_file_size = user.upload.upload_video_max_file_size max_file_size = user.upload.upload_video_max_file_size
if not os.path.exists(media_object.filepath): if not os.path.exists(media_obj.filepath):
raise ValueError( raise ValueError(f'{media_obj.file_type} {media_obj.filepath} not found')
f'{media_object.file_type} {media_object.filepath} not found'
)
file_size = os.stat(media_object.filepath).st_size file_size = os.stat(media_obj.filepath).st_size
if file_size > max_file_size: if file_size > max_file_size:
err_msg = ( err_msg = (
f'{media_object.file_type} file size {file_size} bytes bigger than ' f'{media_obj.file_type} file size {file_size} bytes bigger than '
f'allowed {max_file_size} bytes. Will not upload' f'allowed {max_file_size} bytes. Will not upload'
) )
self._log.warning(err_msg) self._log.warning(err_msg)

2
app_worker/alembic.ini
View file

@ -50,7 +50,7 @@ version_path_separator = os # default: use os.pathsep
# are written from script.py.mako # are written from script.py.mako
# output_encoding = utf-8 # output_encoding = utf-8
sqlalchemy.url = postgresql+asyncpg://yt:yt@yt_postgres:5432/yt sqlalchemy.url = ""
[post_write_hooks] [post_write_hooks]