From 7152ec2e1a946f25d7b06244291c1b9c8ce0cb76 Mon Sep 17 00:00:00 2001 From: ricterz Date: Sat, 9 May 2015 14:00:03 +0800 Subject: [PATCH] global timeout --- hentai/downloader.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/hentai/downloader.py b/hentai/downloader.py index 672b23f..98622b6 100644 --- a/hentai/downloader.py +++ b/hentai/downloader.py @@ -1,6 +1,7 @@ #coding: utf-8 import os import sys +import socket import threading import Queue import requests @@ -8,6 +9,11 @@ from urlparse import urlparse from hentai.logger import logger +# global timeout +timeout = 30 +socket.setdefaulttimeout(timeout) + + class Downloader(object): _instance = None @@ -34,7 +40,7 @@ class Downloader(object): filename = filename if filename else os.path.basename(urlparse(url).path) try: with open(os.path.join(folder, filename), "wb") as f: - response = requests.get(url, stream=True) + response = requests.get(url, stream=True, timeout=10) length = response.headers.get('content-length') if length is None: f.write(response.content)