Added support for custom datatypes and removed some trailing spaces

opcua/common/manage_nodes.py

@@ -37,7 +37,7 @@ def create_folder(parent, *args):
     or namespace index, name
     """
     nodeid, qname = _parse_add_args(*args)
-    return node.Node(parent.server, _create_object(parent.server, parent.nodeid, nodeid, qname, ua.ObjectIds.FolderType))
+    return node.Node(parent.server, _create_object(parent.server, parent.nodeid, nodeid, qname, ua.ObjectIds.FolderType, ua.NodeClass.Object))
 
 
 def create_object(parent, *args):
@@ -63,7 +63,7 @@ def create_object(parent, *args):
             raise
         except Exception as ex:
             raise TypeError("This provided objecttype takes either a index, nodeid or string. Received arguments {} and got exception {}".format(args, ex))
-    return node.Node(parent.server, _create_object(parent.server, parent.nodeid, nodeid, qname, objecttype))
+    return node.Node(parent.server, _create_object(parent.server, parent.nodeid, nodeid, qname, objecttype, ua.NodeClass.Object))
 
 
 def create_property(parent, *args):
@@ -116,19 +116,23 @@ def create_subtype(parent, *args):
     arguments are nodeid, browsename
     or namespace index, name
     """
-    nodeid, qname = _parse_add_args(*args)
-    return node.Node(parent.server, _create_object(parent.server, parent.nodeid, nodeid, qname, None))
+    nodeid, qname = _parse_add_args(*args[:2])
+    if len(args) > 3:
+        node_class = args[3]
+    else:
+        node_class = ua.NodeClass.ObjectType
+    return node.Node(parent.server, _create_object(parent.server, parent.nodeid, nodeid, qname, None, node_class))
 
 
-def _create_object(server, parentnodeid, nodeid, qname, objecttype):
+def _create_object(server, parentnodeid, nodeid, qname, objecttype, node_class):
     addnode = ua.AddNodesItem()
     addnode.RequestedNewNodeId = nodeid
     addnode.BrowseName = qname
     addnode.ParentNodeId = parentnodeid
+    addnode.NodeClass = node_class
     #TODO: maybe move to address_space.py and implement for all node types?
     if not objecttype:
         addnode.ReferenceTypeId = ua.NodeId(ua.ObjectIds.HasSubtype)
-        addnode.NodeClass = ua.NodeClass.ObjectType
         attrs = ua.ObjectTypeAttributes()
         attrs.IsAbstract = True
     else:
@@ -137,7 +141,6 @@ def _create_object(server, parentnodeid, nodeid, qname, objecttype):
         else:
             addnode.ReferenceTypeId = ua.NodeId(ua.ObjectIds.HasComponent)
 
-        addnode.NodeClass = ua.NodeClass.Object
         if isinstance(objecttype, int):
             addnode.TypeDefinition = ua.NodeId(objecttype)
         elif isinstance(objecttype, ua.NodeId):

tests/tests_common.py

@@ -486,13 +486,13 @@ class CommonTests(object):
         o = f.add_object(3, 'MyObject1', ua.ObjectIds.BaseObjectType)
         self.assertEqual(o.get_type_definition(), ua.ObjectIds.BaseObjectType)
 
-        o = f.add_object(3, 'MyObject2', ua.NodeId(ua.ObjectIds.BaseObjectType, 0) )
+        o = f.add_object(3, 'MyObject2', ua.NodeId(ua.ObjectIds.BaseObjectType, 0))
         self.assertEqual(o.get_type_definition(), ua.ObjectIds.BaseObjectType)
 
         base_otype= self.opc.get_node(ua.ObjectIds.BaseObjectType)
         custom_otype = base_otype.add_subtype(2, 'MyFooObjectType')
 
-        o = f.add_object(3, 'MyObject3', custom_otype.nodeid )
+        o = f.add_object(3, 'MyObject3', custom_otype.nodeid)
         self.assertEqual(o.get_type_definition(), custom_otype.nodeid.Identifier)
 
         references = o.get_references(refs=ua.ObjectIds.HasTypeDefinition, direction=ua.BrowseDirection.Forward)