Completed
Push — master ( 9a97a0...4f1130 )
by Olivier
04:03
created

XmlImporter.add_object()   A

Complexity

Conditions 2

Size

Total Lines 10

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 8
CRAP Score 2

Importance

Changes 0
Metric Value
cc 2
dl 0
loc 10
ccs 8
cts 8
cp 1
crap 2
rs 9.4285
c 0
b 0
f 0
1
"""
2
add node defined in XML to address space
3
format is the one from opc-ua specification
4
"""
5 1
import logging
6
7
8 1
from opcua import ua
9 1
from opcua.common import xmlparser
10
11
12 1
class XmlImporter(object):
13
14 1
    def __init__(self, server):
15 1
        self.logger = logging.getLogger(__name__)
16 1
        self.parser = None
17 1
        self.server = server
18
19 1
    def import_xml(self, xmlpath):
20 1
        self.logger.info("Importing XML file %s", xmlpath)
21 1
        self.parser = xmlparser.XMLParser(xmlpath)
22 1
        for node in self.parser:
23 1
            if node.nodetype == 'UAObject':
24 1
                self.add_object(node)
25 1
            elif node.nodetype == 'UAObjectType':
26
                self.add_object_type(node)
27 1
            elif node.nodetype == 'UAVariable':
28 1
                self.add_variable(node)
29
            elif node.nodetype == 'UAVariableType':
30
                self.add_variable_type(node)
31
            elif node.nodetype == 'UAReferenceType':
32
                self.add_reference(node)
33
            elif node.nodetype == 'UADataType':
34
                self.add_datatype(node)
35
            elif node.nodetype == 'UAMethod':
36
                self.add_method(node)
37
            else:
38
                self.logger.info("Not implemented node type: %s ", node.nodetype)
39
40 1
    def _get_node(self, obj):
41 1
        node = ua.AddNodesItem()
42 1
        node.RequestedNewNodeId = ua.NodeId.from_string(obj.nodeid)
43 1
        node.BrowseName = ua.QualifiedName.from_string(obj.browsename)
44 1
        node.NodeClass = getattr(ua.NodeClass, obj.nodetype[2:])
45 1
        if obj.parent:
46 1
            node.ParentNodeId = ua.NodeId.from_string(obj.parent)
47 1
        if obj.parentlink:
48 1
            node.ReferenceTypeId = self.to_nodeid(obj.parentlink)
49 1
        if obj.typedef:
50 1
            node.TypeDefinition = ua.NodeId.from_string(obj.typedef)
51 1
        return node
52
53 1
    def to_nodeid(self, nodeid):
54 1
        if not nodeid:
55
            return ua.NodeId(ua.ObjectIds.String)
56 1
        elif "=" in nodeid:
57
            return ua.NodeId.from_string(nodeid)
58
        elif hasattr(ua.ObjectIds, nodeid):
59 1
            return ua.NodeId(getattr(ua.ObjectIds, nodeid))
60
        else:
61 1
            if nodeid in self.parser.aliases:
62 1
                nodeid = self.parser.aliases[nodeid]
63 1
            else:
64
                nodeid = "i={}".format(getattr(ua.ObjectIds, nodeid))
65
            return ua.NodeId.from_string(nodeid)                
66 1
        
67
    def add_object(self, obj):
68 1
        node = self._get_node(obj)
69
        attrs = ua.ObjectAttributes()
70 1
        if obj.desc:
71 1
            attrs.Description = ua.LocalizedText(obj.desc)
72 1
        attrs.DisplayName = ua.LocalizedText(obj.displayname)
73 1
        attrs.EventNotifier = obj.eventnotifier
74 1
        node.NodeAttributes = attrs
75 1
        self.server.add_nodes([node])
76 1
        self._add_refs(obj)
77 1
78 1
    def add_object_type(self, obj):
79 1
        node = self._get_node(obj)
80
        attrs = ua.ObjectTypeAttributes()
81 1
        if obj.desc:
82
            attrs.Description = ua.LocalizedText(obj.desc)
83
        attrs.DisplayName = ua.LocalizedText(obj.displayname)
84
        attrs.IsAbstract = obj.abstract
85
        node.NodeAttributes = attrs
86
        self.server.add_nodes([node])
87
        self._add_refs(obj)
88
89
    def add_variable(self, obj):
90
        node = self._get_node(obj)
91
        attrs = ua.VariableAttributes()
92 1
        if obj.desc:
93 1
            attrs.Description = ua.LocalizedText(obj.desc)
94 1
        attrs.DisplayName = ua.LocalizedText(obj.displayname)
95 1
        attrs.DataType = self.to_nodeid(obj.datatype)
96
        # if obj.value and len(obj.value) == 1:
97 1
        if obj.value is not None:
98 1
            if obj.valuetype == 'ListOfLocalizedText':
99
                attrs.Value = ua.Variant([ua.LocalizedText(txt) for txt in obj.value], None)
100 1
            else:
101 1
                attrs.Value = ua.Variant(obj.value, getattr(ua.VariantType, obj.valuetype))
102 1
        if obj.rank:
103 1
            attrs.ValueRank = obj.rank
104 1
        if obj.accesslevel:
105
            attrs.AccessLevel = obj.accesslevel
106 1
        if obj.useraccesslevel:
107
            attrs.UserAccessLevel = obj.useraccesslevel
108 1
        if obj.minsample:
109
            attrs.MinimumSamplingInterval = obj.minsample
110 1
        if obj.dimensions:
111
            attrs.ArrayDimensions = obj.dimensions
112 1
        node.NodeAttributes = attrs
113 1 View Code Duplication
        self.server.add_nodes([node])
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
114 1
        self._add_refs(obj)
115
116 1
    def add_variable_type(self, obj):
117
        node = self._get_node(obj)
118
        attrs = ua.VariableTypeAttributes()
119
        if obj.desc:
120
            attrs.Description = ua.LocalizedText(obj.desc)
121
        attrs.DisplayName = ua.LocalizedText(obj.displayname)
122
        attrs.DataType = self.to_nodeid(obj.datatype)
123
        if obj.value and len(obj.value) == 1:
124
            attrs.Value = obj.value[0]
125
        if obj.rank:
126
            attrs.ValueRank = obj.rank
127
        if obj.abstract:
128
            attrs.IsAbstract = obj.abstract
129
        if obj.dimensions:
130
            attrs.ArrayDimensions = obj.dimensions
131
        node.NodeAttributes = attrs
132 View Code Duplication
        self.server.add_nodes([node])
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
133
        self._add_refs(obj)
134
135 1
    def add_method(self, obj):
136
        node = self._get_node(obj)
137
        attrs = ua.MethodAttributes()
138
        if obj.desc:
139
            attrs.Description = ua.LocalizedText(obj.desc)
140
        attrs.DisplayName = ua.LocalizedText(obj.displayname)
141
        if obj.accesslevel:
142
            attrs.AccessLevel = obj.accesslevel
143
        if obj.useraccesslevel:
144
            attrs.UserAccessLevel = obj.useraccesslevel
145
        if obj.minsample:
146
            attrs.MinimumSamplingInterval = obj.minsample
147
        if obj.dimensions:
148
            attrs.ArrayDimensions = obj.dimensions
149
        node.NodeAttributes = attrs
150
        self.server.add_nodes([node])
151
        self._add_refs(obj)
152
153 1
    def add_reference(self, obj):
154
        node = self._get_node(obj)
155
        attrs = ua.ReferenceTypeAttributes()
156
        if obj.desc:
157
            attrs.Description = ua.LocalizedText(obj.desc)
158
        attrs.DisplayName = ua.LocalizedText(obj.displayname)
159
        if obj. inversename:
160
            attrs.InverseName = ua.LocalizedText(obj.inversename)
161
        if obj.abstract:
162
            attrs.IsAbstract = obj.abstract
163
        if obj.symmetric:
164
            attrs.Symmetric = obj.symmetric
165
        node.NodeAttributes = attrs
166
        self.server.add_nodes([node])
167
        self._add_refs(obj)
168
169 1
    def add_datatype(self, obj):
170
        node = self._get_node(obj)
171
        attrs = ua.DataTypeAttributes()
172
        if obj.desc:
173
            attrs.Description = ua.LocalizedText(obj.desc)
174
        attrs.DisplayName = ua.LocalizedText(obj.displayname)
175
        if obj.abstract:
176
            attrs.IsAbstract = obj.abstract
177
        node.NodeAttributes = attrs
178
        self.server.add_nodes([node])
179
        self._add_refs(obj)
180
181 1
    def _add_refs(self, obj):
182 1
        if not obj.refs:
183 1
            return
184
        refs = []
185
        for data in obj.refs:
186
            ref = ua.AddReferencesItem()
187
            ref.IsForward = True
188
            ref.ReferenceTypeId = self.to_nodeid(data.reftype)
189
            ref.SourceNodeId = ua.NodeId.from_string(obj.nodeid)
190
            ref.TargetNodeClass = ua.NodeClass.DataType
191
            ref.TargetNodeId = ua.NodeId.from_string(data.target)
192
            refs.append(ref)
193
        self.server.add_references(refs)
194