rtzip 0.2.0__tar.gz → 0.2.2__tar.gz

rtzip-0.2.2/LICENSE

@@ -0,0 +1,28 @@
+BSD 3-Clause License
+
+Copyright (c) 2026, LittleNightSong
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

{rtzip-0.2.0 → rtzip-0.2.2}/PKG-INFO

@@ -1,5 +1,7 @@
 Metadata-Version: 2.4
 Name: rtzip
-Version: 0.2.0
+Version: 0.2.2
 Requires-Python: >=3.12
+License-File: LICENSE
 Requires-Dist: obstruct==0.5.2
+Dynamic: license-file

rtzip-0.2.2/README.md

@@ -0,0 +1,18 @@
+# Rtzip
+_一个以最小的数据量提取远程 zip 文件内容的库_
+
+
+## 特点
+- 不与任何 HTTP 库耦合
+- 依赖简单（只有用于解析二进制结构的 `obstruct`）
+- 兼容标准 zip 格式、ZIP64 扩展
+
+
+## 安装
+```commandline
+pip install rtzip
+```
+
+> [!TIP]
+> 本库无计划新功能，若是让自行添加功能，可克隆仓库并重写 `rtzip/remotezip.py` 下的代码
+> 大多数库的功能已完成并可以直接投入使用

{rtzip-0.2.0 → rtzip-0.2.2}/pyproject.toml

@@ -1,7 +1,12 @@
 [project]
 name = "rtzip"
-version = "0.2.0"
+version = "0.2.2"
 requires-python = ">=3.12"
 dependencies = [
     "obstruct==0.5.2",
 ]
+
+[dependency-groups]
+dev = [
+    "niquests>=3.21.0",
+]

{rtzip-0.2.0 → rtzip-0.2.2}/src/rtzip/__init__.py

@@ -1,3 +1,3 @@
 from .models import EOCD, Zip64EOCD, CDEntry
 from .errors import *
-from .remotezip import RemoteZip
+from .remotezip import RemoteZip, DataSourceCallbacks

{rtzip-0.2.0 → rtzip-0.2.2}/src/rtzip/const.py

@@ -1,5 +1,5 @@
 EOCD_SIGNATURE = b'PK\x05\x06'
-CDENTITY_SIGNATURE = b'PK\x01\x02'
+CDENTRY_SIGNATURE = b'PK\x01\x02'
 LFH_SIGNATURE = b'PK\x03\x04'
 ZIP64_EOCD_LOCATOR_SIGNATURE = b'PK\x06\x07'
 ZIP64_EOCD_SIGNATURE = b'PK\x06\x06'

{rtzip-0.2.0 → rtzip-0.2.2}/src/rtzip/models.py

@@ -5,6 +5,8 @@ from typing import Annotated, Self
 
 import obstruct as obs
 
+from rtzip.utils import exactly_get_slice
+
 EXTRA_FIELD_HEADER = struct.Struct("<HH")
 
 
@@ -245,13 +247,13 @@ class CDEntry(obs.Struct):
         filename_len, extra_len, comment_len = self.filename_len, self.extra_len, self.comment_len
 
         pos = offset + header_size
-        filename = bytes(buffer[pos: pos + filename_len])
+        filename = bytes(exactly_get_slice(buffer, pos, filename_len))
         pos += filename_len
 
-        extra_attrs = buffer[pos: pos + extra_len].tobytes()
+        extra_attrs = bytes(exactly_get_slice(buffer, pos, extra_len))
         pos += extra_len
 
-        comment = buffer[pos: pos + comment_len].tobytes()
+        comment = bytes(exactly_get_slice(buffer, pos, comment_len))
         pos += comment_len
 
         self.filename = filename

{rtzip-0.2.0 → rtzip-0.2.2}/src/rtzip/remotezip.py

@@ -1,11 +1,11 @@
 import struct
 from typing import AsyncGenerator
 
+from .const import *
 from .errors import UnsupportedAlgorithmError, UnsupportedCryptoError, UnsupportedDataDescriptorError
 from .models import EOCD, Zip64EOCD, CDEntry, LocalFileHeader
 from .utils import deflate_wrapper, single_chunk_wrapper
 
-from .const import *
 
 
 class DataSourceCallbacks:
@@ -18,7 +18,7 @@ class DataSourceCallbacks:
         :return: bytes | bytearray | memoryview
         """
 
-    def read_range_stream(self, offset, length) -> AsyncGenerator[bytes | bytearray | memoryview]:
+    def read_range_stream(self, offset, length) -> AsyncGenerator[bytes | bytearray | memoryview, None]:
         """
         流式读取时指定的一段数据
 
@@ -48,6 +48,7 @@ class RemoteZip:
     :ivar file_mapping: 构建的文件映射表，可能为 None
     :ivar is_zip64: 此 zip 文件是否启动了 zip64 扩展，默认为 False
     """
+
     async def _read_range(self, offset, length):
         return await self.source.read_range(offset, length)
 
@@ -62,7 +63,7 @@ class RemoteZip:
 
         self.cd_info: EOCD | Zip64EOCD | None = None
         self.files: list[CDEntry] = []
-        self.file_mapping: dict[str, CDEntry] | None = None
+        self.file_mapping: dict[bytes, CDEntry] | None = None
 
         self.is_zip64 = False
 
@@ -78,6 +79,7 @@ class RemoteZip:
         chunk_size = initial_chunk
 
         last_start = await self._get_total_size()
+        # print("File Size:", last_start)
         cnt = 0
 
         buffer = b''
@@ -174,6 +176,8 @@ class RemoteZip:
             nonlocal buffer
             entry, nbytes = CDEntry.from_buffer(buffer)
 
+            # print(f'{len(files) + 1:03} Get Entry', entry)
+
             if self.is_zip64:
                 entry.fix_by_zip64()
 
@@ -181,14 +185,15 @@ class RemoteZip:
             buffer = buffer[nbytes + 4:]
 
         async for chunk in self._read_range_stream(eocd.cd_offset, eocd.cd_size):
+            # print("Read Chunk", len(chunk))
             # print(chunk)
             buffer.extend(chunk)
 
-            while buffer and buffer[:4] == CDENTITY_SIGNATURE:
+            while buffer and buffer[:4] == CDENTRY_SIGNATURE:
                 try:
                     consume_entry()
-                except struct.error:
-                    continue
+                except (struct.error, AssertionError):
+                    break
 
         if buffer and not ignore_extra:
             # print("Read Entries: ", len(files))
@@ -244,7 +249,7 @@ class RemoteZip:
 
         return m
 
-    def find_file_entry(self, filename):
+    def find_file_entry(self, filename: bytes):
         """
         从本地已有的数据获取 filename 对应的 CDEntry 对象
 
@@ -265,7 +270,7 @@ class RemoteZip:
             else:
                 raise FileNotFoundError(filename)
 
-    async def _stream_single_file(self, filename):
+    async def _stream_single_file(self, filename: bytes):
         entry = self.find_file_entry(filename)
 
         if entry.is_encrypted:
@@ -292,7 +297,7 @@ class RemoteZip:
             case 0x0008:
                 return deflate_wrapper(raw_generator)
 
-    async def stream_single_file(self, filename):
+    async def stream_single_file(self, filename: bytes):
         """
         流式读取一个压缩包内的文件
 
@@ -303,7 +308,7 @@ class RemoteZip:
         async for i in gen:
             yield i
 
-    async def load_single_file(self, filename) -> bytearray:
+    async def load_single_file(self, filename: bytes) -> bytearray:
         """
         直接加载文件的内容
 

{rtzip-0.2.0 → rtzip-0.2.2}/src/rtzip/utils.py

@@ -13,3 +13,9 @@ async def deflate_wrapper(gen):
 
 async def single_chunk_wrapper(data: Buffer):
     yield data
+
+
+def exactly_get_slice(__obj, __offset, __length):
+    result = __obj[__offset:__offset + __length]
+    assert len(result) == __length
+    return result

{rtzip-0.2.0 → rtzip-0.2.2}/src/rtzip.egg-info/PKG-INFO

@@ -1,5 +1,7 @@
 Metadata-Version: 2.4
 Name: rtzip
-Version: 0.2.0
+Version: 0.2.2
 Requires-Python: >=3.12
+License-File: LICENSE
 Requires-Dist: obstruct==0.5.2
+Dynamic: license-file

{rtzip-0.2.0 → rtzip-0.2.2}/src/rtzip.egg-info/SOURCES.txt

@@ -1,3 +1,5 @@
+LICENSE
+README.md
 pyproject.toml
 src/rtzip/__init__.py
 src/rtzip/const.py