Ensure timeout in requests

merge-requests/154/head
Eliot Berriot 2018-04-08 13:33:36 +02:00
rodzic b8c7e960c3
commit 206ae296b6
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: DD6965E2476E5C27
4 zmienionych plików z 5 dodań i 1 usunięć

Wyświetl plik

@ -73,6 +73,7 @@ def deliver(activity, on_behalf_of, to=[]):
auth=auth, auth=auth,
json=activity, json=activity,
url=recipient_actor.inbox_url, url=recipient_actor.inbox_url,
timeout=5,
headers={ headers={
'Content-Type': 'application/activity+json' 'Content-Type': 'application/activity+json'
} }

Wyświetl plik

@ -31,6 +31,7 @@ def remove_tags(text):
def get_actor_data(actor_url): def get_actor_data(actor_url):
response = session.get_session().get( response = session.get_session().get(
actor_url, actor_url,
timeout=5,
headers={ headers={
'Accept': 'application/activity+json', 'Accept': 'application/activity+json',
} }
@ -42,6 +43,7 @@ def get_actor_data(actor_url):
raise ValueError( raise ValueError(
'Invalid actor payload: {}'.format(response.text)) 'Invalid actor payload: {}'.format(response.text))
def get_actor(actor_url): def get_actor(actor_url):
data = get_actor_data(actor_url) data = get_actor_data(actor_url)
serializer = serializers.ActorSerializer(data=data) serializer = serializers.ActorSerializer(data=data)

Wyświetl plik

@ -47,7 +47,7 @@ def get_resource(resource_string):
username, hostname = clean_acct(resource, ensure_local=False) username, hostname = clean_acct(resource, ensure_local=False)
url = 'https://{}/.well-known/webfinger?resource={}'.format( url = 'https://{}/.well-known/webfinger?resource={}'.format(
hostname, resource_string) hostname, resource_string)
response = session.get_session().get(url) response = session.get_session().get(url, timeout=5)
response.raise_for_status() response.raise_for_status()
serializer = serializers.ActorWebfingerSerializer(data=response.json()) serializer = serializers.ActorWebfingerSerializer(data=response.json())
serializer.is_valid(raise_exception=True) serializer.is_valid(raise_exception=True)

Wyświetl plik

@ -218,6 +218,7 @@ class TrackFileViewSet(viewsets.ReadOnlyModelViewSet):
library_track.audio_url, library_track.audio_url,
auth=auth, auth=auth,
stream=True, stream=True,
timeout=20,
headers={ headers={
'Content-Type': 'application/activity+json' 'Content-Type': 'application/activity+json'
}) })