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,158 @@
+# Copyright (c) 2020-2022, Manfred Moitzi
+# License: MIT License
+from __future__ import annotations
+from typing import Iterable, NamedTuple, BinaryIO
+
+from .const import DXFStructureError
+from ezdxf.tools.codepage import toencoding
+
+
+class IndexEntry(NamedTuple):
+    code: int
+    value: str
+    location: int
+    line: int
+
+
+class FileStructure:
+    """DXF file structure representation stored as file locations.
+
+    Store all DXF structure tags and some other tags as :class:`IndexEntry`
+    tuples:
+
+        - code: group code
+        - value: tag value as string
+        - location: file location as int
+        - line: line number as int
+
+    Indexed tags:
+
+        - structure tags, every tag with group code 0
+        - section names, (2, name) tag following a (0, SECTION) tag
+        - entity handle tags with group code 5, the DIMSTYLE handle group code
+          105 is also stored as group code 5
+
+    """
+
+    def __init__(self, filename: str):
+        # stores the file system name of the DXF document.
+        self.filename: str = filename
+        # DXF version if header variable $ACADVER is present, default is DXFR12
+        self.version: str = "AC1009"
+        # Python encoding required to read the DXF document as text file.
+        self.encoding: str = "cp1252"
+        self.index: list[IndexEntry] = []
+
+    def print(self) -> None:
+        print(f"Filename: {self.filename}")
+        print(f"DXF Version: {self.version}")
+        print(f"encoding: {self.encoding}")
+        for entry in self.index:
+            print(f"Line: {entry.line} - ({entry.code}, {entry.value})")
+
+    def get(self, code: int, value: str, start: int = 0) -> int:
+        """Returns index of first entry matching `code` and `value`."""
+        self_index = self.index
+        index: int = start
+        count: int = len(self_index)
+        while index < count:
+            entry = self_index[index]
+            if entry.code == code and entry.value == value:
+                return index
+            index += 1
+        raise ValueError(f"No entry for tag ({code}, {value}) found.")
+
+    def fetchall(
+        self, code: int, value: str, start: int = 0
+    ) -> Iterable[IndexEntry]:
+        """Iterate over all specified entities.
+
+        e.g. fetchall(0, 'LINE') returns an iterator for all LINE entities.
+
+        """
+        for entry in self.index[start:]:
+            if entry.code == code and entry.value == value:
+                yield entry
+
+
+def load(filename: str) -> FileStructure:
+    """Load DXF file structure for file `filename`, the file has to be seekable.
+
+    Args:
+        filename: file system file name
+
+    Raises:
+        DXFStructureError: Invalid or incomplete DXF file.
+
+    """
+    file_structure = FileStructure(filename)
+    file: BinaryIO = open(filename, mode="rb")
+    line: int = 1
+    eof: bool = False
+    header: bool = False
+    index: list[IndexEntry] = []
+    prev_code: int = -1
+    prev_value: bytes = b""
+    structure = None  # the current structure tag: 'SECTION', 'LINE', ...
+
+    def load_tag() -> tuple[int, bytes]:
+        nonlocal line
+        try:
+            code = int(file.readline())
+        except ValueError:
+            raise DXFStructureError(f"Invalid group code in line {line}")
+
+        if code < 0 or code > 1071:
+            raise DXFStructureError(f"Invalid group code {code} in line {line}")
+        value = file.readline().rstrip(b"\r\n")
+        line += 2
+        return code, value
+
+    def load_header_var() -> str:
+        _, value = load_tag()
+        return value.decode()
+
+    while not eof:
+        location = file.tell()
+        tag_line = line
+        try:
+            code, value = load_tag()
+            if header and code == 9:
+                if value == b"$ACADVER":
+                    file_structure.version = load_header_var()
+                elif value == b"$DWGCODEPAGE":
+                    file_structure.encoding = toencoding(load_header_var())
+                continue
+        except IOError:
+            break
+
+        if code == 0:
+            # All structure tags have group code == 0, store file location
+            structure = value
+            index.append(IndexEntry(0, value.decode(), location, tag_line))
+            eof = value == b"EOF"
+
+        elif code == 2 and prev_code == 0 and prev_value == b"SECTION":
+            # Section name is the tag (2, name) following the (0, SECTION) tag.
+            header = value == b"HEADER"
+            index.append(IndexEntry(2, value.decode(), location, tag_line))
+
+        elif code == 5 and structure != b"DIMSTYLE":
+            # Entity handles have always group code 5.
+            index.append(IndexEntry(5, value.decode(), location, tag_line))
+
+        elif code == 105 and structure == b"DIMSTYLE":
+            # Except the DIMSTYLE table entry has group code 105.
+            index.append(IndexEntry(5, value.decode(), location, tag_line))
+
+        prev_code = code
+        prev_value = value
+
+    file.close()
+    if not eof:
+        raise DXFStructureError(f"Unexpected end of file.")
+
+    if file_structure.version >= "AC1021":  # R2007 and later
+        file_structure.encoding = "utf-8"
+    file_structure.index = index
+    return file_structure