mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
Provide proxy connection. Add example for proxy and emoji util
This commit is contained in:
parent
c94c4e8003
commit
6c21270ccb
5 changed files with 94 additions and 23 deletions
|
|
@ -2,6 +2,6 @@ from .utils.versions import Version, Stage
|
|||
from .bot import Bot
|
||||
|
||||
|
||||
VERSION = Version(0, 3, 3, stage=Stage.FINAL, build=0)
|
||||
VERSION = Version(0, 3, 4, stage=Stage.DEV, build=0)
|
||||
|
||||
__version__ = VERSION.version
|
||||
|
|
|
|||
|
|
@ -104,7 +104,7 @@ def _compose_data(params=None, files=None):
|
|||
return data
|
||||
|
||||
|
||||
async def request(session, token, method, data=None, files=None) -> bool or dict:
|
||||
async def request(session, token, method, data=None, files=None, **kwargs) -> bool or dict:
|
||||
"""
|
||||
Make request to API
|
||||
|
||||
|
|
@ -125,7 +125,7 @@ async def request(session, token, method, data=None, files=None) -> bool or dict
|
|||
method, data or {}, files or {}))
|
||||
data = _compose_data(data, files)
|
||||
url = Methods.api_url(token=token, method=method)
|
||||
async with session.post(url, data=data) as response:
|
||||
async with session.post(url, data=data, **kwargs) as response:
|
||||
return await _check_result(method, response)
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -6,6 +6,7 @@ import aiohttp
|
|||
|
||||
from . import api
|
||||
from .. import types
|
||||
from ..utils import json
|
||||
from ..utils.payload import generate_payload, prepare_arg
|
||||
|
||||
InputFile = TypeVar('InputFile', io.BytesIO, io.FileIO, str)
|
||||
|
|
@ -22,7 +23,7 @@ class BaseBot:
|
|||
|
||||
def __init__(self, token: String,
|
||||
loop: Optional[Union[asyncio.BaseEventLoop, asyncio.AbstractEventLoop]] = None,
|
||||
connections_limit: Optional[Integer] = 10):
|
||||
connections_limit: Optional[Integer] = 10, proxy=None, proxy_auth=None):
|
||||
"""
|
||||
Instructions how to get Bot token is found here: https://core.telegram.org/bots#3-how-do-i-create-a-bot
|
||||
|
||||
|
|
@ -32,6 +33,8 @@ class BaseBot:
|
|||
"""
|
||||
|
||||
self.__token = token
|
||||
self.proxy = proxy
|
||||
self.proxy_auth = proxy_auth
|
||||
|
||||
if loop is None:
|
||||
loop = asyncio.get_event_loop()
|
||||
|
|
@ -39,7 +42,7 @@ class BaseBot:
|
|||
self.loop = loop
|
||||
self.session = aiohttp.ClientSession(
|
||||
connector=aiohttp.TCPConnector(limit=connections_limit),
|
||||
loop=self.loop)
|
||||
loop=self.loop, json_serialize=json.dumps)
|
||||
self._temp_sessions = []
|
||||
api.check_token(token)
|
||||
|
||||
|
|
@ -93,7 +96,8 @@ class BaseBot:
|
|||
:return: Union[List, Dict]
|
||||
:raise: :class:`aiogram.exceptions.TelegramApiError`
|
||||
"""
|
||||
return await api.request(self.session, self.__token, method, data, files)
|
||||
return await api.request(self.session, self.__token, method, data, files,
|
||||
proxy=self.proxy, proxy_auth=self.proxy_auth)
|
||||
|
||||
async def download_file(self, file_path: String,
|
||||
destination: Optional[InputFile] = None,
|
||||
|
|
@ -121,7 +125,7 @@ class BaseBot:
|
|||
|
||||
dest = destination if isinstance(destination, io.IOBase) else open(destination, 'wb')
|
||||
try:
|
||||
async with session.get(url, timeout=timeout) as response:
|
||||
async with session.get(url, timeout=timeout, proxy=self.proxy, proxy_auth=self.proxy_auth) as response:
|
||||
while True:
|
||||
chunk = await response.content.read(chunk_size)
|
||||
if not chunk:
|
||||
|
|
|
|||
|
|
@ -1,5 +1,6 @@
|
|||
import datetime
|
||||
|
||||
from aiogram.utils.helper import Item, HelperMode, Helper
|
||||
from .audio import Audio
|
||||
from .base import Deserializable
|
||||
from .chat import Chat
|
||||
|
|
@ -240,7 +241,7 @@ class Message(Deserializable):
|
|||
return await self.chat.pin_message(self.message_id, disable_notification)
|
||||
|
||||
|
||||
class ContentType:
|
||||
class ContentType(Helper):
|
||||
"""
|
||||
List of message content types
|
||||
|
||||
|
|
@ -258,24 +259,25 @@ class ContentType:
|
|||
:key: SUCCESSFUL_PAYMENT
|
||||
:key: UNKNOWN
|
||||
"""
|
||||
mode = HelperMode.lower_case
|
||||
|
||||
TEXT = 'text'
|
||||
AUDIO = 'audio'
|
||||
DOCUMENT = 'document'
|
||||
GAME = 'game'
|
||||
PHOTO = 'photo'
|
||||
STICKER = 'sticker'
|
||||
VIDEO = 'video'
|
||||
VOICE = 'voice'
|
||||
NEW_CHAT_MEMBERS = 'new_chat_members'
|
||||
LEFT_CHAT_MEMBER = 'left_chat_member'
|
||||
INVOICE = 'invoice'
|
||||
SUCCESSFUL_PAYMENT = 'successful_payment'
|
||||
TEXT = Item() # text
|
||||
AUDIO = Item() # audio
|
||||
DOCUMENT = Item() # document
|
||||
GAME = Item() # game
|
||||
PHOTO = Item() # photo
|
||||
STICKER = Item() # sticker
|
||||
VIDEO = Item() # video
|
||||
VOICE = Item() # voice
|
||||
NEW_CHAT_MEMBERS = Item() # new_chat_members
|
||||
LEFT_CHAT_MEMBER = Item() # left_chat_member
|
||||
INVOICE = Item() # invoice
|
||||
SUCCESSFUL_PAYMENT = Item() # successful_payment
|
||||
|
||||
UNKNOWN = 'unknown'
|
||||
|
||||
|
||||
class ParseMode:
|
||||
class ParseMode(Helper):
|
||||
"""
|
||||
Parse modes
|
||||
|
||||
|
|
@ -283,5 +285,7 @@ class ParseMode:
|
|||
:key: HTML
|
||||
"""
|
||||
|
||||
MARKDOWN = 'markdown'
|
||||
HTML = 'html'
|
||||
mode = HelperMode.lowercase
|
||||
|
||||
MARKDOWN = Item()
|
||||
HTML = Item()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue