Установка параметров `aiohttp.ClientSession` по умолчанию - PullRequest
0 голосов
/ 29 сентября 2018

Я перехожу с requests на aiohttp, и ранее я использовал requests.Session.params, чтобы установить параметры запроса по умолчанию для каждого запроса, сделанного в сеансе.

Как правильно добитьсяэквивалент с aiohttp.ClientSession?

Я пытался:

  • Подклассы aiohttp.ClientSession, чтобы перезаписать метод _request.Это поднимает DepcrecationWarning: Inheritance from ClientSession is discouraged, однако.
  • Подклассы aiohttp.ClientResponse перезаписывают метод __init__ и обновляют params там, а затем передают подкласс в aiohttp.ClientSession.Это не сработает, поскольку мне нужно, чтобы пользовательские параметры определялись (и были изменяемыми) на основе сеанса.

РЕДАКТИРОВАТЬ: пример того, как я хотел бы, чтобы мой код выглядел, для пояснения:

from aiohttp import ClientSession

session = ClientSession()
# Do something fancy here, or in the initialization of `ClientSession`, so that
# `default_params` is set as the default parameters for each request
default_params = dict(some_parameter="a value")

async with session.get("http://httpbin.org/get") as resp:
    j = await resp.json()
assert j["args"] == default_params

# Do something fancy here, where we change the default parameters
default_params.update(some_parameter="another value")

async with session.get("http://httpbin.org/get") as resp:
    j = await resp.json()
assert j["args"] == default_params

1 Ответ

0 голосов
/ 29 сентября 2018

Заголовки пользовательских запросов
Если вам нужно добавить заголовки HTTP к запросу, передайте их в dict параметру заголовков.

он находится вдокументация http://docs.aiohttp.org/en/stable/client_advanced.html#custom-request-headers

url = 'http://example.com/image'
payload = b'GIF89a\x01\x00\x01\x00\x00\xff\x00,\x00\x00'
          b'\x00\x00\x01\x00\x01\x00\x00\x02\x00;'
headers = {'content-type': 'image/gif'}

await session.post(url,
                   data=payload,
                   headers=headers)

Чтобы добавить пользовательские параметры по умолчанию и позже, вы можете сделать это следующим образом без каких-либо подклассов ( на основе этой части документации )

import asyncio
from aiohttp import ClientSession

async def fetch(url, params, loop):
    async with ClientSession() as session:
        async with session.get(url, params=params) as response:
            print(response.request_info)
            print(str(response.url))
            print(response.status)
            args = await response.json()
            resp_text = await response.text()
            print('args:', args['args'])
            print(resp_text)

def main(url, params):
    loop = asyncio.get_event_loop()
    return loop.run_until_complete(fetch(url, params, loop))

if __name__ == '__main__':
    url = 'http://httpbin.org/get'
    default_params = {'param1': 1, 'param2': 2}
    main(url, default_params)
    new_params = {'some_parameter': 'a value', 'other_parameter': 'another value'}
    default_params.update(new_params)
    main(url, default_params)

Результат:

RequestInfo(url=URL('http://httpbin.org/get?param1=1&param2=2'), method='GET', headers=<CIMultiDict('Accept': '*/*', 'Accept-Encoding': 'gzip, deflate', 'Host': 'httpbin.org', 'User-Agent': 'Python/3.5 aiohttp/3.4.4')>, real_url=URL('http://httpbin.org/get?param1=1&param2=2'))
  http://httpbin.org/get?param1=1&param2=2
  200
  args: {'param1': '1', 'param2': '2'}
  {
    "args": {
      "param1": "1", 
      "param2": "2"
    }, 
    "headers": {
      "Accept": "*/*", 
      "Accept-Encoding": "gzip, deflate", 
      "Connection": "close", 
      "Host": "httpbin.org", 
      "User-Agent": "Python/3.5 aiohttp/3.4.4"
    }, 
    "origin": "10.10.10.10", 
    "url": "http://httpbin.org/get?param1=1&param2=2"
  }

  RequestInfo(url=URL('http://httpbin.org/get?param1=1&param2=2&some_parameter=a+value&other_parameter=another+value'), method='GET', headers=<CIMultiDict('Accept': '*/*', 'Accept-Encoding': 'gzip, deflate', 'Host': 'httpbin.org', 'User-Agent': 'Python/3.5 aiohttp/3.4.4')>, real_url=URL('http://httpbin.org/get?param1=1&param2=2&some_parameter=a+value&other_parameter=another+value'))
  http://httpbin.org/get?param1=1&param2=2&some_parameter=a+value&other_parameter=another+value
  200
  args: {'other_parameter': 'another value', 'param1': '1', 'param2': '2', 'some_parameter': 'a value'}
  {
    "args": {
      "other_parameter": "another value", 
      "param1": "1", 
      "param2": "2", 
      "some_parameter": "a value"
    }, 
    "headers": {
      "Accept": "*/*", 
      "Accept-Encoding": "gzip, deflate", 
      "Connection": "close", 
      "Host": "httpbin.org", 
      "User-Agent": "Python/3.5 aiohttp/3.4.4"
    }, 
    "origin": "10.10.10.10", 
    "url": "http://httpbin.org/get?param1=1&param2=2&some_parameter=a+value&other_parameter=another+value"
  }
...