mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 05:18:10 +00:00
Code Issues Releases Wiki Activity

Add __slots__ to every single Pyrogram types

Browse Source
This commit is contained in:
Dan 2019-03-16 16:15:27 +01:00
parent ef9ed31589
commit e0f1f6aaeb
46 changed files with 592 additions and 386 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyrogram/client/types/bots/callback_game.py
View File

@ -25,5 +25,7 @@ class CallbackGame(PyrogramType):
Use BotFather to set up your game. Use BotFather to set up your game.
""" """
__slots__ = []
def __init__(self): def __init__(self):
super().__init__(None) super().__init__(None)

24
pyrogram/client/types/bots/callback_query.py
View File

@ -58,16 +58,20 @@ class CallbackQuery(PyrogramType, Update):
""" """
def __init__(self, __slots__ = ["id", "from_user", "chat_instance", "message", "inline_message_id", "data", "game_short_name"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
id: str, self,
from_user: User, *,
chat_instance: str, client: "pyrogram.client.ext.BaseClient",
message: "pyrogram.Message" = None, id: str,
inline_message_id: str = None, from_user: User,
data: bytes = None, chat_instance: str,
game_short_name: str = None): message: "pyrogram.Message" = None,
inline_message_id: str = None,
data: bytes = None,
game_short_name: str = None
):
super().__init__(client) super().__init__(client)
self.id = id self.id = id

8
pyrogram/client/types/bots/force_reply.py
View File

@ -33,8 +33,12 @@ class ForceReply(PyrogramType):
2) if the bot's message is a reply (has reply_to_message_id), sender of the original message. 2) if the bot's message is a reply (has reply_to_message_id), sender of the original message.
""" """
def __init__(self, __slots__ = ["selective"]
selective: bool = None):
def __init__(
self,
selective: bool = None
):
super().__init__(None) super().__init__(None)
self.selective = selective self.selective = selective

16
pyrogram/client/types/bots/game_high_score.py
View File

@ -37,12 +37,16 @@ class GameHighScore(PyrogramType):
Position in high score table for the game. Position in high score table for the game.
""" """
def __init__(self, __slots__ = ["user", "score", "position"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
user: User, self,
score: int, *,
position: int = None): client: "pyrogram.client.ext.BaseClient",
user: User,
score: int,
position: int = None
):
super().__init__(client) super().__init__(client)
self.user = user self.user = user

14
pyrogram/client/types/bots/game_high_scores.py
View File

@ -35,11 +35,15 @@ class GameHighScores(PyrogramType):
Game scores. Game scores.
""" """
def __init__(self, __slots__ = ["total_count", "game_high_scores"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
total_count: int, self,
game_high_scores: List[GameHighScore]): *,
client: "pyrogram.client.ext.BaseClient",
total_count: int,
game_high_scores: List[GameHighScore]
):
super().__init__(client) super().__init__(client)
self.total_count = total_count self.total_count = total_count

20
pyrogram/client/types/bots/inline_keyboard_button.py
View File

@ -54,13 +54,19 @@ class InlineKeyboardButton(PyrogramType):
# TODO: Add callback_game and pay fields # TODO: Add callback_game and pay fields
def __init__(self, __slots__ = [
text: str, "text", "url", "callback_data", "switch_inline_query", "switch_inline_query_current_chat", "callback_game"
callback_data: bytes = None, ]
url: str = None,
switch_inline_query: str = None, def __init__(
switch_inline_query_current_chat: str = None, self,
callback_game: CallbackGame = None): text: str,
callback_data: bytes = None,
url: str = None,
switch_inline_query: str = None,
switch_inline_query_current_chat: str = None,
callback_game: CallbackGame = None
):
super().__init__(None) super().__init__(None)
self.text = str(text) self.text = str(text)

8
pyrogram/client/types/bots/inline_keyboard_markup.py
View File

@ -31,8 +31,12 @@ class InlineKeyboardMarkup(PyrogramType):
List of button rows, each represented by a List of InlineKeyboardButton objects. List of button rows, each represented by a List of InlineKeyboardButton objects.
""" """
def __init__(self, __slots__ = ["inline_keyboard"]
inline_keyboard: List[List[InlineKeyboardButton]]):
def __init__(
self,
inline_keyboard: List[List[InlineKeyboardButton]]
):
super().__init__(None) super().__init__(None)
self.inline_keyboard = inline_keyboard self.inline_keyboard = inline_keyboard

12
pyrogram/client/types/bots/keyboard_button.py
View File

@ -40,10 +40,14 @@ class KeyboardButton(PyrogramType):
Available in private chats only. Available in private chats only.
""" """
def __init__(self, __slots__ = ["text", "request_contact", "request_location"]
text: str,
request_contact: bool = None, def __init__(
request_location: bool = None): self,
text: str,
request_contact: bool = None,
request_location: bool = None
):
super().__init__(None) super().__init__(None)
self.text = str(text) self.text = str(text)

14
pyrogram/client/types/bots/reply_keyboard_markup.py
View File

@ -49,11 +49,15 @@ class ReplyKeyboardMarkup(PyrogramType):
select the new language. Other users in the group don't see the keyboard. select the new language. Other users in the group don't see the keyboard.
""" """
def __init__(self, __slots__ = ["keyboard", "resize_keyboard", "one_time_keyboard", "selective"]
keyboard: List[List[Union[KeyboardButton, str]]],
resize_keyboard: bool = None, def __init__(
one_time_keyboard: bool = None, self,
selective: bool = None): keyboard: List[List[Union[KeyboardButton, str]]],
resize_keyboard: bool = None,
one_time_keyboard: bool = None,
selective: bool = None
):
super().__init__(None) super().__init__(None)
self.keyboard = keyboard self.keyboard = keyboard

8
pyrogram/client/types/bots/reply_keyboard_remove.py
View File

@ -35,8 +35,12 @@ class ReplyKeyboardRemove(PyrogramType):
keyboard for that user, while still showing the keyboard with poll options to users who haven't voted yet. keyboard for that user, while still showing the keyboard with poll options to users who haven't voted yet.
""" """
def __init__(self, __slots__ = ["selective"]
selective: bool = None):
def __init__(
self,
selective: bool = None
):
super().__init__(None) super().__init__(None)
self.selective = selective self.selective = selective

12
pyrogram/client/types/input_media/input_media.py
View File

@ -18,10 +18,14 @@
class InputMedia: class InputMedia:
def __init__(self, __slots__ = ["media", "caption", "parse_mode"]
media: str,
caption: str, def __init__(
parse_mode: str): self,
media: str,
caption: str,
parse_mode: str
):
self.media = media self.media = media
self.caption = caption self.caption = caption
self.parse_mode = parse_mode self.parse_mode = parse_mode

20
pyrogram/client/types/input_media/input_media_animation.py
View File

@ -52,14 +52,18 @@ class InputMediaAnimation(InputMedia):
Animation duration. Animation duration.
""" """
def __init__(self, __slots__ = ["thumb", "width", "height", "duration"]
media: str,
thumb: str = None, def __init__(
caption: str = "", self,
parse_mode: str = "", media: str,
width: int = 0, thumb: str = None,
height: int = 0, caption: str = "",
duration: int = 0): parse_mode: str = "",
width: int = 0,
height: int = 0,
duration: int = 0
):
super().__init__(media, caption, parse_mode) super().__init__(media, caption, parse_mode)
self.thumb = thumb self.thumb = thumb

20
pyrogram/client/types/input_media/input_media_audio.py
View File

@ -53,14 +53,18 @@ class InputMediaAudio(InputMedia):
Title of the audio Title of the audio
""" """
def __init__(self, __slots__ = ["thumb", "duration", "performer", "title"]
media: str,
thumb: str = None, def __init__(
caption: str = "", self,
parse_mode: str = "", media: str,
duration: int = 0, thumb: str = None,
performer: int = "", caption: str = "",
title: str = ""): parse_mode: str = "",
duration: int = 0,
performer: int = "",
title: str = ""
):
super().__init__(media, caption, parse_mode) super().__init__(media, caption, parse_mode)
self.thumb = thumb self.thumb = thumb

14
pyrogram/client/types/input_media/input_media_document.py
View File

@ -43,11 +43,15 @@ class InputMediaDocument(InputMedia):
Defaults to Markdown. Defaults to Markdown.
""" """
def __init__(self, __slots__ = ["thumb"]
media: str,
thumb: str = None, def __init__(
caption: str = "", self,
parse_mode: str = ""): media: str,
thumb: str = None,
caption: str = "",
parse_mode: str = ""
):
super().__init__(media, caption, parse_mode) super().__init__(media, caption, parse_mode)
self.thumb = thumb self.thumb = thumb

12
pyrogram/client/types/input_media/input_media_photo.py
View File

@ -39,8 +39,12 @@ class InputMediaPhoto(InputMedia):
Defaults to Markdown. Defaults to Markdown.
""" """
def __init__(self, __slots__ = []
media: str,
caption: str = "", def __init__(
parse_mode: str = ""): self,
media: str,
caption: str = "",
parse_mode: str = ""
):
super().__init__(media, caption, parse_mode) super().__init__(media, caption, parse_mode)

22
pyrogram/client/types/input_media/input_media_video.py
View File

@ -57,15 +57,19 @@ class InputMediaVideo(InputMedia):
Pass True, if the uploaded video is suitable for streaming. Pass True, if the uploaded video is suitable for streaming.
""" """
def __init__(self, __slots__ = ["thumb", "width", "height", "duration", "supports_streaming"]
media: str,
thumb: str = None, def __init__(
caption: str = "", self,
parse_mode: str = "", media: str,
width: int = 0, thumb: str = None,
height: int = 0, caption: str = "",
duration: int = 0, parse_mode: str = "",
supports_streaming: bool = True): width: int = 0,
height: int = 0,
duration: int = 0,
supports_streaming: bool = True
):
super().__init__(media, caption, parse_mode) super().__init__(media, caption, parse_mode)
self.thumb = thumb self.thumb = thumb

12
pyrogram/client/types/input_media/input_phone_contact.py
View File

@ -35,10 +35,14 @@ class InputPhoneContact:
Contact's last name Contact's last name
""" """
def __init__(self, __slots__ = []
phone: str,
first_name: str, def __init__(
last_name: str = ""): self,
phone: str,
first_name: str,
last_name: str = ""
):
pass pass
def __new__(cls, def __new__(cls,

28
pyrogram/client/types/messages_and_media/animation.py
View File

@ -57,18 +57,22 @@ class Animation(PyrogramType):
Date the animation was sent in Unix time. Date the animation was sent in Unix time.
""" """
def __init__(self, __slots__ = ["file_id", "thumb", "file_name", "mime_type", "file_size", "date", "width", "height", "duration"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
file_id: str, self,
width: int, *,
height: int, client: "pyrogram.client.ext.BaseClient",
duration: int, file_id: str,
thumb: PhotoSize = None, width: int,
file_name: str = None, height: int,
mime_type: str = None, duration: int,
file_size: int = None, thumb: PhotoSize = None,
date: int = None): file_name: str = None,
mime_type: str = None,
file_size: int = None,
date: int = None
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

28
pyrogram/client/types/messages_and_media/audio.py
View File

@ -57,18 +57,22 @@ class Audio(PyrogramType):
Title of the audio as defined by sender or by audio tags. Title of the audio as defined by sender or by audio tags.
""" """
def __init__(self, __slots__ = ["file_id", "thumb", "file_name", "mime_type", "file_size", "date", "duration", "performer", "title"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
file_id: str, self,
duration: int, *,
thumb: PhotoSize = None, client: "pyrogram.client.ext.BaseClient",
file_name: str = None, file_id: str,
mime_type: str = None, duration: int,
file_size: int = None, thumb: PhotoSize = None,
date: int = None, file_name: str = None,
performer: str = None, mime_type: str = None,
title: str = None): file_size: int = None,
date: int = None,
performer: str = None,
title: str = None
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

20
pyrogram/client/types/messages_and_media/contact.py
View File

@ -42,14 +42,18 @@ class Contact(PyrogramType):
Additional data about the contact in the form of a vCard. Additional data about the contact in the form of a vCard.
""" """
def __init__(self, __slots__ = ["phone_number", "first_name", "last_name", "user_id", "vcard"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
phone_number: str, self,
first_name: str, *,
last_name: str = None, client: "pyrogram.client.ext.BaseClient",
user_id: int = None, phone_number: str,
vcard: str = None): first_name: str,
last_name: str = None,
user_id: int = None,
vcard: str = None
):
super().__init__(client) super().__init__(client)
self.phone_number = phone_number self.phone_number = phone_number

22
pyrogram/client/types/messages_and_media/document.py
View File

@ -48,15 +48,19 @@ class Document(PyrogramType):
Date the document was sent in Unix time. Date the document was sent in Unix time.
""" """
def __init__(self, __slots__ = ["file_id", "thumb", "file_name", "mime_type", "file_size", "date"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
file_id: str, self,
thumb: PhotoSize = None, *,
file_name: str = None, client: "pyrogram.client.ext.BaseClient",
mime_type: str = None, file_id: str,
file_size: int = None, thumb: PhotoSize = None,
date: int = None): file_name: str = None,
mime_type: str = None,
file_size: int = None,
date: int = None
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

22
pyrogram/client/types/messages_and_media/game.py
View File

@ -48,15 +48,19 @@ class Game(PyrogramType):
Upload via BotFather. Upload via BotFather.
""" """
def __init__(self, __slots__ = ["id", "title", "short_name", "description", "photo", "animation"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
id: int, self,
title: str, *,
short_name: str, client: "pyrogram.client.ext.BaseClient",
description: str, id: int,
photo: Photo, title: str,
animation: Animation = None): short_name: str,
description: str,
photo: Photo,
animation: Animation = None
):
super().__init__(client) super().__init__(client)
self.id = id self.id = id

14
pyrogram/client/types/messages_and_media/location.py
View File

@ -33,11 +33,15 @@ class Location(PyrogramType):
Latitude as defined by sender. Latitude as defined by sender.
""" """
def __init__(self, __slots__ = ["longitude", "latitude"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
longitude: float, self,
latitude: float): *,
client: "pyrogram.client.ext.BaseClient",
longitude: float,
latitude: float
):
super().__init__(client) super().__init__(client)
self.longitude = longitude self.longitude = longitude

133
pyrogram/client/types/messages_and_media/message.py
View File

@ -234,65 +234,80 @@ class Message(PyrogramType, Update):
# TODO: Add game missing field. Also invoice, successful_payment, connected_website # TODO: Add game missing field. Also invoice, successful_payment, connected_website
def __init__(self, __slots__ = [
*, "message_id", "date", "chat", "from_user", "forward_from", "forward_from_chat", "forward_from_message_id",
client: "pyrogram.client.ext.BaseClient", "forward_signature", "forward_date", "reply_to_message", "mentioned", "empty", "service", "media", "edit_date",
message_id: int, "media_group_id", "author_signature", "text", "entities", "caption_entities", "audio", "document", "photo",
date: int = None, "sticker", "animation", "game", "video", "voice", "video_note", "caption", "contact", "location", "venue",
chat: Chat = None, "web_page", "poll", "new_chat_members", "left_chat_member", "new_chat_title", "new_chat_photo",
from_user: User = None, "delete_chat_photo", "group_chat_created", "supergroup_chat_created", "channel_chat_created",
forward_from: User = None, "migrate_to_chat_id", "migrate_from_chat_id", "pinned_message", "game_high_score", "views", "via_bot",
forward_from_chat: Chat = None, "outgoing", "matches", "command", "reply_markup"
forward_from_message_id: int = None, ]
forward_signature: str = None,
forward_date: int = None, def __init__(
reply_to_message: "Message" = None, self,
mentioned: bool = None, *,
empty: bool = None, client: "pyrogram.client.ext.BaseClient",
service: bool = None, message_id: int,
media: bool = None, date: int = None,
edit_date: int = None, chat: Chat = None,
media_group_id: str = None, from_user: User = None,
author_signature: str = None, forward_from: User = None,
text: str = None, forward_from_chat: Chat = None,
entities: List["pyrogram.MessageEntity"] = None, forward_from_message_id: int = None,
caption_entities: List["pyrogram.MessageEntity"] = None, forward_signature: str = None,
audio: "pyrogram.Audio" = None, forward_date: int = None,
document: "pyrogram.Document" = None, reply_to_message: "Message" = None,
photo: "pyrogram.Photo" = None, mentioned: bool = None,
sticker: "pyrogram.Sticker" = None, empty: bool = None,
animation: "pyrogram.Animation" = None, service: bool = None,
game: "pyrogram.Game" = None, media: bool = None,
video: "pyrogram.Video" = None, edit_date: int = None,
voice: "pyrogram.Voice" = None, media_group_id: str = None,
video_note: "pyrogram.VideoNote" = None, author_signature: str = None,
caption: str = None, text: str = None,
contact: "pyrogram.Contact" = None, entities: List["pyrogram.MessageEntity"] = None,
location: "pyrogram.Location" = None, caption_entities: List["pyrogram.MessageEntity"] = None,
venue: "pyrogram.Venue" = None, audio: "pyrogram.Audio" = None,
web_page: bool = None, document: "pyrogram.Document" = None,
poll: "pyrogram.Poll" = None, photo: "pyrogram.Photo" = None,
new_chat_members: List[User] = None, sticker: "pyrogram.Sticker" = None,
left_chat_member: User = None, animation: "pyrogram.Animation" = None,
new_chat_title: str = None, game: "pyrogram.Game" = None,
new_chat_photo: "pyrogram.Photo" = None, video: "pyrogram.Video" = None,
delete_chat_photo: bool = None, voice: "pyrogram.Voice" = None,
group_chat_created: bool = None, video_note: "pyrogram.VideoNote" = None,
supergroup_chat_created: bool = None, caption: str = None,
channel_chat_created: bool = None, contact: "pyrogram.Contact" = None,
migrate_to_chat_id: int = None, location: "pyrogram.Location" = None,
migrate_from_chat_id: int = None, venue: "pyrogram.Venue" = None,
pinned_message: "Message" = None, web_page: bool = None,
game_high_score: int = None, poll: "pyrogram.Poll" = None,
views: int = None, new_chat_members: List[User] = None,
via_bot: User = None, left_chat_member: User = None,
outgoing: bool = None, new_chat_title: str = None,
matches: List[Match] = None, new_chat_photo: "pyrogram.Photo" = None,
command: List[str] = None, delete_chat_photo: bool = None,
reply_markup: Union["pyrogram.InlineKeyboardMarkup", group_chat_created: bool = None,
"pyrogram.ReplyKeyboardMarkup", supergroup_chat_created: bool = None,
"pyrogram.ReplyKeyboardRemove", channel_chat_created: bool = None,
"pyrogram.ForceReply"] = None): migrate_to_chat_id: int = None,
migrate_from_chat_id: int = None,
pinned_message: "Message" = None,
game_high_score: int = None,
views: int = None,
via_bot: User = None,
outgoing: bool = None,
matches: List[Match] = None,
command: List[str] = None,
reply_markup: Union[
"pyrogram.InlineKeyboardMarkup",
"pyrogram.ReplyKeyboardMarkup",
"pyrogram.ReplyKeyboardRemove",
"pyrogram.ForceReply"
] = None
):
super().__init__(client) super().__init__(client)
self.message_id = message_id self.message_id = message_id

20
pyrogram/client/types/messages_and_media/message_entity.py
View File

@ -47,6 +47,8 @@ class MessageEntity(PyrogramType):
For "text_mention" only, the mentioned user. For "text_mention" only, the mentioned user.
""" """
__slots__ = ["type", "offset", "length", "url", "user"]
ENTITIES = { ENTITIES = {
types.MessageEntityMention.ID: "mention", types.MessageEntityMention.ID: "mention",
types.MessageEntityHashtag.ID: "hashtag", types.MessageEntityHashtag.ID: "hashtag",
@ -63,14 +65,16 @@ class MessageEntity(PyrogramType):
types.MessageEntityPhone.ID: "phone_number" types.MessageEntityPhone.ID: "phone_number"
} }
def __init__(self, def __init__(
*, self,
client: "pyrogram.client.ext.BaseClient", *,
type: str, client: "pyrogram.client.ext.BaseClient",
offset: int, type: str,
length: int, offset: int,
url: str = None, length: int,
user: User = None): url: str = None,
user: User = None
):
super().__init__(client) super().__init__(client)
self.type = type self.type = type

14
pyrogram/client/types/messages_and_media/messages.py
View File

@ -37,11 +37,15 @@ class Messages(PyrogramType, Update):
Requested messages. Requested messages.
""" """
def __init__(self, __slots__ = ["total_count", "messages"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
total_count: int, self,
messages: List[Message]): *,
client: "pyrogram.client.ext.BaseClient",
total_count: int,
messages: List[Message]
):
super().__init__(client) super().__init__(client)
self.total_count = total_count self.total_count = total_count

16
pyrogram/client/types/messages_and_media/photo.py
View File

@ -41,12 +41,16 @@ class Photo(PyrogramType):
Available sizes of this photo. Available sizes of this photo.
""" """
def __init__(self, __slots__ = ["id", "date", "sizes"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
id: str, self,
date: int, *,
sizes: List[PhotoSize]): client: "pyrogram.client.ext.BaseClient",
id: str,
date: int,
sizes: List[PhotoSize]
):
super().__init__(client) super().__init__(client)
self.id = id self.id = id

18
pyrogram/client/types/messages_and_media/photo_size.py
View File

@ -42,13 +42,17 @@ class PhotoSize(PyrogramType):
File size. File size.
""" """
def __init__(self, __slots__ = ["file_id", "width", "height", "file_size"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
file_id: str, self,
width: int, *,
height: int, client: "pyrogram.client.ext.BaseClient",
file_size: int): file_id: str,
width: int,
height: int,
file_size: int
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

22
pyrogram/client/types/messages_and_media/poll.py
View File

@ -47,15 +47,19 @@ class Poll(PyrogramType):
The index of your chosen option (in case you voted already), None otherwise. The index of your chosen option (in case you voted already), None otherwise.
""" """
def __init__(self, __slots__ = ["id", "closed", "question", "options", "total_voters", "option_chosen"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
id: int, self,
closed: bool, *,
question: str, client: "pyrogram.client.ext.BaseClient",
options: List[PollOption], id: int,
total_voters: int, closed: bool,
option_chosen: int = None): question: str,
options: List[PollOption],
total_voters: int,
option_chosen: int = None
):
super().__init__(client) super().__init__(client)
self.id = id self.id = id

16
pyrogram/client/types/messages_and_media/poll_option.py
View File

@ -34,12 +34,16 @@ class PollOption(PyrogramType):
Unique data that identifies this option among all the other options in a poll. Unique data that identifies this option among all the other options in a poll.
""" """
def __init__(self, __slots__ = ["text", "voters", "data"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
text: str, self,
voters: int, *,
data: bytes): client: "pyrogram.client.ext.BaseClient",
text: str,
voters: int,
data: bytes
):
super().__init__(client) super().__init__(client)
self.text = text self.text = text

32
pyrogram/client/types/messages_and_media/sticker.py
View File

@ -64,19 +64,25 @@ class Sticker(PyrogramType):
# TODO: Add mask position # TODO: Add mask position
def __init__(self, __slots__ = [
*, "file_id", "thumb", "file_name", "mime_type", "file_size", "date", "width", "height", "emoji", "set_name"
client: "pyrogram.client.ext.BaseClient", ]
file_id: str,
width: int, def __init__(
height: int, self,
thumb: PhotoSize = None, *,
file_name: str = None, client: "pyrogram.client.ext.BaseClient",
mime_type: str = None, file_id: str,
file_size: int = None, width: int,
date: int = None, height: int,
emoji: str = None, thumb: PhotoSize = None,
set_name: str = None): file_name: str = None,
mime_type: str = None,
file_size: int = None,
date: int = None,
emoji: str = None,
set_name: str = None
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

14
pyrogram/client/types/messages_and_media/user_profile_photos.py
View File

@ -34,11 +34,15 @@ class UserProfilePhotos(PyrogramType):
Requested profile pictures. Requested profile pictures.
""" """
def __init__(self, __slots__ = ["total_count", "photos"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
total_count: int, self,
photos: List[Photo]): *,
client: "pyrogram.client.ext.BaseClient",
total_count: int,
photos: List[Photo]
):
super().__init__(client) super().__init__(client)
self.total_count = total_count self.total_count = total_count

20
pyrogram/client/types/messages_and_media/venue.py
View File

@ -44,14 +44,18 @@ class Venue(PyrogramType):
""" """
def __init__(self, __slots__ = ["location", "title", "address", "foursquare_id", "foursquare_type"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
location: Location, self,
title: str, *,
address: str, client: "pyrogram.client.ext.BaseClient",
foursquare_id: str = None, location: Location,
foursquare_type: str = None): title: str,
address: str,
foursquare_id: str = None,
foursquare_type: str = None
):
super().__init__(client) super().__init__(client)
self.location = location self.location = location

28
pyrogram/client/types/messages_and_media/video.py
View File

@ -57,18 +57,22 @@ class Video(PyrogramType):
Date the video was sent in Unix time. Date the video was sent in Unix time.
""" """
def __init__(self, __slots__ = ["file_id", "thumb", "file_name", "mime_type", "file_size", "date", "width", "height", "duration"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
file_id: str, self,
width: int, *,
height: int, client: "pyrogram.client.ext.BaseClient",
duration: int, file_id: str,
thumb: PhotoSize = None, width: int,
file_name: str = None, height: int,
mime_type: str = None, duration: int,
file_size: int = None, thumb: PhotoSize = None,
date: int = None): file_name: str = None,
mime_type: str = None,
file_size: int = None,
date: int = None
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

24
pyrogram/client/types/messages_and_media/video_note.py
View File

@ -51,16 +51,20 @@ class VideoNote(PyrogramType):
Date the video note was sent in Unix time. Date the video note was sent in Unix time.
""" """
def __init__(self, __slots__ = ["file_id", "thumb", "mime_type", "file_size", "date", "length", "duration"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
file_id: str, self,
length: int, *,
duration: int, client: "pyrogram.client.ext.BaseClient",
thumb: PhotoSize = None, file_id: str,
mime_type: str = None, length: int,
file_size: int = None, duration: int,
date: int = None): thumb: PhotoSize = None,
mime_type: str = None,
file_size: int = None,
date: int = None
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

22
pyrogram/client/types/messages_and_media/voice.py
View File

@ -47,15 +47,19 @@ class Voice(PyrogramType):
Date the voice was sent in Unix time. Date the voice was sent in Unix time.
""" """
def __init__(self, __slots__ = ["file_id", "duration", "waveform", "mime_type", "file_size", "date"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
file_id: str, self,
duration: int, *,
waveform: bytes = None, client: "pyrogram.client.ext.BaseClient",
mime_type: str = None, file_id: str,
file_size: int = None, duration: int,
date: int = None): waveform: bytes = None,
mime_type: str = None,
file_size: int = None,
date: int = None
):
super().__init__(client) super().__init__(client)
self.file_id = file_id self.file_id = file_id

44
pyrogram/client/types/user_and_chats/chat.py
View File

@ -80,24 +80,32 @@ class Chat(PyrogramType):
Information about the chat default permissions. Information about the chat default permissions.
""" """
def __init__(self, __slots__ = [
*, "id", "type", "title", "username", "first_name", "last_name", "photo", "description", "invite_link",
client: "pyrogram.client.ext.BaseClient", "pinned_message", "sticker_set_name", "can_set_sticker_set", "members_count", "restriction_reason",
id: int, "permissions"
type: str, ]
title: str = None,
username: str = None, def __init__(
first_name: str = None, self,
last_name: str = None, *,
photo: ChatPhoto = None, client: "pyrogram.client.ext.BaseClient",
description: str = None, id: int,
invite_link: str = None, type: str,
pinned_message=None, title: str = None,
sticker_set_name: str = None, username: str = None,
can_set_sticker_set: bool = None, first_name: str = None,
members_count: int = None, last_name: str = None,
restriction_reason: str = None, photo: ChatPhoto = None,
permissions: "pyrogram.ChatPermissions" = None): description: str = None,
invite_link: str = None,
pinned_message=None,
sticker_set_name: str = None,
can_set_sticker_set: bool = None,
members_count: int = None,
restriction_reason: str = None,
permissions: "pyrogram.ChatPermissions" = None
):
super().__init__(client) super().__init__(client)
self.id = id self.id = id

24
pyrogram/client/types/user_and_chats/chat_member.py
View File

@ -51,16 +51,20 @@ class ChatMember(PyrogramType):
Information about the member permissions. Information about the member permissions.
""" """
def __init__(self, __slots__ = ["user", "status", "date", "invited_by", "promoted_by", "restricted_by", "permissions"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
user: "pyrogram.User", self,
status: str, *,
date: int = None, client: "pyrogram.client.ext.BaseClient",
invited_by: "pyrogram.User" = None, user: "pyrogram.User",
promoted_by: "pyrogram.User" = None, status: str,
restricted_by: "pyrogram.User" = None, date: int = None,
permissions: "pyrogram.ChatPermissions" = None): invited_by: "pyrogram.User" = None,
promoted_by: "pyrogram.User" = None,
restricted_by: "pyrogram.User" = None,
permissions: "pyrogram.ChatPermissions" = None
):
super().__init__(client) super().__init__(client)
self.user = user self.user = user

15
pyrogram/client/types/user_and_chats/chat_members.py
View File

@ -21,7 +21,6 @@ from typing import List
import pyrogram import pyrogram
from pyrogram.api import types from pyrogram.api import types
from .chat_member import ChatMember from .chat_member import ChatMember
from .user import User
from ..pyrogram_type import PyrogramType from ..pyrogram_type import PyrogramType
@ -36,11 +35,15 @@ class ChatMembers(PyrogramType):
Requested chat members. Requested chat members.
""" """
def __init__(self, __slots__ = ["total_count", "chat_members"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
total_count: int, self,
chat_members: List[ChatMember]): *,
client: "pyrogram.client.ext.BaseClient",
total_count: int,
chat_members: List[ChatMember]
):
super().__init__(client) super().__init__(client)
self.total_count = total_count self.total_count = total_count

7
pyrogram/client/types/user_and_chats/chat_permissions.py
View File

@ -94,6 +94,13 @@ class ChatPermissions(PyrogramType):
True, if polls can be sent, implies can_send_media_messages. True, if polls can be sent, implies can_send_media_messages.
""" """
__slots__ = [
"until_date", "can_be_edited", "can_change_info", "can_post_messages", "can_edit_messages",
"can_delete_messages", "can_restrict_members", "can_invite_users", "can_pin_messages", "can_promote_members",
"can_send_messages", "can_send_media_messages", "can_send_other_messages", "can_add_web_page_previews",
"can_send_polls"
]
def __init__( def __init__(
self, self,
*, *,

14
pyrogram/client/types/user_and_chats/chat_photo.py
View File

@ -35,11 +35,15 @@ class ChatPhoto(PyrogramType):
Unique file identifier of big (640x640) chat photo. This file_id can be used only for photo download. Unique file identifier of big (640x640) chat photo. This file_id can be used only for photo download.
""" """
def __init__(self, __slots__ = ["small_file_id", "big_file_id"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
small_file_id: str, self,
big_file_id: str): *,
client: "pyrogram.client.ext.BaseClient",
small_file_id: str,
big_file_id: str
):
super().__init__(client) super().__init__(client)
self.small_file_id = small_file_id self.small_file_id = small_file_id

20
pyrogram/client/types/user_and_chats/chat_preview.py
View File

@ -45,14 +45,18 @@ class ChatPreview(PyrogramType):
Preview of some of the chat members. Preview of some of the chat members.
""" """
def __init__(self, __slots__ = ["title", "photo", "type", "members_count", "members"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
title: str, self,
photo: ChatPhoto, *,
type: str, client: "pyrogram.client.ext.BaseClient",
members_count: int, title: str,
members: List[User] = None): photo: ChatPhoto,
type: str,
members_count: int,
members: List[User] = None
):
super().__init__(client) super().__init__(client)
self.title = title self.title = title

22
pyrogram/client/types/user_and_chats/dialog.py
View File

@ -46,15 +46,19 @@ class Dialog(PyrogramType):
True, if the dialog is pinned. True, if the dialog is pinned.
""" """
def __init__(self, __slots__ = ["chat", "top_message", "unread_messages_count", "unread_mentions_count", "unread_mark", "is_pinned"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
chat: Chat, self,
top_message: "pyrogram.Message", *,
unread_messages_count: int, client: "pyrogram.client.ext.BaseClient",
unread_mentions_count: int, chat: Chat,
unread_mark: bool, top_message: "pyrogram.Message",
is_pinned: bool): unread_messages_count: int,
unread_mentions_count: int,
unread_mark: bool,
is_pinned: bool
):
super().__init__(client) super().__init__(client)
self.chat = chat self.chat = chat

14
pyrogram/client/types/user_and_chats/dialogs.py
View File

@ -36,11 +36,15 @@ class Dialogs(PyrogramType):
Requested dialogs. Requested dialogs.
""" """
def __init__(self, __slots__ = ["total_count", "dialogs"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
total_count: int, self,
dialogs: List[Dialog]): *,
client: "pyrogram.client.ext.BaseClient",
total_count: int,
dialogs: List[Dialog]
):
super().__init__(client) super().__init__(client)
self.total_count = total_count self.total_count = total_count

41
pyrogram/client/types/user_and_chats/user.py
View File

@ -70,23 +70,30 @@ class User(PyrogramType):
The reason why this bot might be unavailable to some users. The reason why this bot might be unavailable to some users.
""" """
def __init__(self, __slots__ = [
*, "id", "is_self", "is_contact", "is_mutual_contact", "is_deleted", "is_bot", "first_name", "last_name", "status",
client: "pyrogram.client.ext.BaseClient", "username", "language_code", "phone_number", "photo", "restriction_reason"
id: int, ]
is_self: bool,
is_contact: bool, def __init__(
is_mutual_contact: bool, self,
is_deleted: bool, *,
is_bot: bool, client: "pyrogram.client.ext.BaseClient",
first_name: str, id: int,
last_name: str = None, is_self: bool,
status: UserStatus = None, is_contact: bool,
username: str = None, is_mutual_contact: bool,
language_code: str = None, is_deleted: bool,
phone_number: str = None, is_bot: bool,
photo: ChatPhoto = None, first_name: str,
restriction_reason: str = None): last_name: str = None,
status: UserStatus = None,
username: str = None,
language_code: str = None,
phone_number: str = None,
photo: ChatPhoto = None,
restriction_reason: str = None
):
super().__init__(client) super().__init__(client)
self.id = id self.id = id

26
pyrogram/client/types/user_and_chats/user_status.py
View File

@ -65,17 +65,21 @@ class UserStatus(PyrogramType, Update):
always shown to blocked users), None otherwise. always shown to blocked users), None otherwise.
""" """
def __init__(self, __slots__ = ["user_id", "online", "offline", "date", "recently", "within_week", "within_month", "long_time_ago"]
*,
client: "pyrogram.client.ext.BaseClient", def __init__(
user_id: int, self,
online: bool = None, *,
offline: bool = None, client: "pyrogram.client.ext.BaseClient",
date: int = None, user_id: int,
recently: bool = None, online: bool = None,
within_week: bool = None, offline: bool = None,
within_month: bool = None, date: int = None,
long_time_ago: bool = None): recently: bool = None,
within_week: bool = None,
within_month: bool = None,
long_time_ago: bool = None
):
super().__init__(client) super().__init__(client)
self.user_id = user_id self.user_id = user_id