351 lines
14 KiB
Python
351 lines
14 KiB
Python
""" This module contains utils to configure your account. """
|
||
|
||
from os import remove
|
||
from telethon.errors import ImageProcessFailedError, PhotoCropSizeSmallError
|
||
from telethon.errors.rpcerrorlist import PhotoExtInvalidError, UsernameOccupiedError, AboutTooLongError, \
|
||
FirstNameInvalidError, UsernameInvalidError, UsernameNotModifiedError
|
||
from telethon.tl.functions.account import UpdateProfileRequest, UpdateUsernameRequest
|
||
from telethon.tl.functions.photos import DeletePhotosRequest, GetUserPhotosRequest, UploadProfilePhotoRequest
|
||
from telethon.tl.functions.users import GetFullUserRequest
|
||
from telethon.tl.functions.contacts import BlockRequest, UnblockRequest
|
||
from telethon.tl.types import InputPhoto, MessageMediaPhoto, MessageEntityMentionName, MessageEntityPhone, User
|
||
from struct import error as StructError
|
||
from pagermaid import bot, log, silent
|
||
from pagermaid.utils import lang, alias_command
|
||
from pagermaid.listener import listener
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('username'),
|
||
description=lang('username_des'),
|
||
parameters="<username>")
|
||
async def username(context):
|
||
""" Reconfigure your username. """
|
||
if len(context.parameter) > 1:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}｝")
|
||
if len(context.parameter) == 1:
|
||
result = context.parameter[0]
|
||
else:
|
||
result = ""
|
||
try:
|
||
await bot(UpdateUsernameRequest(result))
|
||
except UsernameOccupiedError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('username_exist')}")
|
||
return
|
||
except UsernameInvalidError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('username_vaild')}")
|
||
return
|
||
except UsernameNotModifiedError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('username_exist')}")
|
||
return
|
||
await context.edit(lang('username_set'))
|
||
if result == "":
|
||
await log(lang('username_cancel'))
|
||
return
|
||
await log(f"{lang('username_whatset')}`{result}`")
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('name'),
|
||
description=lang('name_des'),
|
||
parameters="<first name> <last name>")
|
||
async def name(context):
|
||
""" Updates your display name. """
|
||
if len(context.parameter) == 2:
|
||
first_name = context.parameter[0]
|
||
last_name = context.parameter[1]
|
||
elif len(context.parameter) == 1:
|
||
first_name = context.parameter[0]
|
||
last_name = " "
|
||
else:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
return
|
||
try:
|
||
await bot(UpdateProfileRequest(
|
||
first_name=first_name,
|
||
last_name=last_name))
|
||
except FirstNameInvalidError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('name_vaild')}")
|
||
return
|
||
await context.edit(lang('name_set'))
|
||
if last_name != " ":
|
||
await log(f"{lang('name_whatset')}`{first_name} {last_name}`.")
|
||
else:
|
||
await log(f"{lang('name_whatset')}`{first_name}`.")
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('pfp'),
|
||
description=lang('pfp_des'))
|
||
async def pfp(context):
|
||
""" Sets your profile picture. """
|
||
reply = await context.get_reply_message()
|
||
photo = None
|
||
if not silent:
|
||
await context.edit(lang('pfp_process'))
|
||
if reply:
|
||
if reply.media:
|
||
if isinstance(reply.media, MessageMediaPhoto):
|
||
photo = await bot.download_media(message=reply.photo)
|
||
elif "image" in reply.media.document.mime_type.split('/'):
|
||
photo = await bot.download_file(reply.media.document)
|
||
else:
|
||
await context.edit(f"{lang('error_prefix')}{lang('pfp_e_notp')}")
|
||
|
||
if photo:
|
||
try:
|
||
await bot(UploadProfilePhotoRequest(
|
||
await bot.upload_file(photo)
|
||
))
|
||
try:
|
||
remove(photo)
|
||
except:
|
||
pass
|
||
await context.edit("头像修改成功啦 ~")
|
||
return
|
||
except PhotoCropSizeSmallError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('pfp_e_size')}")
|
||
except ImageProcessFailedError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('pfp_e_img')}")
|
||
except PhotoExtInvalidError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('pfp_e_notp')}")
|
||
await context.edit(f"{lang('error_prefix')}{lang('pfp_e_notp')}")
|
||
return
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('bio'),
|
||
description="设置咱的公开情报",
|
||
parameters="<string>")
|
||
async def bio(context):
|
||
""" Sets your bio. """
|
||
try:
|
||
await bot(UpdateProfileRequest(about=context.arguments))
|
||
except AboutTooLongError:
|
||
await context.edit(f"{lang('error_prefix')}{lang('bio_too_lang')}")
|
||
return
|
||
await context.edit(lang('bio_set'))
|
||
if context.arguments == "":
|
||
await log(lang('bio_cancel'))
|
||
return
|
||
await log(f"{lang('bio_whatset')}`{context.arguments}`.")
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('rmpfp'),
|
||
description=lang('rmpfp_des'),
|
||
parameters=f"<{lang('int')}>")
|
||
async def rmpfp(context):
|
||
""" Removes your profile picture. """
|
||
group = context.text[8:]
|
||
if group == 'all':
|
||
limit = 0
|
||
elif group.isdigit():
|
||
limit = int(group)
|
||
else:
|
||
limit = 1
|
||
|
||
pfp_list = await bot(GetUserPhotosRequest(
|
||
user_id=context.from_id,
|
||
offset=0,
|
||
max_id=0,
|
||
limit=limit))
|
||
input_photos = []
|
||
for sep in pfp_list.photos:
|
||
input_photos.append(
|
||
InputPhoto(
|
||
id=sep.id,
|
||
access_hash=sep.access_hash,
|
||
file_reference=sep.file_reference
|
||
)
|
||
)
|
||
await bot(DeletePhotosRequest(id=input_photos))
|
||
await context.edit(f"`{lang('rmpfp_p')}{len(input_photos)} {lang('rmpfp_l')}`")
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('profile'),
|
||
description=lang('profile_des'),
|
||
parameters="<username>")
|
||
async def profile(context):
|
||
""" Queries profile of a user. """
|
||
if len(context.parameter) > 1:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
return
|
||
if not silent:
|
||
await context.edit(lang('profile_process'))
|
||
if context.reply_to_msg_id:
|
||
reply_message = await context.get_reply_message()
|
||
if not reply_message:
|
||
return await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
user = reply_message.from_id
|
||
target_user = await context.client(GetFullUserRequest(user))
|
||
else:
|
||
if len(context.parameter) == 1:
|
||
user = context.parameter[0]
|
||
if user.isnumeric():
|
||
user = int(user)
|
||
else:
|
||
user_object = await context.client.get_me()
|
||
user = user_object.id
|
||
if context.message.entities is not None:
|
||
if isinstance(context.message.entities[0], MessageEntityMentionName):
|
||
user = context.message.entities[0].user_id
|
||
elif isinstance(context.message.entities[0], MessageEntityPhone):
|
||
user = int(context.parameter[0])
|
||
else:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
return
|
||
try:
|
||
user_object = await context.client.get_entity(user)
|
||
target_user = await context.client(GetFullUserRequest(user_object.id))
|
||
except (TypeError, ValueError, OverflowError, StructError) as exception:
|
||
if str(exception).startswith("Cannot find any entity corresponding to"):
|
||
await context.edit(f"{lang('error_prefix')}{lang('profile_e_no')}")
|
||
return
|
||
if str(exception).startswith("No user has"):
|
||
await context.edit(f"{lang('error_prefix')}{lang('profile_e_nou')}")
|
||
return
|
||
if str(exception).startswith("Could not find the input entity for") or isinstance(exception, StructError):
|
||
await context.edit(f"{lang('error_prefix')}{lang('profile_e_nof')}")
|
||
return
|
||
if isinstance(exception, OverflowError):
|
||
await context.edit(f"{lang('error_prefix')}{lang('profile_e_long')}")
|
||
return
|
||
raise exception
|
||
user_type = "Bot" if target_user.user.bot else lang('profile_user')
|
||
username_system = f"@{target_user.user.username}" if target_user.user.username is not None else (
|
||
lang('profile_noset'))
|
||
if not target_user.user.first_name:
|
||
await context.edit(f"{lang('error_prefix')}{lang('profile_e_no')}")
|
||
return
|
||
first_name = target_user.user.first_name.replace("\u2060", "")
|
||
last_name = target_user.user.last_name.replace("\u2060", "") if target_user.user.last_name is not None else (
|
||
lang('profile_noset')
|
||
)
|
||
biography = target_user.about if target_user.about is not None else lang('profile_nobio')
|
||
verified = lang('profile_yes') if target_user.user.verified else lang('profile_no')
|
||
restricted = lang('profile_yes') if target_user.user.restricted else lang('profile_no')
|
||
caption = f"**{lang('profile_name')}:** \n" \
|
||
f"{lang('profile_username')}: {username_system} \n" \
|
||
f"ID: {target_user.user.id} \n" \
|
||
f"{lang('profile_fname')}: {first_name} \n" \
|
||
f"{lang('profile_lname')}: {last_name} \n" \
|
||
f"{lang('profile_bio')}: {biography} \n" \
|
||
f"{lang('profile_gic')}: {target_user.common_chats_count} \n" \
|
||
f"{lang('profile_verified')}: {verified} \n" \
|
||
f"{lang('profile_restricted')}: {restricted} \n" \
|
||
f"{lang('profile_type')}: {user_type} \n" \
|
||
f"[{first_name}](tg://user?id={target_user.user.id})"
|
||
photo = await context.client.download_profile_photo(
|
||
target_user.user.id,
|
||
"./" + str(target_user.user.id) + ".jpg",
|
||
download_big=True
|
||
)
|
||
try:
|
||
reply_to = context.message.reply_to_msg_id
|
||
try:
|
||
await context.client.send_file(
|
||
context.chat_id,
|
||
photo,
|
||
caption=caption,
|
||
link_preview=False,
|
||
force_document=False,
|
||
reply_to=reply_to
|
||
)
|
||
if not photo.startswith("http"):
|
||
remove(photo)
|
||
await context.delete()
|
||
try:
|
||
remove(photo)
|
||
except:
|
||
pass
|
||
return
|
||
except TypeError:
|
||
await context.edit(caption)
|
||
except:
|
||
try:
|
||
await context.client.send_file(
|
||
context.chat_id,
|
||
photo,
|
||
caption=caption,
|
||
link_preview=False,
|
||
force_document=False
|
||
)
|
||
if not photo.startswith("http"):
|
||
remove(photo)
|
||
await context.delete()
|
||
try:
|
||
remove(photo)
|
||
except:
|
||
pass
|
||
return
|
||
except TypeError:
|
||
await context.edit(caption)
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('block'),
|
||
description=lang('block_des'),
|
||
parameters="(username/uid/reply)")
|
||
async def block_user(context):
|
||
""" Block an user. """
|
||
current_chat = await context.get_chat()
|
||
if len(context.parameter) > 1:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
return
|
||
if not silent:
|
||
await context.edit(lang('block_process'))
|
||
user = None
|
||
# Priority: reply > argument > current_chat
|
||
if context.reply_to_msg_id: # Reply to a user
|
||
reply_message = await context.get_reply_message()
|
||
if reply_message and reply_message.sender_id is not None:
|
||
user = reply_message.sender_id
|
||
if not user and len(context.parameter) == 1: # Argument provided
|
||
(raw_user,) = context.parameter
|
||
if raw_user.isnumeric():
|
||
user = int(raw_user)
|
||
elif context.message.entities is not None:
|
||
if isinstance(context.message.entities[0], MessageEntityMentionName):
|
||
user = context.message.entities[0].user_id
|
||
if not user and isinstance(current_chat, User):
|
||
user = current_chat.id
|
||
if not user:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
return
|
||
try:
|
||
if await context.client(BlockRequest(id=user)):
|
||
await context.edit(f"{lang('block_success')} `{user}`")
|
||
except Exception: # noqa
|
||
pass
|
||
await context.edit(f"`{user}` {lang('block_exist')}")
|
||
if isinstance(current_chat, User) and current_chat.id == user:
|
||
await context.delete()
|
||
|
||
|
||
@listener(is_plugin=False, outgoing=True, command=alias_command('unblock'),
|
||
description=lang('unblock_des'),
|
||
parameters="<username/uid/reply>")
|
||
async def unblock_user(context):
|
||
""" Unblock an user. """
|
||
if len(context.parameter) > 1:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
return
|
||
if not silent:
|
||
await context.edit(lang('unblock_process'))
|
||
user = None
|
||
if context.reply_to_msg_id:
|
||
reply_message = await context.get_reply_message()
|
||
if reply_message and reply_message.sender_id is not None:
|
||
user = reply_message.sender_id
|
||
if not user and len(context.parameter) == 1:
|
||
(raw_user,) = context.parameter
|
||
if raw_user.isnumeric():
|
||
user = int(raw_user)
|
||
elif context.message.entities is not None:
|
||
if isinstance(context.message.entities[0], MessageEntityMentionName):
|
||
user = context.message.entities[0].user_id
|
||
if not user:
|
||
await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
return
|
||
try:
|
||
if await context.client(UnblockRequest(id=user)):
|
||
await context.edit(f"{lang('unblock_success')} `{user}`")
|
||
except Exception:
|
||
pass
|
||
await context.edit(f"`{user}` {lang('unblock_exist')}")
|