1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
# Copyright (c) 2020, Manfred Moitzi
# License: MIT License
import os
import pytest
import random
from ezdxf import recover
from ezdxf.audit import AuditError
from ezdxf.lldxf.tagger import tag_compiler, ascii_tags_loader
BASEDIR = os.path.dirname(__file__)
DATADIR = "data"
RECOVER1 = "recover01.dxf"
RECOVER2 = "recover02.dxf"
CC_DXFLIB = "cc_dxflib.dxf"
EMPTY_HANDLES = "empty_handles.dxf"
def fullpath(name):
filename = os.path.join(BASEDIR, DATADIR, name)
if not os.path.exists(filename):
pytest.skip(f"File {filename} not found.")
return filename
@pytest.fixture
def tags01():
filename = fullpath(RECOVER1)
tool = recover.Recover()
with open(filename, "rb") as fp:
return list(tool.load_tags(fp, errors="ignore"))
def test_bytes_loader():
filename = fullpath(RECOVER1)
with open(filename, "rb") as fp:
tags = list(recover.bytes_loader(fp))
assert len(tags) == 14736
def test_safe_tag_loader():
filename = fullpath(RECOVER1)
with open(filename, "rt", encoding="cp1252") as fp:
expected = list(tag_compiler(iter(ascii_tags_loader(fp))))
with open(filename, "rb") as fp:
tags = list(recover.safe_tag_loader(fp))
assert len(tags) == len(expected)
assert tags[:100] == tags[:100]
def test_rebuild_sections(tags01):
tool = recover.Recover()
sections = tool.rebuild_sections(tags01)
expected = sum(int(tag == (0, "SECTION")) for tag in tags01)
orphans = sections.pop()
assert len(sections) == expected
assert len(orphans) == 4
def test_build_section_dict(tags01):
tool = recover.Recover()
sections = tool.rebuild_sections(tags01)
tool.load_section_dict(sections)
assert len(tool.section_dict) == 2
header = tool.section_dict["HEADER"][0]
assert len(header) == 6
assert header[0] == (0, "SECTION")
assert header[1] == (2, "HEADER")
assert len(tool.section_dict["ENTITIES"]) == 1505
def test_readfile_recover01_dxf():
doc, auditor = recover.readfile(fullpath(RECOVER1))
assert doc.dxfversion == "AC1009"
assert auditor.has_errors is False
@pytest.fixture
def tags02():
filename = fullpath(RECOVER2)
tool = recover.Recover()
with open(filename, "rb") as fp:
return list(tool.load_tags(fp, errors="ignore"))
def test_rebuild_tables(tags02):
recover_tool = recover.Recover()
sections = recover_tool.rebuild_sections(tags02)
recover_tool.load_section_dict(sections)
tables = recover_tool.section_dict.get("TABLES")
random.shuffle(tables)
tables = recover_tool.rebuild_tables(tables)
assert tables[0] == [(0, "SECTION"), (2, "TABLES")]
assert tables[1][0] == (0, "TABLE")
assert tables[1][1] == (2, "VPORT")
assert tables[2][0] == (0, "VPORT")
assert tables[3][0] == (0, "ENDTAB")
assert tables[4][0] == (0, "TABLE")
assert tables[4][1] == (2, "LTYPE")
assert tables[5][0] == (0, "LTYPE")
assert tables[8][0] == (0, "ENDTAB")
assert tables[-5][0] == (0, "TABLE")
assert tables[-5][1] == (2, "BLOCK_RECORD")
assert tables[-4][0] == (0, "BLOCK_RECORD")
assert tables[-1][0] == (0, "ENDTAB")
def test_readfile_recover02_dxf():
doc, auditor = recover.readfile(fullpath(RECOVER2))
assert doc.dxfversion == "AC1032"
assert auditor.has_errors is False
# Auditor should restore deleted BLOCK-RECORD table head:
blkrec_head = doc.block_records.head
assert blkrec_head.dxf.handle is not None
assert blkrec_head.dxf.handle in doc.entitydb
# Auditor should update/fix BLOCK_RECORD entries owner handle:
for entry in doc.block_records:
assert (
entry.dxf.owner == blkrec_head.dxf.handle
), "Auditor() should update table-entry owner handle."
# Auditor should restore invalid VPORT table-head owner handle:
vport_head = doc.viewports.head
assert (
vport_head.dxf.owner == "0"
), "Auditor() should repair invalid table-head owner handle."
# Auditor should fix invalid VPORT table-entry owner handle:
vport = doc.viewports.get("*Active")[0]
assert (
vport.dxf.owner == vport_head.dxf.handle
), "Auditor() should update table-entry owner handle."
def test_read_cc_dxflib_file():
doc, auditor = recover.readfile(fullpath(CC_DXFLIB))
codes = {fix.code for fix in auditor.fixes}
assert AuditError.REMOVED_UNSUPPORTED_SECTION in codes
assert AuditError.REMOVED_UNSUPPORTED_TABLE in codes
msp = doc.modelspace()
polyline = msp.query("POLYLINE").first
assert polyline is not None
def test_readfile_empty_handles_dxf():
doc, auditor = recover.readfile(fullpath(EMPTY_HANDLES))
msp = doc.modelspace()
assert doc.dxfversion == "AC1009"
assert auditor.has_errors is False
assert len(msp.query("LINE")) == 8
assert len(msp.query("*[layer=='GEOMETRY-CUTTING']")) == 4
assert len(msp.query("*[layer=='GEOMETRY-ENGRAVING']")) == 4
|
