lazysdk 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

lazysdk/lazym3u8.py

@@ -5,6 +5,7 @@ from . import lazytime
 from . import lazyfile
 from . import lazypath
 from . import lazymd5
+from . import lazyprocess
 import subprocess
 import threading
 import requests
@@ -503,6 +504,105 @@ def download_fragment(
     return download_res
 
 
+def download_fragment_single(
+        task_index,
+        task_info,
+):
+    """
+    模块功能：下载单个碎片到指定文件夹
+    :param fragment_url: 碎片下载地址
+    :param fragment_url_name: 碎片名
+    :param fragment_path:碎片保存路径
+    :param aes_key: aes解密key
+    :param headers:请求头
+    :param fragment_suffix:碎片文件后缀名
+    :return:
+    """
+    fragment_url = task_info['fragment_url']
+    fragment_url_name = task_info['fragment_url_name']
+    fragment_path = task_info['fragment_path']
+    aes_key = None,
+    headers = None,
+    fragment_suffix = task_info['fragment_suffix']
+
+    if headers is None:
+        headers_local = default_headers
+    else:
+        headers_local = headers
+
+    print('正在下载', fragment_url_name, fragment_url)
+    fragment_file_name = '%s%s%s.%s' % (fragment_path, path_separator, fragment_url_name, fragment_suffix)  # 生成碎片文件名
+    # 开始下载碎片文件
+    while True:
+        try:
+            with open(fragment_file_name, "ab+") as f:
+                response = requests.get(
+                    url=fragment_url,
+                    headers=headers_local,
+                    stream=True,
+                    timeout=timeout
+                )
+                total_length = response.headers.get('content-length')
+
+                if total_length is None:
+                    print("Attention: There is no content length in header!")  # 未返回长度信息
+                else:
+                    dl = 0
+                    total_length = int(total_length)
+                    for data in track(
+                            sequence=response.iter_content(chunk_size=8192),
+                            description='下载中',
+                            total=total_length/8192,
+                            show_speed=True
+                    ):
+                        dl += len(data)
+                        if aes_key is None:
+                            data_decode = data
+                        else:
+                            data_decode = lazyCrypto.aes_decode(data, aes_key)
+                        f.write(data_decode)
+                        f.flush()
+            break
+        except:
+            showlog.warning('下载超时，将在1秒后重试...')
+            os.remove(fragment_file_name)
+            time.sleep(1)
+            pass
+
+
+def download_fragment_quick(
+        url_list,
+        subprocess_limit=None
+):
+    fragment_path = 'fragment_%s' % str(time.time()).replace('.', '')  # 使用时间戳命名
+    fragment_suffix = 'ts'
+    lazypath.make_path(fragment_path)
+    task_list = list()
+    fragment_file_name_list = list()
+    for url_index, each_url in enumerate(url_list):
+        task_list.append(
+            {
+                'fragment_url': each_url,
+                'fragment_url_name': url_index,
+                'fragment_path': fragment_path,
+                'fragment_suffix': fragment_suffix
+            }
+        )
+        fragment_file_name = '%s%s%s.%s' % (fragment_path, path_separator, url_index, fragment_suffix)  # 生成碎片文件名
+        fragment_file_name_list.append(fragment_file_name)
+    lazyprocess.run(
+        task_list=task_list,
+        task_function=download_fragment_single,
+        subprocess_limit=subprocess_limit
+    )
+    download_res = {
+        'fragment_path': fragment_path,
+        'fragment_suffix': fragment_suffix,
+        'fragment_file_name_list': fragment_file_name_list
+    }
+    return download_res
+
+
 def download_m3u8(
         m3u8_link=None,
         m3u8_file_path=None,

{lazysdk-0.1.3.dist-info → lazysdk-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lazysdk
-Version: 0.1.3
+Version: 0.1.5
 Summary: 基于Python的懒人包
 Home-page: https://gitee.com/ZeroSeeker/lazysdk
 Author: ZeroSeeker

{lazysdk-0.1.3.dist-info → lazysdk-0.1.5.dist-info}/RECORD

@@ -8,7 +8,7 @@ lazysdk/lazyfile.py,sha256=Rdj9f11IUyd1lLAZu-G4JHZqSKrvBQGkLCeJ9CxSzPQ,20267
 lazysdk/lazyflask.py,sha256=VUhJqtNz_evVyGOS40FKkkQWWrqGEUZ26cv1RQxS388,727
 lazysdk/lazyid.py,sha256=pEmRpN78-6vdz4hzLxgoCSHJw6gvnOVQyNjqusZZLQE,607
 lazysdk/lazyip.py,sha256=ZM4N0nCimW4r4z1_Bymb-ZAfabR_3L-hB5Biyyz77-I,451
-lazysdk/lazym3u8.py,sha256=8lBBzIl_LajvmOuKvbp17nT2XItHDzHDfhVSd4K1mQs,28027
+lazysdk/lazym3u8.py,sha256=PyBOpgFlkKNQqi9MyVcueBjEdqDKaSkvUBG5ggvdT3k,31559
 lazysdk/lazymd5.py,sha256=JSKIcUmH2zwPDDJAd9XajTYd59eKidoKAkLLNU8OHM4,1112
 lazysdk/lazypath.py,sha256=cPjp5kfJcbp6884HgJY0SIUAhQ85qA-o9N-MKhk4uBU,4305
 lazysdk/lazyprocess.py,sha256=ifRnQmQMwhP8wmy0HF6_sYZ747XlQ7esp0jiS2JnOPk,5079
@@ -20,7 +20,7 @@ lazysdk/lazytext.py,sha256=1f2B0HJVweWVkmCx3NcwXaRw0ZQHYc7PKTv38hdgAGQ,1448
 lazysdk/lazytime.py,sha256=lOQ5U8iWa9dp1ZJ8SjfZ2FRBj9t43SsteYbxDcXkFzo,23199
 lazysdk/lazyua.py,sha256=QI4EgZAMF9AqXJJvaDEzsMSTbsR47_y8CkJkfPbX-kM,3783
 lazysdk/lazywebhook.py,sha256=HwJIZ5j3wUpnEfPkFY-Xq9T_9xdhCJ71dYsWNX22_3Y,8386
-lazysdk-0.1.3.dist-info/METADATA,sha256=n8O2bO3ZjwxSLlVRiFF5HlLuFhWk2ufs0stpfZb6EGI,1730
-lazysdk-0.1.3.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-lazysdk-0.1.3.dist-info/top_level.txt,sha256=--bGS42ZHUhVeO83y1wfvKFg6OjkTLxQ4V4riqPQljY,8
-lazysdk-0.1.3.dist-info/RECORD,,
+lazysdk-0.1.5.dist-info/METADATA,sha256=6khY7vOS649UU4oFmYW5XsCAHvJiS2HHewQIfd0HFnU,1730
+lazysdk-0.1.5.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+lazysdk-0.1.5.dist-info/top_level.txt,sha256=--bGS42ZHUhVeO83y1wfvKFg6OjkTLxQ4V4riqPQljY,8
+lazysdk-0.1.5.dist-info/RECORD,,