Add BookWyrm user-agent to http requests

This allows other software to identify BookWyrm in calls, as well as
will allow BookWyrm to differentiate between calls done from other
fediverse software and BookWyrm to answer with specific BookWyrm data.
This commit is contained in:
Renato "Lond" Cerqueira
2020-12-30 12:35:11 +01:00
parent 26c0880fe4
commit fb10cb35ad
3 changed files with 17 additions and 3 deletions

View File

@ -8,7 +8,7 @@ import requests
from requests import HTTPError
from requests.exceptions import SSLError
from bookwyrm import activitypub, models
from bookwyrm import activitypub, models, settings
class ConnectorException(HTTPError):
@ -42,6 +42,7 @@ class AbstractMinimalConnector(ABC):
'%s%s' % (self.search_url, query),
headers={
'Accept': 'application/json; charset=utf-8',
'User-Agent': settings.USER_AGENT,
},
)
if not resp.ok:
@ -196,6 +197,7 @@ def get_data(url):
url,
headers={
'Accept': 'application/json; charset=utf-8',
'User-Agent': settings.USER_AGENT,
},
)
except RequestError:
@ -213,7 +215,12 @@ def get_data(url):
def get_image(url):
''' wrapper for requesting an image '''
try:
resp = requests.get(url)
resp = requests.get(
url,
headers={
'User-Agent': settings.USER_AGENT,
},
)
except (RequestError, SSLError):
return None
if not resp.ok: