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
|
from typing import List
from xsdata.codegen.mixins import HandlerInterface
from xsdata.codegen.models import Attr, Class
from xsdata.codegen.utils import ClassUtils
from xsdata.utils import collections
class MergeAttributes(HandlerInterface):
"""Merge same type attributes and their restrictions."""
__slots__ = ()
@classmethod
def process(cls, target: Class):
"""
Detect and process duplicate attributes.
- Remove duplicates for enumerations.
- Merge duplicates with restrictions and types.
"""
if target.is_enumeration:
cls.filter_duplicate_attrs(target)
else:
cls.merge_duplicate_attrs(target)
@classmethod
def filter_duplicate_attrs(cls, target: Class):
attrs = collections.unique_sequence(target.attrs, key="default")
target.attrs = attrs
@classmethod
def merge_duplicate_attrs(self, target: Class):
result: List[Attr] = []
for attr in target.attrs:
pos = collections.find(result, attr)
existing = result[pos] if pos > -1 else None
if not existing:
result.append(attr)
elif not (attr.is_attribute or attr.is_enumeration):
existing.help = existing.help or attr.help
e_res = existing.restrictions
a_res = attr.restrictions
min_occurs = e_res.min_occurs or 0
max_occurs = e_res.max_occurs or 1
attr_min_occurs = a_res.min_occurs or 0
attr_max_occurs = a_res.max_occurs or 1
e_res.min_occurs = min(min_occurs, attr_min_occurs)
e_res.max_occurs = max_occurs + attr_max_occurs
if a_res.sequence is not None:
e_res.sequence = a_res.sequence
existing.fixed = False
existing.types.extend(attr.types)
target.attrs = result
ClassUtils.cleanup_class(target)
|
