diff --git a/pyrogram/__init__.py b/pyrogram/__init__.py index 09545a6c..767a4cab 100644 --- a/pyrogram/__init__.py +++ b/pyrogram/__init__.py @@ -29,3 +29,4 @@ from .api.errors import Error from .client import ChatAction from .client import Client from .client import ParseMode +from .client.input_media import InputMedia diff --git a/pyrogram/client/input_media.py b/pyrogram/client/input_media.py new file mode 100644 index 00000000..e61ccc15 --- /dev/null +++ b/pyrogram/client/input_media.py @@ -0,0 +1,24 @@ +class InputMedia: + class Photo: + def __init__(self, + media: str, + caption: str = "", + parse_mode: str = ""): + self.media = media + self.caption = caption + self.parse_mode = parse_mode + + class Video: + def __init__(self, + media: str, + caption: str = "", + parse_mode: str = "", + width: int = 0, + height: int = 0, + duration: int = 0): + self.media = media + self.caption = caption + self.parse_mode = parse_mode + self.width = width + self.height = height + self.duration = duration