U Sc?@sddlZddlmZddlmZddlmZddddgZzeWnek r\d d ZYnXGd dde Z Gd dde Z Gd dde Z Gddde Z dS)N)error)tag)tagmap NamedTypeOptionalNamedTypeDefaultedNamedType NamedTypescCsttt|SN)boolfilter)xr 9/tmp/pip-unpacked-wheel-hsjy238j/pyasn1/type/namedtype.pyrc@seZdZdZdZdZd$ddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZeddZeddZeddZd d!Zd"d#ZdS)%raCreate named field object for a constructed ASN.1 type. The |NamedType| object represents a single name and ASN.1 type of a constructed ASN.1 type. |NamedType| objects are immutable and duck-type Python :class:`tuple` objects holding *name* and *asn1Object* components. Parameters ---------- name: :py:class:`str` Field name asn1Object: ASN.1 type object FNcCs ||_||_||f|_||_dSr )_NamedType__name_NamedType__type_NamedType__nameAndType_NamedType__openType)selfname asn1ObjectopenTyper r r__init__*s zNamedType.__init__cCs4d|j|jf}|jr$|d|j7}d|jj|fS)Nz%s=%rz, open type %rz<%s object, type %s>)rrr __class____name__rZrepresentationr r r__repr__0szNamedType.__repr__cCs |j|kSr rrotherr r r__eq__9szNamedType.__eq__cCs |j|kSr rrr r r__ne__<szNamedType.__ne__cCs |j|kSr rrr r r__lt__?szNamedType.__lt__cCs |j|kSr rrr r r__le__BszNamedType.__le__cCs |j|kSr rrr r r__gt__EszNamedType.__gt__cCs |j|kSr rrr r r__ge__HszNamedType.__ge__cCs t|jSr )hashrrr r r__hash__KszNamedType.__hash__cCs |j|Sr rridxr r r __getitem__NszNamedType.__getitem__cCs t|jSr )iterrr(r r r__iter__QszNamedType.__iter__cCs|jSr )rr(r r rrTszNamedType.namecCs|jSr )rr(r r rrXszNamedType.asn1ObjectcCs|jSr )rr(r r rr\szNamedType.openTypecCs|jSr rr(r r rgetNamebszNamedType.getNamecCs|jSr rr(r r rgetTypeeszNamedType.getType)N)r __module__ __qualname____doc__ isOptional isDefaultedrrr!r"r#r$r%r&r)r,r.propertyrrrr0r2r r r rrs,     c@seZdZejZdZdS)rTN)rr3r4rr5r6r r r rrisc@seZdZejZdZdS)rTN)rr3r4rr5r7r r r rrosc@steZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZejddkrddZnddZd d!Zd"d#Zd$d%Zd&d'Zd(d)ZGd*d+d+eZd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Z d8d9Z!d:d;Z"dd?Z$e%d@dAZ&dBdCZ'e%dDdEZ(e%dFdGZ)e%dHdIZ*e%dJdKZ+e%dLdMZ,e%dNdOZ-dPS)Qra+Create a collection of named fields for a constructed ASN.1 type. The NamedTypes object represents a collection of named fields of a constructed ASN.1 type. *NamedTypes* objects are immutable and duck-type Python :class:`dict` objects holding *name* as keys and ASN.1 type object as values. Parameters ---------- *namedTypes: :class:`~pyasn1.type.namedtype.NamedType` Examples -------- .. code-block:: python class Description(Sequence): ''' ASN.1 specification: Description ::= SEQUENCE { surname IA5String, first-name IA5String OPTIONAL, age INTEGER DEFAULT 40 } ''' componentType = NamedTypes( NamedType('surname', IA5String()), OptionalNamedType('first-name', IA5String()), DefaultedNamedType('age', Integer(40)) ) descr = Description() descr['surname'] = 'Smith' descr['first-name'] = 'John' cOs||_t|j|_||_||_||_d|kr@| pBi|_ |j dd|_ |j dd|_ tdd|jD|_tdd|jD|_tddt|jD|_td d|jD|_td d|jD|_td d|jD|_dS) NterminalT)uniqueFcSsg|]}|js|jrdqST)r7r6.0 namedTyper r r sz'NamedTypes.__init__..cSsg|]}|jrdqSr;)rr<r r rr?scSs g|]\}}|js|js|qSr )r6r7)r=r+ntr r rr?scSsg|] }|jqSr r/r<r r rr?scSsg|] }|jqSr r1r<r r rr?scSsg|]}|j|jfqSr )rrr<r r rr?s)_NamedTypes__namedTypeslen_NamedTypes__namedTypesLen_NamedTypes__computeMinTagSet_NamedTypes__minTagSet _NamedTypes__computeNameToPosMap_NamedTypes__nameToPosMap_NamedTypes__computeTagToPosMap_NamedTypes__tagToPosMap"_NamedTypes__computeAmbiguousTypes_NamedTypes__ambiguousTypes_NamedTypes__computeTagMaps_NamedTypes__uniqueTagMap_NamedTypes__nonUniqueTagMapany!_NamedTypes__hasOptionalOrDefault_NamedTypes__hasOpenTypes frozenset enumerate_NamedTypes__requiredComponents_NamedTypes__keystuple_NamedTypes__values_NamedTypes__items)r namedTypeskwargsr r rrs     zNamedTypes.__init__cCs&ddd|jD}d|jj|fS)Nz, cSsg|] }d|qS)z%rr r=r r r rr?sz'NamedTypes.__repr__..z<%s object, types %s>)joinrArrrr r rrs zNamedTypes.__repr__cCs |j|kSr rArr r rr!szNamedTypes.__eq__cCs |j|kSr r]rr r rr"szNamedTypes.__ne__cCs |j|kSr r]rr r rr#szNamedTypes.__lt__cCs |j|kSr r]rr r rr$szNamedTypes.__le__cCs |j|kSr r]rr r rr%szNamedTypes.__gt__cCs |j|kSr r]rr r rr&szNamedTypes.__ge__cCs t|jSr )r'rAr(r r rr)szNamedTypes.__hash__cCs6z |j|WStk r0|j|j|YSXdSr )rA TypeErrorrGr*r r rr,s zNamedTypes.__getitem__cCs ||jkSr )rG)rkeyr r r __contains__szNamedTypes.__contains__cCsdd|jDS)Ncss|]}|dVqdS)rNr r[r r r sz&NamedTypes.__iter__..r]r(r r rr.szNamedTypes.__iter__rcCs |jdkSNrrCr(r r r __nonzero__szNamedTypes.__nonzero__cCs |jdkSrcrdr(r r r__bool__szNamedTypes.__bool__cCs|jSr rdr(r r r__len__szNamedTypes.__len__cCs|jSr )rWr(r r rvaluesszNamedTypes.valuescCs|jSr )rUr(r r rkeysszNamedTypes.keyscCs|jSr )rXr(r r ritemsszNamedTypes.itemscCs |j|jSr )rrAr(r r rcloneszNamedTypes.clonec@seZdZddZddZdS)zNamedTypes.PostponedErrorcCs ||_dSr )_PostponedError__errorMsg)rZerrorMsgr r rrsz"NamedTypes.PostponedError.__init__cCst|jdSr )r PyAsn1Errorrl)ritemr r rr,sz%NamedTypes.PostponedError.__getitem__N)rr3r4rr,r r r rPostponedErrorsrocCsti}t|jD]`\}}|jj}t|tjr2|S|s8q|jD].}||krdtd||fS|||<q>q|S)Nz Duplicate component tag %s at %s)rSrArtagMap isinstancerro presentTypes)rZ tagToPosMapr+r>rpZ_tagSetr r rZ__computeTagToPosMaps   zNamedTypes.__computeTagToPosMapcCsHi}t|jD]4\}}|j|kr8td|j|fS|||j<q|S)Nz!Duplicate component name %s at %s)rSrArrro)rZ nameToPosMapr+r>r r rZ__computeNameToPosMaps   z NamedTypes.__computeNameToPosMapcCsvi}d}ttt|jD]V\}}|js.|jr:|f|}n|f}t|t|jkr\|||<qt|tdd||<q|S)Nr T)r9) reversedrVrSrAr6r7rBrdict)rZambiguousTypesZpartialAmbiguousTypesr+r>r r rZ__computeAmbiguousTypes s   z"NamedTypes.__computeAmbiguousTypescCs2z|j|jWStk r,tdYnXdS)aTReturn ASN.1 type object by its position in fields set. Parameters ---------- idx: :py:class:`int` Field index Returns ------- : ASN.1 type Raises ------ ~pyasn1.error.PyAsn1Error If given position is out of fields range Type position out of rangeN)rAr IndexErrorrrmr*r r rgetTypeByPositionszNamedTypes.getTypeByPositioncCs6z |j|WStk r0td|fYnXdS)aReturn field position by its ASN.1 type. Parameters ---------- tagSet: :class:`~pysnmp.type.tag.TagSet` ASN.1 tag set distinguishing one ASN.1 type from others. Returns ------- : :py:class:`int` ASN.1 type position in fields set Raises ------ ~pyasn1.error.PyAsn1Error If *tagSet* is not present or ASN.1 types are not unique within callee *NamedTypes* zType %s not foundN)rIKeyErrorrrm)rtagSetr r rgetPositionByType3s zNamedTypes.getPositionByTypecCs2z|j|jWStk r,tdYnXdS)anReturn field name by its position in fields set. Parameters ---------- idx: :py:class:`idx` Field index Returns ------- : :py:class:`str` Field name Raises ------ ~pyasn1.error.PyAsn1Error If given field name is not present in callee *NamedTypes* ruN)rArrvrrmr*r r rgetNameByPositionKszNamedTypes.getNameByPositioncCs6z |j|WStk r0td|fYnXdS)a|Return field position by filed name. Parameters ---------- name: :py:class:`str` Field name Returns ------- : :py:class:`int` Field position in fields set Raises ------ ~pyasn1.error.PyAsn1Error If *name* is not present or not unique within callee *NamedTypes* zName %s not foundN)rGrxrrm)rrr r rgetPositionByNamecs zNamedTypes.getPositionByNamecCs2z|j|jWStk r,tdYnXdS)aReturn ASN.1 types that are allowed at or past given field position. Some ASN.1 serialisation allow for skipping optional and defaulted fields. Some constructed ASN.1 types allow reordering of the fields. When recovering such objects it may be important to know which types can possibly be present at any given position in the field sets. Parameters ---------- idx: :py:class:`int` Field index Returns ------- : :class:`~pyasn1.type.tagmap.TagMap` Map if ASN.1 types allowed at given field position Raises ------ ~pyasn1.error.PyAsn1Error If given position is out of fields range ruN)rKrprxrrmr*r r rgetTagMapNearPosition{sz NamedTypes.getTagMapNearPositioncCs:z||j||WStk r4tdYnXdS)aReturn the closest field position where given ASN.1 type is allowed. Some ASN.1 serialisation allow for skipping optional and defaulted fields. Some constructed ASN.1 types allow reordering of the fields. When recovering such objects it may be important to know at which field position, in field set, given *tagSet* is allowed at or past *idx* position. Parameters ---------- tagSet: :class:`~pyasn1.type.tag.TagSet` ASN.1 type which field position to look up idx: :py:class:`int` Field position at or past which to perform ASN.1 type look up Returns ------- : :py:class:`int` Field position in fields set Raises ------ ~pyasn1.error.PyAsn1Error If *tagSet* is not present or not unique within callee *NamedTypes* or *idx* is out of fields range ruN)rKrzrxrrm)rryr+r r rgetPositionNearTypeszNamedTypes.getPositionNearTypec Cs\d}|jD]D}|j}z |j}Wntk r8|j}YnX|dksJ||kr |}q |pZtSr )rAr minTagSetAttributeErrorryrZTagSet)rrr>rryr r rZ__computeMinTagSets   zNamedTypes.__computeMinTagSetcCs|jS)aReturn the minimal TagSet among ASN.1 type in callee *NamedTypes*. Some ASN.1 types/serialisation protocols require ASN.1 types to be arranged based on their numerical tag value. The *minTagSet* property returns that. Returns ------- : :class:`~pyasn1.type.tagset.TagSet` Minimal TagSet among ASN.1 types in callee *NamedTypes* )rEr(r r rrs zNamedTypes.minTagSetcCsi}i}d}|jD]}|jj}t|tjr2|S|D]6}|rb||krbtd|||fS|j||<q6||j|dkr|j}q|jdk rtd|fSqt |||S)Nz Non-unique tagSet %s of %s at %sz"Duplicate default ASN.1 type at %s) rArrprqrroupdate skipTypes defaultTyperZTagMap)rr:rrrrr>rpryr r rZ__computeTagMapss"      zNamedTypes.__computeTagMapscCs|jS)aReturn a *TagMap* object from tags and types recursively. Return a :class:`~pyasn1.type.tagmap.TagMap` object by combining tags from *TagMap* objects of children types and associating them with their immediate child type. Example ------- .. code-block:: python OuterType ::= CHOICE { innerType INTEGER } Calling *.tagMap* on *OuterType* will yield a map like this: .. code-block:: python Integer.tagSet -> Choice )rNr(r r rrpszNamedTypes.tagMapcCs|jS)aReturn a *TagMap* object from unique tags and types recursively. Return a :class:`~pyasn1.type.tagmap.TagMap` object by combining tags from *TagMap* objects of children types and associating them with their immediate child type. Example ------- .. code-block:: python OuterType ::= CHOICE { innerType INTEGER } Calling *.tagMapUnique* on *OuterType* will yield a map like this: .. code-block:: python Integer.tagSet -> Choice Note ---- Duplicate *TagSet* objects found in the tree of children types would cause error. )rMr(r r r tagMapUniqueszNamedTypes.tagMapUniquecCs|jSr )rPr(r r rhasOptionalOrDefault#szNamedTypes.hasOptionalOrDefaultcCs|jSr )rQr(r r r hasOpenTypes'szNamedTypes.hasOpenTypescCs t|jSr )rVrAr(r r rrY+szNamedTypes.namedTypescCs|jSr )rTr(r r rrequiredComponents/szNamedTypes.requiredComponentsN).rr3r4r5rrr!r"r#r$r%r&r)r,r`r.sys version_inforerfrgrhrirjrkobjectrorHrFrJrwrzr{r|r}r~rDr8rrLrprrrrYrr r r rrus^$  !      )rZpyasn1rZ pyasn1.typerr__all__rO NameErrorrrrrrr r r rs   R