68839e9dd0
Co-authored-by: New-dev0 <New-dev0@notavailable.live> Co-authored-by: Danish <danish@ultroid.tech> Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com> Co-authored-by: Programming Error <error@notavailable.live> Co-authored-by: Aakash <BLUE-DEVIL1134@users.noreply.github.com> Co-authored-by: Aditya <me@xditya.me> Co-authored-by: sppidy <sppidy@users.noreply.github.com> Co-authored-by: Arnab Paryali <Arnabxd@users.noreply.github.com> Co-authored-by: divkix <divkix@users.noreply.github.com> Co-authored-by: hellboi_atul <hellboi-atul@users.noreply.github.com>
111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
# Ultroid - UserBot
|
|
# Copyright (C) 2020 TeamUltroid
|
|
#
|
|
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
|
|
# PLease read the GNU Affero General Public License in
|
|
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
|
|
|
|
"""
|
|
✘ Commands Available -
|
|
|
|
•`qrcode <text/reply to text>`
|
|
Makes qrcode of text
|
|
|
|
•`addqr <reply image> <text>`
|
|
Makes qr of text and add it to image.
|
|
|
|
•`qrdecode <reply to qrcode>`
|
|
It decodes the qrcode.
|
|
"""
|
|
|
|
|
|
import os
|
|
|
|
import cv2
|
|
import qrcode
|
|
from PIL import Image
|
|
from telethon.tl.types import MessageMediaDocument as doc
|
|
from telethon.tl.types import MessageMediaPhoto as photu
|
|
|
|
from . import *
|
|
|
|
|
|
@ultroid_cmd(pattern="qrcode ?(.*)")
|
|
async def cd(e):
|
|
reply = await e.get_reply_message()
|
|
msg = e.pattern_match.group(1)
|
|
if reply and reply.text:
|
|
msg = reply.text
|
|
elif msg:
|
|
msg = msg
|
|
else:
|
|
return await eod(e, "`Give Some Text or Reply")
|
|
kk = await eor(e, "`processing`")
|
|
pfp = await ultroid_bot.get_profile_photos(ultroid_bot.uid)
|
|
img = "resources/extras/teamultroid.jpg"
|
|
if len(pfp) >= 1:
|
|
img = await ultroid_bot.download_media(pfp[0])
|
|
ok = Image.open(img)
|
|
logo = ok.resize((60, 60))
|
|
cod = qrcode.QRCode(error_correction=qrcode.constants.ERROR_CORRECT_H)
|
|
cod.add_data(msg)
|
|
cod.make()
|
|
imgg = cod.make_image().convert("RGB")
|
|
pstn = ((imgg.size[0] - logo.size[0]) // 2, (imgg.size[1] - logo.size[1]) // 2)
|
|
imgg.paste(logo, pstn)
|
|
imgg.save(img)
|
|
await ultroid_bot.send_file(e.chat_id, img, support_stream=True)
|
|
await kk.delete()
|
|
os.remove(img)
|
|
|
|
|
|
@ultroid_cmd(pattern="addqr ?(.*)")
|
|
async def qrwater(e):
|
|
msg = e.pattern_match.group(1)
|
|
r = await e.get_reply_message()
|
|
if not (msg and r and r.media):
|
|
return await eod(e, "`Reply Any Media and Give Text`")
|
|
kk = await eor(e, "`processing`")
|
|
if isinstance(r.media, photu):
|
|
dl = await ultroid_bot.download_media(r.media)
|
|
elif isinstance(r.media, doc):
|
|
dl = await ultroid_bot.download_media(r, thumb=-1)
|
|
else:
|
|
return
|
|
img_bg = Image.open(dl)
|
|
qr = qrcode.QRCode(box_size=5)
|
|
qr.add_data(msg)
|
|
qr.make()
|
|
img_qr = qr.make_image()
|
|
pos = (img_bg.size[0] - img_qr.size[0], img_bg.size[1] - img_qr.size[1])
|
|
img_bg.paste(img_qr, pos)
|
|
img_bg.save(dl)
|
|
await ultroid_bot.send_file(e.chat_id, dl, support_stream=True)
|
|
await kk.delete()
|
|
os.remove(dl)
|
|
|
|
|
|
@ultroid_cmd(pattern="qrdecode$")
|
|
async def decod(e):
|
|
r = await e.get_reply_message()
|
|
if not (r and r.media):
|
|
return await eod(e, "`Reply to Qrcode Media`")
|
|
kk = await eor(e, "`processing`")
|
|
if isinstance(r.media, photu):
|
|
dl = await ultroid_bot.download_media(r.media)
|
|
elif isinstance(r.media, doc):
|
|
dl = await ultroid_bot.download_media(r, thumb=-1)
|
|
else:
|
|
return
|
|
im = cv2.imread(dl)
|
|
try:
|
|
det = cv2.QRCodeDetector()
|
|
tx, y, z = det.detectAndDecode(im)
|
|
await kk.edit("**Decoded Text:\n\n**" + tx)
|
|
except BaseException:
|
|
await kk.edit("`Reply To Media in Which Qr image present.`")
|
|
os.remove(dl)
|
|
|
|
|
|
HELP.update({f"{__name__.split('.')[1]}": f"{__doc__.format(i=HNDLR)}"})
|