Spaces:
Paused
Paused
File size: 6,098 Bytes
ea7f80a c46d195 be3098f ea7f80a be3098f ea7f80a c46d195 ea7f80a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 |
# Ultroid - UserBot
# Copyright (C) 2021-2022 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/>.
import random
from pyUltroid.functions.misc import Quotly
from pyUltroid.functions.tools import TgConverter
from telethon import errors
from telethon.errors.rpcerrorlist import StickersetInvalidError
from telethon.tl.functions.messages import GetStickerSetRequest as GetSticker
from telethon.tl.functions.messages import UploadMediaRequest
from telethon.tl.functions.stickers import AddStickerToSetRequest as AddSticker
from telethon.tl.functions.stickers import CreateStickerSetRequest
from telethon.tl.types import InputPeerSelf
from telethon.tl.types import InputStickerSetItem as SetItem
from telethon.tl.types import InputStickerSetShortName, User
from telethon.utils import get_display_name, get_input_document
from . import LOGS, asst, asst_cmd, udB
@asst_cmd(
pattern="kang",
)
async def kang_cmd(ult):
sender = await ult.get_sender()
if not isinstance(sender, User):
return
if not ult.is_reply:
return await ult.eor("`Reply to a sticker/photo..`", time=5)
reply = await ult.get_reply_message()
if sender.username:
pre = sender.username[:4]
else:
pre = random.random_string(length=3)
animated, dl = None, None
try:
emoji = ult.text.split(maxsplit=1)[1]
except IndexError:
emoji = None
if reply.sticker:
file = get_input_document(reply.sticker)
emoji = emoji or reply.file.emoji
if reply.file.name.endswith(".tgs"):
animated = True
dl = await reply.download_media()
elif reply.photo:
dl = await reply.download_media()
name = "sticker.webp"
image = TgConverter.resize_photo_sticker(dl)
image.save(name, "WEBP")
elif reply.text:
dl = await Quotly().create_quotly(reply)
else:
return await ult.eor("`Reply to sticker or text to add it in your pack...`")
if not emoji:
emoji = "🏵"
if dl:
upl = await ult.client.upload_file(dl)
file = get_input_document(
await ult.client(UploadMediaRequest(InputPeerSelf(), upl))
)
get_ = udB.get_key("STICKERS") or {}
type_ = "static" if not animated else "anim"
if not get_.get(ult.sender_id) or not get_.get(ult.sender_id, {}).get(type_):
sn = f"{pre}_{ult.sender_id}"
title = f"{get_display_name(sender)}'s Kang Pack"
if animated:
type_ = "anim"
sn += "_anim"
title += " (Animated)"
sn += f"_by_{asst.me.username}"
try:
await asst(GetSticker(InputStickerSetShortName(sn), hash=0))
sn = sn.replace(str(ult.sender_id), f"{ult.sender_id}_{ult.id}")
except StickersetInvalidError:
pass
try:
pack = await ult.client(
CreateStickerSetRequest(
user_id=sender.id,
title=title,
short_name=sn,
stickers=[SetItem(file, emoji=emoji)],
animated=animated,
)
)
except Exception as er:
return await ult.eor(str(er))
sn = pack.set.short_name
if not get_.get(ult.sender_id):
get_.update({ult.sender_id: {type_: [sn]}})
else:
get_[ult.sender_id].update({type_: [sn]})
udB.set_key("STICKERS", get_)
return await ult.reply(
f"**Kanged Successfully!\nEmoji :** {emoji}\n**Link :** [Click Here](https://t.me/addstickers/{sn})"
)
name = get_[ult.sender_id][type_][-1]
try:
await asst(GetSticker(InputStickerSetShortName(name), hash=0))
except StickersetInvalidError:
get_[ult.sender_id][type_].remove(name)
try:
await asst(
AddSticker(InputStickerSetShortName(name), SetItem(file, emoji=emoji))
)
except (errors.StickerpackStickersTooMuchError, errors.StickersTooMuchError):
sn = f"{pre}{ult.sender_id}_{ult.id}"
title = f"{get_display_name(sender)}'s Kang Pack"
if animated:
sn += "_anim"
title += " (Animated)"
sn += f"_by_{asst.me.username}"
try:
pack = await ult.client(
CreateStickerSetRequest(
user_id=sender.id,
title=title,
short_name=sn,
stickers=[SetItem(file, emoji=emoji)],
animated=animated,
)
)
except Exception as er:
return await ult.eor(str(er))
get_[ult.sender_id][type_].append(pack.set.short_name)
udB.set_key("STICKERS", get_)
return await ult.reply(
f"**Created New Kang Pack!\nEmoji :** {emoji}\n**Link :** [Click Here](https://t.me/addstickers/{sn})"
)
except Exception as er:
LOGS.exception(er)
return await ult.reply(str(er))
await ult.reply(
f"Sticker Added to Pack Successfully\n**Link :** [Click Here](https://t.me/addstickers/{name})"
)
@asst_cmd(pattern="listpack")
async def do_magic(ult):
ko = udB.get_key("STICKERS") or {}
if not ko.get(ult.sender_id):
return await ult.reply("No Sticker Pack Found!")
al_ = []
ul = ko[ult.sender_id]
if ul.get("static"):
al_.extend(ul["static"])
if ul.get("anim"):
al_.extend(ul["anim"])
msg = "• **Stickers Owned by You!**\n\n"
for _ in al_:
try:
pack = await ult.client(GetSticker(InputStickerSetShortName(_), hash=0))
msg += f"• [{pack.set.title}](https://t.me/addstickers/{_})\n"
except StickerSetInvalidError:
if ul.get("anim") and _ in ul["anim"]:
ul["anim"].remove(_)
else:
ul["static"].remove(_)
udB.set_key("STICKERS", ko)
await ult.reply(msg)
|