nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Have a 30 second cooldown between transcodes

Browse source
This commit is contained in:
Nexus 2024-05-05 20:13:39 +01:00
parent 7a43c204d3
commit 5e1b0bb22e
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/cogs/auto_responder.py
View file

@ -88,6 +88,13 @@ class AutoResponder(commands.Cog):
links.append(url.geturl() if raw is False else url) links.append(url.geturl() if raw is False else url)
return links return links
def _cooldown_transcode(self, timeout: float = 30.0):
async def _i():
async with self.transcode_lock:
await asyncio.sleep(timeout)
self.bot.loop.create_task(_i())
async def _transcode_hevc_to_h264( async def _transcode_hevc_to_h264(
self, uri: str | pathlib.Path, *, update: discord.Message = None self, uri: str | pathlib.Path, *, update: discord.Message = None
) -> tuple[discord.File, pathlib.Path] | None: ) -> tuple[discord.File, pathlib.Path] | None:
@ -261,6 +268,7 @@ class AutoResponder(commands.Cog):
_r = await self._transcode_hevc_to_h264(link, update=message) _r = await self._transcode_hevc_to_h264(link, update=message)
if not _r: if not _r:
continue continue
self._cooldown_transcode()
file, _p = _r file, _p = _r
if file: if file:
if _p.stat().st_size <= 24.5 * 1024 * 1024: if _p.stat().st_size <= 24.5 * 1024 * 1024: