From cecc4121a6394824eeda96a0f77e8246523774a7 Mon Sep 17 00:00:00 2001
From: pl <0x00-pl@gmail.com>
Date: Sat, 11 Oct 2014 18:34:22 +0800
Subject: [PATCH] support stream vedio.

---
 src/you_get/common.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/you_get/common.py b/src/you_get/common.py
index 34c1fda7..6ed888be 100644
--- a/src/you_get/common.py
+++ b/src/you_get/common.py
@@ -299,7 +299,7 @@ def url_save(url, filepath, bar, refer = None, is_part = False, faker = False):
     else:
         open_mode = 'wb'
 
-    if received < file_size:
+    if file_size==None or received < file_size:
         if faker:
             headers = fake_headers
         else:
@@ -315,9 +315,9 @@ def url_save(url, filepath, bar, refer = None, is_part = False, faker = False):
             end_length = end = int(response.headers['content-range'][6:].split('/')[1])
             range_length = end_length - range_start
         except:
-            range_length = int(response.headers['content-length'])
+            range_length = response.headers['content-length'] and int(response.headers['content-length'])
 
-        if file_size != received + range_length:
+        if range_length==None or file_size != received + range_length:
             received = 0
             if bar:
                 bar.received = 0