hs-m3u8 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

hs_m3u8/__init__.py

@@ -1,3 +1,3 @@
-from hs_m3u8.main import M3u8Downloader
+from hs_m3u8.main import M3u8Downloader, M3u8Key
 
-__version__ = "0.1.3"
+__version__ = "0.1.4"

hs_m3u8/main.py

@@ -54,8 +54,9 @@ class M3u8Key:
         :param key: 密钥
         :param iv: 偏移
         """
+        iv = bytes.fromhex(iv[2:]) if iv and iv.startswith("0x") else iv
         self.key = key
-        self.iv = iv or key
+        self.iv = iv
 
 
 class M3u8Downloader:
@@ -78,6 +79,7 @@ class M3u8Downloader:
         decrypt=False,
         max_workers=None,
         headers=None,
+        key: M3u8Key = None,
         get_m3u8_func: Callable = None,
     ):
         """
@@ -101,6 +103,7 @@ class M3u8Downloader:
         self.save_dir = Path(save_path) / "hls"
         self.save_name = Path(save_path).name
         self.key_path = self.save_dir / "key.key"
+        self.custom_key = key
 
         if not self.save_dir.exists():
             self.save_dir.mkdir(parents=True)
@@ -216,10 +219,16 @@ class M3u8Downloader:
 
         # 保存解密key
         if len(m3u8_obj.keys) > 0 and m3u8_obj.keys[0]:
-            resp = await self.net.get(m3u8_obj.keys[0].absolute_uri, headers=self.headers)
-            key_data = resp.content
+            iv = m3u8_obj.keys[0].iv
+            if not self.custom_key:
+                resp = await self.net.get(m3u8_obj.keys[0].absolute_uri, headers=self.headers)
+                key_data = resp.content
+            else:
+                key_data = self.custom_key.key
+                iv = self.custom_key.iv or iv
+
             self.save_file(key_data, self.key_path)
-            self.ts_key = M3u8Key(key=key_data, iv=m3u8_obj.keys[0].iv)
+            self.ts_key = M3u8Key(key=key_data, iv=iv)
             key = m3u8_obj.segments[0].key
             key.uri = "key.key"
             m3u8_obj.segments[0].key = key
@@ -244,7 +253,7 @@ class M3u8Downloader:
         if Path(ts_path).exists():
             self.ts_path_list[index] = str(ts_path)
             return
-        resp = await self.net.get(ts_item["uri"])
+        resp = await self.net.get(ts_item["uri"], self.headers)
         ts_content = resp.content
         if ts_content is None:
             return

{hs_m3u8-0.1.3.dist-info → hs_m3u8-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hs-m3u8
-Version: 0.1.3
+Version: 0.1.4
 Summary: m3u8 下载器
 Project-URL: homepage, https://github.com/x-haose/hs-m3u8
 Project-URL: repository, https://github.com/x-haose/hs-m3u8

hs_m3u8-0.1.4.dist-info/RECORD

@@ -0,0 +1,5 @@
+hs_m3u8/__init__.py,sha256=WxYMLrnGwdYDEmbxZmUV-l2DPs9jrYDP0cAnHxkqtFs,72
+hs_m3u8/main.py,sha256=RELt4z7AM7OuXYiRkCe-yf4A1JGQ9FtIaBj6Vov3Ig0,11574
+hs_m3u8-0.1.4.dist-info/METADATA,sha256=ddVI5SkdoINuFHSPS9uFGDeOmCybhT1al4_4SmXPHRs,2204
+hs_m3u8-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hs_m3u8-0.1.4.dist-info/RECORD,,

hs_m3u8-0.1.3.dist-info/RECORD

@@ -1,5 +0,0 @@
-hs_m3u8/__init__.py,sha256=t0R3Q2zVnUtOiJ8e2sL29N4QkkBHXKgC54n6OqMbwJs,63
-hs_m3u8/main.py,sha256=kJ4HV1eISz80ttxsr2E9K-tfTKK79URiEJviPcc5tfg,11259
-hs_m3u8-0.1.3.dist-info/METADATA,sha256=nskQSk_KjDfVtGPhdr6cYXdJbfrbHJXnf7JnvtxIcjg,2204
-hs_m3u8-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hs_m3u8-0.1.3.dist-info/RECORD,,