refactor: excel parse

2026-04-16 10:01:11 +08:00
parent 680ecc320f
commit f62f95ec02
7941 changed files with 2899112 additions and 0 deletions
@@ -0,0 +1,87 @@
+# Copyright (c) 2014-2023, Manfred Moitzi
+# License: MIT License
+from __future__ import annotations
+from typing import BinaryIO, cast, TextIO, Optional, Iterator
+import zipfile
+from contextlib import contextmanager
+
+from ezdxf.lldxf.validator import is_dxf_stream, dxf_info
+
+CRLF = b"\r\n"
+LF = b"\n"
+
+
+class ZipReader:
+    def __init__(self, zip_archive_name: str, errors="surrogateescape"):
+        if not zipfile.is_zipfile(zip_archive_name):
+            raise IOError(f"'{zip_archive_name}' is not a zip archive.")
+        self.zip_archive_name = zip_archive_name
+        self.zip_archive: Optional[zipfile.ZipFile] = None
+        self.dxf_file_name: Optional[str] = None
+        self.dxf_file: Optional[BinaryIO] = None
+        self.encoding = "cp1252"
+        self.errors = errors
+        self.dxfversion = "AC1009"
+
+    def open(self, dxf_file_name: Optional[str] = None) -> None:
+        def open_dxf_file() -> BinaryIO:
+            # Open always in binary mode:
+            return cast(BinaryIO, self.zip_archive.open(self.dxf_file_name))  # type: ignore
+
+        self.zip_archive = zipfile.ZipFile(self.zip_archive_name)
+        self.dxf_file_name = (
+            dxf_file_name
+            if dxf_file_name is not None
+            else self.get_first_dxf_file_name()
+        )
+        self.dxf_file = open_dxf_file()
+
+        # Reading with standard encoding 'cp1252' - readline() fails if leading
+        # comments contain none ASCII characters.
+        if not is_dxf_stream(cast(TextIO, self)):
+            raise IOError(f"'{self.dxf_file_name}' is not a DXF file.")
+        self.dxf_file = open_dxf_file()  # restart
+        self.get_dxf_info()
+        self.dxf_file = open_dxf_file()  # restart
+
+    def get_first_dxf_file_name(self) -> str:
+        dxf_file_names = self.get_dxf_file_names()
+        if len(dxf_file_names) > 0:
+            return dxf_file_names[0]
+        else:
+            raise IOError("No DXF files found.")
+
+    def get_dxf_file_names(self) -> list[str]:
+        assert self.zip_archive is not None
+        return [
+            name
+            for name in self.zip_archive.namelist()
+            if name.lower().endswith(".dxf")
+        ]
+
+    def get_dxf_info(self) -> None:
+        info = dxf_info(cast(TextIO, self))
+        # Since DXF R2007 (AC1021) file encoding is always 'utf-8'
+        self.encoding = info.encoding if info.version < "AC1021" else "utf-8"
+        self.dxfversion = info.version
+
+    def readline(self) -> str:
+        assert self.dxf_file is not None
+        next_line = self.dxf_file.readline().replace(CRLF, LF)
+        return str(next_line, self.encoding, self.errors)
+
+    def close(self) -> None:
+        assert self.zip_archive is not None
+        self.zip_archive.close()
+
+
+@contextmanager
+def ctxZipReader(
+    zipfilename: str,
+    filename: Optional[str] = None,
+    errors: str = "surrogateescape",
+) -> Iterator[ZipReader]:
+    zip_reader = ZipReader(zipfilename, errors=errors)
+    zip_reader.open(filename)
+    yield zip_reader
+    zip_reader.close()