U ji<@sddlmZddlZddlZddlZddlZddlZejdkrddlZej Z ejdkrpej Z ej Z ej ZejZqej Z ej Z ej ZejZnej Z ej Z ej ZejZej Z ejdkredZnejZddlmZejddd Zed Zejd e d Zejdd GdddejeefZejZejZdddddZddddddZddddddZddd d!d"d#Z d$d%d&d'd(Z!d)d*d+d,d-Z"d)d*d+d.d/Z#ejdkrej$dd0d)d)d+d1d2Z%ej$dd0d)d)d+d3d4Z&n8ej$dd0d)d)d+d5d2Z%ej$dd0d)d)d+d6d4Z&ejdd Gd7d8d8ejeZ'ej(Z(ej)j*Z*ej)j+Z+ej,Z,ej-Z-ej.Z.ej/Z/ej0Z0ej1Z1ej2Z3ej4Z4dS)9) annotationsN) )r r )declarative_asn1TT) covariantUTag)bound)frozenc@s"eZdZUdZded<ded<dS)Varianta A tagged variant for CHOICE fields with the same underlying type. Use this when you have multiple CHOICE alternatives with the same type and need to distinguish between them: foo: ( Annotated[Variant[int, typing.Literal["IntA"]], Implicit(0)] | Annotated[Variant[int, typing.Literal["IntB"]], Implicit(1)] ) Usage: example = Example(foo=Variant(5, "IntA")) decoded.foo.value # The int value decoded.foo.tag # "IntA" or "IntB" r valuestrtagN)__name__ __module__ __qualname____doc____annotations__rrA/tmp/pip-unpacked-wheel-w0_gbkxc/cryptography/hazmat/asn1/asn1.pyr2s rtypebool) field_typereturncCs*ttdrtjtjfntjf}t||kS)N UnionType)hasattrtypesrtypingUnionget_type_origin)rZ union_typesrrr _is_unionMs r$tuplerzdeclarative_asn1.Annotation)metadata field_namercCsd}d}d}|D]}t|tr>|dk r6td|d|j}qt|tjrh|dk rbtd|d|}qt|tjr|dk rtd|d|}qtd|qtj|||dS)Nz-multiple DEFAULT annotations found in field ''z7multiple IMPLICIT/EXPLICIT annotations found in field 'z*multiple SIZE annotations found in field 'zunsupported annotation: )defaultencodingsize) isinstanceDefault TypeErrorrrEncodingSize Annotation)r&r'r)r*r+Zraw_annotationrrr_extract_annotationYs8      r2z typing.Anyzdeclarative_asn1.AnnotatedType)rr'rc st|tkr&t|j}t|^}}nt}|jdk rpt|tj t fkrp|tj tj t ttfkrptdd|tkrt|jtrtddn|jdk rtddt|dr|j}t|tjjtjjfstd|tttj||St|rPt|}t|dkrt |kr|dt!dkr>|d n|d}|tkrXtd t"|}|j#$svtd |jdk rtd tj%|}nt|jtrtd fdd|D} | d j&dk t'fdd| Drtdrdd| D} t| t| krtdtj(| } t |krJtj%t| tn| }nht|tj krt"t|d } tj)| }n8t|t krt"t|d } tj | }n t*|}t||S)Nzfield 'z' has a SIZE annotation, but SIZE annotations are only supported for fields of types: [SEQUENCE OF, SET OF, BIT STRING, OCTET STRING, UTF8String, PrintableString, IA5String]zW' has an IMPLICIT annotation, but IMPLICIT annotations are not supported for TLV types.zT' has a DEFAULT annotation, but DEFAULT annotations are not supported for TLV types. __asn1_root__zunsupported root type: rz=optional TLV types (`TLV | None`) are not currently supportedzoptional (`X | None`) types cannot have `X` annotated: annotations must apply to the union (i.e: `Annotated[X | None, annotation]`)z@optional (`X | None`) types should not have a DEFAULT annotationzCCHOICE (`X | Y | ...`) types should not have an IMPLICIT annotationcs"g|]}|tdk rt|qSN)r_type_to_variant).0arg)r'rr s z)_normalize_field_type..c3s|]}|jdk kVqdSr6tag_namer8v)are_union_types_taggedrr sz(_normalize_field_type..zbWhen using `asn1.Variant` in a union, all the other types in the union must also be `asn1.Variant`cSsh|] }|jqSrr;r=rrr sz(_normalize_field_type..zBWhen using `asn1.Variant` in a union, the tags used must be unique)+r# Annotatedr2 __metadata__ get_type_argsrr1r+builtinslistSetOfbytesr BitString IA5StringPrintableStringr.TLVr,r*Implicitr)rr3TypeSequenceSet AnnotatedTyper!castr$lenNoneTyper_normalize_field_type annotationZis_emptyOptionr<anyZChoiceZ SequenceOfnon_root_python_to_rust) rr'rV_Z root_typeZ union_argsZ optional_typeannotated_typeZrust_field_typevariantstagsZrust_choice_type inner_typer)r?r'rrU|s                          rUzdeclarative_asn1.Variant)tr'rc Cst|tk}|rt|dn|}t|tkrt|\}}t|tjk rNtdt|d}t|drl|j}n t |}|rt |t |j |}nt |t }t t||St |t||dSdS)NrzWhen using `asn1.Variant` in a type annotation, the second type parameter must be a `typing.Literal` type. E.g: `Variant[int, typing.Literal["MyInt"]]`.r3)r#rBrDrr!Literalr.rr3rrYrQr2rCr1rU) r_r'Z is_annotatedr^Z value_typeZ tag_literalr<Z rust_typeZann_typerrrr7 s6       r7zdict[str, type]z)dict[str, declarative_asn1.AnnotatedType]) raw_fieldsrcCs,i}|D]\}}t||}|||<q |Sr6)itemsrU)rafieldsr'rZannotated_field_typerrr_annotate_fields7s   rdztype[U]None)clsrcCs.t|dd}tj|t|}t|d|dSNT)Zinclude_extrasr3)get_type_hintsrrNrOrdsetattrrfrarootrrr_register_asn1_sequenceDs rlcCs.t|dd}tj|t|}t|d|dSrg)rhrrNrPrdrirjrrr_register_asn1_setKs rm)Zkw_only_defaultcCs@tjdkr"tjddddd|}ntjddd|}t||SNrFTrepreqZ match_argsZkw_only)rprq)sys version_info dataclasses dataclassrlrfZ dataclass_clsrrrsequenceZs"   rwcCs@tjdkr"tjddddd|}ntjddd|}t||Srn)rrrsrtrurmrvrrrsetqs"   rxcCs"tjddddd|}t||SNFTro)rtrurlrvrrrrwscCs"tjddddd|}t||Sry)rtrurmrvrrrrxsc@seZdZUded<dS)r-r rN)rrrrrrrrr-s r-)5 __future__rrErtrrr r!rsZtyping_extensionsZ LiteralStringrhget_argsrD get_originr#rBrrTZ"cryptography.hazmat.bindings._rustrTypeVarr r r ruGenericrZ decode_derZ encode_derr$r2rUr7rdrlrmZdataclass_transformrwrxr-rGr/ZExplicitrMr0rKrJZUTCTimeZGeneralizedTimerIZTlvrLZNullrrrrs~         #+