artie/artemis
1
0
Fork 0
mirror of https://github.com/artiemis/artemis.git synced 2026-02-14 08:31:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

types

Browse Source
This commit is contained in:
artie 2024-03-02 15:19:06 +01:00
parent b84cedad3b
commit 40f9e8e4d5
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cogs/ocr.py
View File

@ -27,8 +27,8 @@ class OCR(commands.Cog):
async def ocr_impl( async def ocr_impl(
self, self,
ctx: commands.Context, ctx: commands.Context,
flags: OCRFlags | OCRTranslateFlags, flags: OCRFlags | OCRTranslateFlags | None,
translate: Literal["gt", "deepl"] = None, translate: Literal["gt", "deepl"] | None = None,
): ):
if flags: if flags:
url = flags.url url = flags.url
@ -76,7 +76,7 @@ class OCR(commands.Cog):
return await ctx.reply(file=discord.File(StringIO(text), "ocr.txt")) return await ctx.reply(file=discord.File(StringIO(text), "ocr.txt"))
await ctx.reply(self.bot.codeblock(text, "")) await ctx.reply(self.bot.codeblock(text, ""))
async def lens_impl(self, ctx: commands.Context[Artemis], url: str) -> str: async def lens_impl(self, ctx: commands.Context[Artemis], url: str | None) -> str:
headers = {"User-Agent": self.bot.user_agent} headers = {"User-Agent": self.bot.user_agent}
cookies = { cookies = {
"CONSENT": "PENDING+137", "CONSENT": "PENDING+137",
@ -138,7 +138,9 @@ class OCR(commands.Cog):
""" """
await self.ocr_impl(ctx, flags, translate="gt") await self.ocr_impl(ctx, flags, translate="gt")
@commands.command(usage="[source:auto] [lang:eng] [l:eng] [s:auto] [dest:en] [d:en] <url>") @commands.command(
aliases=["ocrdl"], usage="[source:auto] [lang:eng] [l:eng] [s:auto] [dest:en] [d:en] <url>"
)
@commands.cooldown(1, 2, commands.BucketType.default) @commands.cooldown(1, 2, commands.BucketType.default)
async def ocrdeepl(self, ctx: commands.Context, *, flags: Optional[OCRTranslateFlags]): async def ocrdeepl(self, ctx: commands.Context, *, flags: Optional[OCRTranslateFlags]):
""" """
@ -159,7 +161,7 @@ class OCR(commands.Cog):
return await ctx.reply(file=discord.File(StringIO(text), "lens.txt")) return await ctx.reply(file=discord.File(StringIO(text), "lens.txt"))
await ctx.reply(self.bot.codeblock(text, "")) await ctx.reply(self.bot.codeblock(text, ""))
@commands.command(aliases=["ocrtr"]) @commands.command()
@commands.max_concurrency(1) @commands.max_concurrency(1)
@commands.cooldown(1, 10, commands.BucketType.default) @commands.cooldown(1, 10, commands.BucketType.default)
async def lensgt(self, ctx: commands.Context, *, url: Optional[str]): async def lensgt(self, ctx: commands.Context, *, url: Optional[str]):
@ -171,7 +173,7 @@ class OCR(commands.Cog):
cmd = self.bot.get_command("gt") cmd = self.bot.get_command("gt")
await cmd(ctx, flags=flags) await cmd(ctx, flags=flags)
@commands.command(aliases=["lenstr"]) @commands.command(aliases=["lensdl", "lenstr"])
@commands.max_concurrency(1) @commands.max_concurrency(1)
@commands.cooldown(1, 10, commands.BucketType.default) @commands.cooldown(1, 10, commands.BucketType.default)
async def lensdeepl(self, ctx: commands.Context, *, url: Optional[str]): async def lensdeepl(self, ctx: commands.Context, *, url: Optional[str]):