zedfrigg/bookwyrm
1
0
Fork 0
Code Activity

Add BookWyrm user-agent to http requests

Browse source 
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
Download patch file Download diff file Expand all files Collapse all files

11
bookwyrm/connectors/abstract_connector.py
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: