a
#j Q @ s d dl mZ d dlZd dlm mZ d dlZd dlZd dlZd dl m
Z
m Z d dl T d dl
m
Z
d dlmZ d dlZd dlZd dlZddgZeeZejd d krd
d ZeZneZe
d Z g fd
dZ dd Z dd Z G dd de Z!dS ) )print_functionN)NodeIdvalueIsInternalType)*)opaque_type_mapping)CSVBSDTypeParserNodeSet
getSubTypesOf c C s | S N )sr r a/mnt/data/dev_pub/qqyu/v2.12.x/application/opensource/open62541/tools/nodeset_compiler/nodeset.pyunicode! s r z ns=0;i=45c C s ||v r g S t }|| |jD ]}|jtkr"t }|| || |j|jkrz|jrz|t | | j
|j |d q"|j |jkr"|js"|t | | j
|j |d q"|S )N) skipNodes) setadd
references
referenceType
hassubtypeupdatesourceid isForwardr nodestarget)nodesetnoder rerefZskipAllr r r r - s
c C s dg}t j| jdd}d}d}|D ]:}d| v r8d}nd| v rP|| } q^|r"|| }q"t|d krt|d
}|d jD ].}|j |j
krq|j j |v rq|
|j j q| |S ) N http://opcfoundation.org/UA/utf-8)encodingF zTzr Z
NamespaceUris)codecsopennamelowerlendom parseStringgetElementsByTagName
childNodesnodeType ELEMENT_NODE
firstChilddataappendclose)xmlfile
namespacesZinfileZ foundURIsZnslinelinensurir r r extractNamespaces= s,
r8 c C sH i }| j D ]8}|j|jkr
|dr
|d}t|jj}|||< q
|S )zParses the XML Element present in must XML NodeSet definitions.
Contents the Alias element are stored in a dictionary for further
dereferencing during pointer linkage (see linkOpenPointer()).ZAlias)r, r- r. hasAttribute getAttributer r/ r0 )
xmlelementaliasesZalZaliasstZaliasndr r r buildAliasList] s
r= c @ s e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z d0ddZ
d1ddZ dd Z dd Z
dd Zd2dd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/S )3r a This class handles parsing XML description of namespaces, instantiating
nodes, linking references, graphing the namespace and compiling a binary
representation.
Note that nodes assigned to this class are not restricted to having a
single namespace ID. This class represents the entire physical address
space of the binary representation and all nodes that are to be included
in that segment of memory.
c C s i | _ i | _dg| _i | _d S )Nr )r r< r4 namespaceMappingselfr r r __init__u s zNodeSet.__init__c C s | j D ] }| dkr
tdt| q
| j D ]z}|jD ]n}|j|jksdtdt| d |j| j vrtdt| d |j | j vr@t
| j tdt| d q@q6d S )NFzFailed to sanitize node z
Reference z has an invalid sourcez has an unknown reference typez has an unknown target) r valuessanitize Exceptionstrr r r r r printr4 )r@ nr r r r rC { s
zNodeSet.sanitizec C s || j vr| j | d S r )r4 r1 )r@ ZnsURLr r r addNamespace s
zNodeSet.addNamespacec C s. i | _ t|D ]\}}| j|| j |< qdS )zsCreates a dict that maps from the nsindex in the original nodeset to the
nsindex in the combined nodesetN)r> enumerater4 index)r@ orig_namespacesrJ r& r r r createNamespaceMapping s z NodeSet.createNamespaceMappingc s t fdd| j D d S )Nc 3 s | ]} |j jkr|V qd S r )
browseNamer& ).0rG idstringr r z.NodeSet.getNodeByBrowseName..)nextr rB )r@ rP r rO r getNodeByBrowseName s zNodeSet.getNodeByBrowseNamec C s
| dS )NZRoot)rT r? r r r getRoot s zNodeSet.getRootFc C s |j }|d d dkr&|dd }d }|dkr:t|}|dkrJt|}|dkrZt|}|dkrjt|}|dkrzt|}|dkrt|}|d krt|}|d
krt|}|d u rd S ||_ ||_
|S ) N ZuaZvariableobjectmethodZ
objecttypeZ variabletypeZ
methodtypeZdatatypeZ
referencetype) localNamer' VariableNode
ObjectNode
MethodNodeObjectTypeNodeVariableTypeNode DataTypeNodeReferenceTypeNodemodelUrihidden)r@ r; ra rb ndtyper r r r
createNode s2
zNodeSet.createNodeTc C s" || j vrdS | j | }||_dS )NFT)r rb )r@ nodeIdrb r r r r hide_node s
zNodeSet.hide_nodec G s i }|D ]}| | q|S )z
Given any number of dicts, shallow copy and merge into a new dict,
precedence goes to key value pairs in latter dicts.
)r )r@ dict_argsresult
dictionaryr r r merge_dicts s zNodeSet.merge_dictsc C sn t d|}|rN|d}| sN|| jvr0d S | j|}d||d}t|}|| jvrdd S | j| S )Nzns=([^;]+);(.*) zns={};{}rV ) r matchgroupisdigitr4 rJ formatr r )r@ ZidStrmr6 re r r r getNodeByIDString s
zNodeSet.getNodeByIDStringc s fdd j D ]jjkrbj| jvr0qtt fdd| jj j | jj _ qjjkrj| jvr|qtt fdd| jj j | jj _ q| jj= d S )Nc s& | j |j kp$|j jkp"|j jk S r )r r r r )rrt)r r r filterRef s z&NodeSet.remove_node..filterRefc s | d S Nr r rs rt rr r r rR z%NodeSet.remove_node..c s | d S ru r rv rw r r rx rR )r r r r r dictfilteritems)r@ r r )rt r rr r remove_node s"
zNodeSet.remove_nodeUA_TYPESc C s& | }|tjr |tj}tjdkr4|d}t dd|}t
| d}t
|dksjt
|dkrzt| | jd |d }z(| d d d
d }|jd j}W n ty d }Y n0 t|} |d u rt
| dkr| d }|d u rt| | jd | D ]}
| |
q| | |jD ]@} | j| jkr:q$| j } d
| v r$| | jt | | _q$i }
|jD ]} | j| jkrqp| | ||}|sqp| | j | | j ||_!|j"| j#v rtdt$|j" || j#|j"< ||
|j"< qp|
% D ]"}t&|t'r|j(| | j d qd S )Nr
r r! <([/]?)uax:(.+?)([/]?)><\g<1>\g<2>\g<3>> UANodeSetr rk z# contains no or more then 1 nodesetZModelsZModelZModelUrizQ does not define the nodeset URI in Models/Model/ModelUri or NamespaceUris array.r< z XMLElement with duplicate ID )r> ))read
startswithr$ BOM_UTF8lstripsys version_infodecoder subr) r* r+ r( rD Z originXML
attributes nodeValuer8 rH rL r, r- r. rY r' rj r< r= rd ZreplaceAliasesZreplaceNamespacesr>
typesArrayr r rE rB
isinstancer_ Z
buildEncoding)r@ r3 rb r fileContentnodesetsr ZmodelTagra rK r6 ndrc Znewnodesr rG r r r
addNodeSet s\
zNodeSet.addNodeSetc C sd | j | }|jD ]>}|jjdkr|jjdkr| j |j }|jjdkr|j S qtdt | dS )a
The node should have a 'HasEncoding' forward reference which points to the encoding ids.
These can be XML Encoding or Binary Encoding. Therefore we also need to check if the SymbolicName
of the target node is "DefaultBinary"
r & Z
DefaultBinaryz+No DefaultBinary encoding defined for node N)
r r r r6 ir symbolicNamevaluerD rE )r@ re r r ZrefNoder r r getBinaryEncodingIdForNode? s
z"NodeSet.getBinaryEncodingIdForNodec C s( | j D ]}t|tr
|| q
d S r )r rB r rZ Z
allocateValue)r@ rG r r r allocateVariablesM s
zNodeSet.allocateVariablesc C sV |d u r d S |j jtvr |S |jD ].}|jr.q"|jjdkr"| | j|j S q"|S )N- )
rM r& r r r r r getBaseDataTyper r r@ r r r r r r R s
zNodeSet.getBaseDataTypec C s2 |j D ]&}|jjdkr|jr| j|j S qd S )N( )r r r r r r r r r r getNodeTypeDefinition^ s
z NodeSet.getNodeTypeDefinitionc C s t |tr8t|s$td| d S | jt| j| S t |tr|jdkrPd S | j| }t |t s~tdt
| d d S |S d S )Nz Not a valid dataType string: r zNode id z# is not reference a valid dataType.) r string_typesr loggererrorr r r< r r_ rE )r@ dataTypeZ dataTypeNoder r r getDataTypeNodee s
zNodeSet.getDataTypeNodec C s` t }|t| | dg |t| | dg |t| | dg ttdd |S )NHierarchicalReferencesZ HasEncodingZHasTypeDefinitionc S s | j S r r xr r r rx z rR z7NodeSet.getRelevantOrderingReferences..)r r r rT listmap)r@ Zrelevant_typesr r r getRelevantOrderingReferencesu s
z%NodeSet.getRelevantOrderingReferencesc C sN | j D ]>}|j D ].}t|j|j|j|j }d | j |j j|< qq
d S r ) r rB r copyZ Referencer r r r )r@ ur Zbackr r r addInverseReferences| s z NodeSet.addInverseReferencesc C s t | | d}ttdd |}| j D ]f}|jjdkrJ|jjdv rJq,| |}|d ur,| j|j
|_ |j st d
t|j| j|j |_q,d S )Nr c S s | j S r r r r r r rx rR z'NodeSet.setNodeParent..r )N P T z#Node {}: Did not find parent node: )r rT r r r rB r r6 r ZgetParentReferencer parent RuntimeErrorro rE r ZparentReference)r@ Zparentreftypesr Z parentrefr r r
setNodeParent s
zNodeSet.setNodeParentc
C s| g | _ g }g }|D ]p}|jdkr$qt|jd}| }|tjrP|tj}tj dkrd|
d}t dd|}t
|d} t| dkst| d krtd
| d }
|
d } | D ]} | d rt d
| jd js| jd jdks| dr| d st d| jd jr| dd dd }
tjddd}|t|
j j | | j |j | d|j }| | qq|D ]}t|jd}| }|tjr|tj}tj dkr|
d}t dd|}t
|d} t| dkst| d krtd
| d }
|
d } | D ]} | d rdt d
| jd js| jd jdks| dr,| d s,t d| jd jr,| dd dd }
tjdd}|t|
j j | | d t!"|}| j |j t!j#|ddd}d|_$| | q,qg }g }g }g }d }d }|D ]}| | q8t%||||||||| j& | _'| j'( | d S ) Nz)deps/ua-nodeset/DI/Opc.Ua.Di.NodeSet2.xmlrbr~ r! r r r r rk z"contains no or more then 1 nodesetZ
UAVariable SymbolicNamez.*_BinarySchemaZTypeDictionary_BinarySchemaZ ParentNodeIdzi=93Value
ByteStringFz.bsd)deletesuffixz UA_TYPES#)r zUTF-8
)newlinerr Toutname))Z all_filesr& r% r r r$ r r r r r r r r) r* r+ r( rD r9 rl r r fullmatchtempfileZNamedTemporaryFilewritebase64 b64decoder/ flushr1 r2 seekioBufferedReader
TextIOWrappermoder r4 parser create_types)r@ existinginfilesbsdFileZ
import_bsdtype_bsdr3 nodeset_baser r r Z
variableNodesr Z type_contentfZbsdZbftmp
opaque_mapselected_typestype_csvtype_xml
no_builtinr r r r generateParser s
Z
b
zNodeSet.generateParserN)F)T)Fr} )__name__
__module__ __qualname____doc__rA rC rH rL rT rU rd rf rj rq r| r r r r r r r r r r r r r r r j s,
K )"Z
__future__r r xml.dom.minidomr) minidomloggingr$ r datatypesr r r r type_parserr r r r __all__ getLoggerr r r r rE r Z
basestringr r r8 r= rW r r r r r s.