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)