mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 17:44:50 +00:00
a7008c664e
* ci: Update .deepsource.toml
* ci: Update .deepsource.toml
* style: format code with black and isort
Format code with black and isort
This commit fixes the style issues introduced in 0fb651a according to the output
from Black and isort.
Details: https://app.deepsource.com/gh/yasirarism/MissKatyPyro/transform/d8f2f66e-b496-4686-aca6-9830236eda12/
---------
Co-authored-by: deepsource-io[bot] <42547082+deepsource-io[bot]@users.noreply.github.com>
Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
62 lines
2 KiB
Python
62 lines
2 KiB
Python
"""
|
|
* @author yasir <yasiramunandar@gmail.com>
|
|
* @date 2022-12-01 09:12:27
|
|
* @projectName MissKatyPyro
|
|
* Copyright @YasirPedia All rights reserved
|
|
"""
|
|
|
|
|
|
import os
|
|
|
|
from pyrogram import Client, filters
|
|
from pyrogram.types import Message
|
|
from telegraph.aio import Telegraph
|
|
|
|
from misskaty import app
|
|
from misskaty.core.decorator.errors import capture_err
|
|
from misskaty.core.decorator.ratelimiter import ratelimiter
|
|
from misskaty.helper.http import http
|
|
from misskaty.helper.localization import use_chat_lang
|
|
from misskaty.vars import COMMAND_HANDLER
|
|
|
|
__MODULE__ = "OCR"
|
|
__HELP__ = "/ocr [reply to photo] - Read Text From Image"
|
|
|
|
|
|
@app.on_message(filters.command(["ocr"], COMMAND_HANDLER))
|
|
@capture_err
|
|
@ratelimiter
|
|
@use_chat_lang()
|
|
async def ocr(self: Client, ctx: Message, strings):
|
|
reply = ctx.reply_to_message
|
|
if (
|
|
not reply
|
|
or not reply.sticker
|
|
and not reply.photo
|
|
and (not reply.document or not reply.document.mime_type.startswith("image"))
|
|
):
|
|
return await ctx.reply_msg(
|
|
strings("no_photo").format(cmd=ctx.command[0]), quote=True
|
|
)
|
|
msg = await ctx.reply_msg(strings("read_ocr"), quote=True)
|
|
try:
|
|
file_path = await reply.download()
|
|
if reply.sticker:
|
|
file_path = await reply.download(
|
|
f"ocr_{ctx.from_user.id if ctx.from_user else ctx.sender_chat.id}.jpg"
|
|
)
|
|
response = await Telegraph().upload_file(file_path)
|
|
url = f"https://telegra.ph{response[0]['src']}"
|
|
req = (
|
|
await http.get(
|
|
f"https://script.google.com/macros/s/AKfycbwURISN0wjazeJTMHTPAtxkrZTWTpsWIef5kxqVGoXqnrzdLdIQIfLO7jsR5OQ5GO16/exec?url={url}",
|
|
follow_redirects=True,
|
|
)
|
|
).json()
|
|
await msg.edit_msg(strings("result_ocr").format(result=req["text"]))
|
|
if os.path.exists(file_path):
|
|
os.remove(file_path)
|
|
except Exception as e:
|
|
await msg.edit_msg(str(e))
|
|
if os.path.exists(file_path):
|
|
os.remove(file_path)
|