fixed txt download error

Author: zaxtyson

README.md

@@ -34,6 +34,10 @@
 
 # 更新日志
 
+## `v2.6.1`
+
+- 修复下载某些 txt 文件失败的问题[#53](https://github.com/zaxtyson/LanZouCloud-API/issues/53)
+
 ## `v2.6.0`
 
 - 修复无法上传文件的问题 [#52](https://github.com/zaxtyson/LanZouCloud-API/pull/52)

lanzou/api/__init__.py

@@ -1,5 +1,5 @@
 from lanzou.api.core import LanZouCloud
 
-version = '2.6.0'
+version = '2.6.1'
 
 __all__ = ['utils', 'types', 'models', 'LanZouCloud', 'version']

lanzou/api/core.py

@@ -947,8 +947,26 @@ def down_file_by_url(self, share_url, pwd='', save_path='./Download', *, callbac
         resp = self._get(info.durl, stream=True)
         if not resp:
             return LanZouCloud.FAILED
-        total_size = int(resp.headers['Content-Length'])
 
+        content_length = resp.headers.get('Content-Length', None)
+        # 如果无法获取 Content-Length, 先读取一点数据, 再尝试获取一次
+        # 通常只需读取 1 字节数据
+        data_iter = resp.iter_content(chunk_size=1)
+        while not content_length:
+            logger.warning("Not found Content-Length in response headers")
+            logger.debug("Read 1 byte from stream...")
+            try:
+                next(data_iter)
+            except StopIteration:
+                logger.debug("Please wait for a moment before downloading")
+                return LanZouCloud.FAILED
+            resp_ = self._get(info.durl, stream=True)
+            if not resp_:
+                return LanZouCloud.FAILED
+            content_length = resp_.headers.get('Content-Length', None)
+            logger.debug(f"Content-Length: {content_length}")
+
+        total_size = int(content_length)
         file_path = save_path + os.sep + info.name
         if os.path.exists(file_path):
             if overwrite: