Copy the pattern from requests.api to use a slightly more stable API
# Copyright (c) 2017 LINE Corporation
# These sources are released under the terms of the MIT license: see LICENSE
-import requests
+import requests.sessions
from promgen.version import __version__
-def post(url, *args, **kwargs):
- '''Wraps requests.post with our user-agent'''
- if 'headers' not in kwargs:
- kwargs['headers'] = {}
- kwargs['headers']['user-agent'] = 'promgen/{}'.format(__version__)
+def post(url, **kwargs):
+ with requests.sessions.Session() as session:
+ session.headers['User-Agent'] = 'promgen/{}'.format(__version__)
+ return session.post(url, **kwargs)
- return requests.post(url, *args, **kwargs)
-
-def get(url, *args, **kwargs):
- '''Wraps requests.post with our user-agent'''
- if 'headers' not in kwargs:
- kwargs['headers'] = {}
- kwargs['headers']['user-agent'] = 'promgen/{}'.format(__version__)
-
- return requests.get(url, *args, **kwargs)
+def get(url, **kwargs):
+ with requests.sessions.Session() as session:
+ session.headers['User-Agent'] = 'promgen/{}'.format(__version__)
+ return session.get(url, **kwargs)
