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
|
# This script is dedicated to the public domain under the terms of the CC0 license.
from collections import OrderedDict
import os
import sys
import re
from typing import Optional
class MaterialProfilesValidator:
def __init__(self, root_dir: str) -> None:
self._repo_dir = os.path.abspath(root_dir)
self._materials_dir = self._repo_dir
self._guid_pattern = re.compile(r"<GUID>.*</GUID>")
def _get_guid(self, content: str) -> Optional[str]:
guid = None
for line in content.splitlines():
line = line.strip()
if self._guid_pattern.match(line):
guid = line.strip("<GUID>").strip("</GUID>")
break
return guid
def get_materials_dir(self, dirpath: str) -> str:
for root_dir, dirnames, filenames in os.walk(dirpath):
has_materials_file = any(fn.endswith(".xml.fdm_material") for fn in filenames)
if not has_materials_file:
for dirname in dirnames:
full_dir_path = os.path.join(root_dir, dirname)
return self.get_materials_dir(full_dir_path)
return dirpath
## Validates the preset settings files and returns ``True`` or ``False``
# indicating whether there are invalid files.
def validate(self) -> bool:
# parse the definition file
guid_dict = OrderedDict()
materials_dir = self.get_materials_dir(self._materials_dir)
# go through all the preset settings files
for _, _, filenames in os.walk(materials_dir):
for filename in filenames:
file_path = os.path.join(materials_dir, filename)
if not filename.endswith(".xml.fdm_material"):
continue
with open(file_path, "r", encoding = "utf-8") as f:
content = f.read()
guid = self._get_guid(content)
if guid not in guid_dict:
guid_dict[guid] = []
item_list = guid_dict[guid]
item_list.append({"file_name": filename,
"file_path": file_path})
break
has_invalid_files = False
for guid, file_item_list in guid_dict.items():
if len(file_item_list) <= 1:
continue
has_invalid_files = True
if guid is not None:
print("-> The following files contain the same GUID [%s]:" % guid)
else:
print("-> The following files DO NOT contain any GUID:")
for file_item in file_item_list:
print(" -- [%s]" % file_item["file_name"])
print("-> PLEASE make sure to generate unique GUIDs for each material.")
return not has_invalid_files
if __name__ == "__main__":
script_dir = os.path.dirname(os.path.realpath(__file__))
root_dir = os.path.abspath(os.path.join(script_dir, ".."))
validator = MaterialProfilesValidator(root_dir)
is_everything_validate = validator.validate()
ret_code = 0 if is_everything_validate else 1
sys.exit(ret_code)
|
