Fix wrong imports in examples

delivrance · delivrance · commit c139d78b34f9 · 2020-10-31T19:29:39.000+01:00
diff --git a/pyrogram/methods/bots/answer_inline_query.py b/pyrogram/methods/bots/answer_inline_query.py
@@ -84,7 +84,7 @@ async def answer_inline_query(
         Example:
             .. code-block:: python
 
-                from pyrogram import InlineQueryResultArticle, InputTextMessageContent
+                from pyrogram.types import InlineQueryResultArticle, InputTextMessageContent
 
                 app.answer_inline_query(
                     inline_query_id,
diff --git a/pyrogram/methods/chats/restrict_chat_member.py b/pyrogram/methods/chats/restrict_chat_member.py
@@ -60,7 +60,7 @@ async def restrict_chat_member(
 
                 from time import time
 
-                from pyrogram import ChatPermissions
+                from pyrogram.types import ChatPermissions
 
                 # Completely restrict chat member (mute) forever
                 app.restrict_chat_member(chat_id, user_id, ChatPermissions())
diff --git a/pyrogram/methods/chats/set_chat_permissions.py b/pyrogram/methods/chats/set_chat_permissions.py
@@ -47,7 +47,7 @@ async def set_chat_permissions(
         Example:
             .. code-block:: python
 
-                from pyrogram import ChatPermissions
+                from pyrogram.types import ChatPermissions
 
                 # Completely restrict chat
                 app.set_chat_permissions(chat_id, ChatPermissions())
diff --git a/pyrogram/methods/contacts/add_contacts.py b/pyrogram/methods/contacts/add_contacts.py
@@ -40,7 +40,7 @@ async def add_contacts(
         Example:
             .. code-block:: python
 
-                from pyrogram import InputPhoneContact
+                from pyrogram.types import InputPhoneContact
 
                 app.add_contacts([
                     InputPhoneContact("39123456789", "Foo"),
diff --git a/pyrogram/methods/messages/edit_inline_media.py b/pyrogram/methods/messages/edit_inline_media.py
@@ -54,7 +54,7 @@ async def edit_inline_media(
         Example:
             .. code-block:: python
 
-                from pyrogram import InputMediaPhoto, InputMediaVideo, InputMediaAudio
+                from pyrogram.types import InputMediaPhoto, InputMediaVideo, InputMediaAudio
 
                 # Bots only
 
diff --git a/pyrogram/methods/messages/edit_inline_reply_markup.py b/pyrogram/methods/messages/edit_inline_reply_markup.py
@@ -44,7 +44,7 @@ async def edit_inline_reply_markup(
         Example:
             .. code-block:: python
 
-                from pyrogram import InlineKeyboardMarkup, InlineKeyboardButton
+                from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
 
                 # Bots only
                 app.edit_inline_reply_markup(
diff --git a/pyrogram/methods/messages/edit_message_media.py b/pyrogram/methods/messages/edit_message_media.py
@@ -65,7 +65,7 @@ async def edit_message_media(
         Example:
             .. code-block:: python
 
-                from pyrogram import InputMediaPhoto, InputMediaVideo, InputMediaAudio
+                from pyrogram.types import InputMediaPhoto, InputMediaVideo, InputMediaAudio
 
                 # Replace the current media with a local photo
                 app.edit_message_media(chat_id, message_id, InputMediaPhoto("new_photo.jpg"))
diff --git a/pyrogram/methods/messages/edit_message_reply_markup.py b/pyrogram/methods/messages/edit_message_reply_markup.py
@@ -50,7 +50,7 @@ async def edit_message_reply_markup(
         Example:
             .. code-block:: python
 
-                from pyrogram import InlineKeyboardMarkup, InlineKeyboardButton
+                from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
 
                 # Bots only
                 app.edit_message_reply_markup(
diff --git a/pyrogram/methods/messages/send_message.py b/pyrogram/methods/messages/send_message.py
@@ -98,7 +98,7 @@ async def send_message(
                 # For bots only, send messages with keyboards attached
                 ##
 
-                from pyrogram import (
+                from pyrogram.types import (
                     ReplyKeyboardMarkup, InlineKeyboardMarkup, InlineKeyboardButton)
 
                 # Send a normal keyboard
diff --git a/pyrogram/methods/utilities/add_handler.py b/pyrogram/methods/utilities/add_handler.py
@@ -44,7 +44,8 @@ def add_handler(self, handler: "Handler", group: int = 0):
             .. code-block:: python
                 :emphasize-lines: 8
 
-                from pyrogram import Client, MessageHandler
+                from pyrogram import Client
+                from pyrogram.handlers import MessageHandler
 
                 def dump(client, message):
                     print(message)
diff --git a/pyrogram/methods/utilities/remove_handler.py b/pyrogram/methods/utilities/remove_handler.py
@@ -39,7 +39,8 @@ def remove_handler(self, handler: "Handler", group: int = 0):
             .. code-block:: python
                 :emphasize-lines: 11
 
-                from pyrogram import Client, MessageHandler
+                from pyrogram import Client
+                from pyrogram.handlers import MessageHandler
 
                 def dump(client, message):
                     print(message)
