(i61ddlZddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z ddl Z ddlmZmZmZmZGddeZGdd ee ZGd d e Zy) N) OrderedDict defaultdict)ListTuple NamedTuple)keccak)Array EIP712Typefrom_solidity_typeBytesJSONEncoderc eZdZdZedZy)OrderedAttributesMetaz=Metaclass to ensure struct attribute order is preserved. ctSN)r)mcsnamebasess K/var/www/br/venv/lib/python3.12/site-packages/poly_eip712_structs/struct.py __prepare__z!OrderedAttributesMeta.__prepare__s }N)__name__ __module__ __qualname____doc__ classmethodrrrrrsrrcpeZdZdZfdZefdZddZdZdZ dZ ede d e fd Z ed Zed Zed efd Zd efdZed eee effdZedZdddd efdZdddd e fdZdddd efdZeded dfdZedZedZdZ dZ!dZ"dZ#dZ$xZ%S) EIP712Structa A representation of an EIP712 struct. Subclass it to use it. Example: from eip712_structs import EIP712Struct, String class MyStruct(EIP712Struct): some_param = String() struct_instance = MyStruct(some_param='some_value') c tt| |jd|j }t |_|D]=\}}|j|}t|t r|di|}||j |<?yNr) superr__init__ type_name get_membersdictvaluesget isinstance)selfkwargsmembersrtypvalue __class__s rr"zEIP712Struct.__init__!ss lD*4>>4@""$f   &ID#JJt$E%& e  %DKK   &rc Ft|di||j|_yr )r!__init_subclass__rr#)clsr*r.s rr0zEIP712Struct.__init_subclass__+s !+F+  rcdt}|jD]\}}t|trAt |t r1|j |}|j|jW|j|j|j|dj|S)zReturns the struct's encoded value. A struct's encoded value is a concatenation of the bytes32 representation of each member of the struct. Order is preserved. :param value: This parameter is not used for structs. r) listr$r(type issubclassrget_data_valueappend hash_struct encode_valuer&join)r)r-encoded_valuesrr, sub_structs rr9zEIP712Struct.encode_value0s))+ KID##t$C)F!006 %%j&<&<&>?%%c&6&6t{{47H&IJ Kxx''rc8|jj|S)z5Get the value of the given struct parameter. )r&r')r)rs rr6zEIP712Struct.get_data_valueCs{{t$$rc@||jvr||j|<yy)z5Set the value of the given struct parameter. N)r&)r)rr-s rset_data_valuezEIP712Struct.set_data_valueHs# 4;;  %DKK  rct}|jjD].\}}t|tr|j ||<*|||<0|S)zProvide the entire data dictionary representing the struct. Nested structs instances are also converted to dict form. )r%r&itemsr(r data_dict)r)resultkvs rrBzEIP712Struct.data_dictNsT KK%%' DAq!\*KKMq q    rresolve_referencesreturncjjDcgc]\}}|jd|}}}jddj|d}|rWt}j |t t fd|Dd}|D]}||jd z }|Scc}}w) N (,)c3.K|] }|k7s |ywrr).0sr1s r z,EIP712Struct._encode_type..cs(Rqc(Rs c|jSr)r#)rOs rz+EIP712Struct._encode_type..cs bcbmbmr)keyF)rF)r$r#r:set_gather_reference_structssortedr3 _encode_type) r1rFrr, member_sigs struct_sigreference_structssorted_structsstructs ` rrWzEIP712Struct._encode_type[sBE//BSTYT3#--$0T T a(='>a@  #   ) )*; <#D(R4E(R$RXmnN( Lf11U1KK  LUsB/c|jDcgc]/}t|dtst|dts+|d1}}|D])}||vs|j ||j |+ycc}w)zbFinds reference structs defined in this struct type, and inserts them into the given set. N)r$r(r4r5raddrU)r1 struct_setmstructsr\s rrUz&EIP712Struct._gather_reference_structshsx"%!2pAj1t6LQ[\]^_\`bnQo1Q4pp =FZ'v&00< =qsA8A8A8c$|jdS)zGet the encoded type signature of the struct. Nested structs are also encoded, and appended in alphabetical order. T)rWr1s r encode_typezEIP712Struct.encode_typers %%rc6t|jS)z1Get the keccak hash of the struct's encoded type.)text)rrerds r type_hashzEIP712Struct.type_hashzs3??,--rcrtdj|j|jgS)zYThe hash of the struct. hash_struct => keccak(type_hash || encode_data) r)rr:rhr9)r)s rr8zEIP712Struct.hash_structs- chh 0$2C2C2EFGHHrc|jjDcgc]=}t|dts&t|dtrt |dt r|?}}|Scc}w)zA list of tuples of supported parameters. Each tuple is (, ). The list's order is determined by definition order. r^)__dict__rAr(r r4r5r)r1rar+s rr$zEIP712Struct.get_memberssc "ll002Sj1z6R!!A$-*QqT<2PSSSsAA#cH|xstj}|s td|S)NzFDomain must be provided, or eip712_structs.default_domain must be set.)poly_eip712_structsdefault_domain ValueError)domainrCs r_assert_domainzEIP712Struct._assert_domains'=.==ef f rrpcV|j|}||h}|j|t}|D]A}|jDcgc]}|d|djd}}|||j<C|j||j |j d}|Scc}w)aConvert a struct into a dictionary suitable for messaging. Dictionary is of the form: { 'primaryType': Name of the primary type, 'types': Definition of each included struct type (including the domain type) 'domain': Values for the domain struct, 'message': Values for the message struct, } :returns: This struct + the domain in dict form, structured as specified for EIP712 messages. rr^)rr4) primaryTypetypesrpmessage)rqrUr%r$r#rB)r)rprbrtr\ra members_jsonrCs r to_messagezEIP712Struct.to_messages$$V,4. &&w/ 3F))+-!!-L-'3E&"" #  3 >>&&(~~'   -sB&cZ|j|}tj|tS)Nrd)rwjsondumpsr )r)rprus rto_message_jsonzEIP712Struct.to_message_jsons"//&)zz''788rcp|j|}d|jz|jz}|S)a^Return a ``bytes`` object suitable for signing, as specified for EIP712. As per the spec, bytes are constructed as follows: ``b'' + domain_hash_bytes + struct_hash_bytes`` :param domain: The domain to include in the hash bytes. If None, uses ``eip712_structs.default_domain`` :return: The bytes object s)rqr8)r)rprCs rsignable_byteszEIP712Struct.signable_bytess:$$V,v1133d6F6F6HH r message_dict StructTuplec t}tt}|dD]b}t|tfi}|d|D]>}|d}t |d}t ||||&||j||df@|||<d|jD]\} } d} || } | D]|\} }tj| |}|jd}||}|jdr-|jdxsd}t | | t||pt | | |~||d }|d }|di|d }|di|d }t|| }|S)aLConvert a message dictionary into two EIP712Struct objects - one for domain, another for the message struct. Returned as a StructTuple, which has the attributes ``message`` and ``domain``. Example: my_msg = { .. } deserialized = EIP712Struct.from_message(my_msg) msg_struct = deserialized.message domain_struct = deserialized.domain :param message_dict: The dictionary, such as what is produced by EIP712Struct.to_message. :return: A StructTuple object, containing the message and domain structs. rtrr4z([a-zA-Z0-9_]+)(\[(\d+)?\])?r^rrs EIP712Domainrurp)rurpr)r%rr3r4rr setattrr7rArematchgroupr r)r1r~rbunfulfilled_struct_paramsr#StructFromJSONmember member_namemember_sol_type struct_nameunfulfilled_member_names regex_pattern struct_classrrbase_type_name ref_structarr_lenprimary_struct domain_structprimary_result domain_resultrCs r from_messagezEIP712Struct.from_messages&$/$5!%g. 0I!)l_bAN&w/ : _$Vn "4VF^"D _E"*-i8??fU[n@]^ _"0GI  0 6O5T5T5V < 1K1;M";/L#; <i :!&Q$^4 ;;q>#kk!n1GL$j'0JKL$ ; < <!m!<=/ 'B,y*AB%? X(>? ^MJ rct|jDchc]}|d }}||vrtd|dycc}w)Nr"z!" is not defined for this struct.)r$KeyError)r1rStup member_namess r_assert_key_is_memberz"EIP712Struct._assert_key_is_membersG*-//*;<3A< < l "Qse#DEF F #=s 5ct|j}||}t|tr^t |t rNt|t r#|j d|j dk7rtdt|d|y |j|y#t$r!}tdt|d|d|d}~wwxYw)z'Eagerly check for a correct member typeFzGiven value is of type z, but we expected zThe python type z/ does not appear to be supported for data type .N) r%r$r(r4r5rrWror9 Exception)r1rSr-r+r,es r_assert_property_typez"EIP712Struct._assert_property_type ss()cl c4 Z\%Be\2e6H6H6OSVScScdiSj6j #:4;-GYZ]Y^!_``7k Q  ' Q #3DK=ABBEa"IJOPQ QsB C )CC cZ|j||jj|Sz:Provide access directly to the underlying value dictionary)rr& __getitem__)r)rSs rrzEIP712Struct.__getitem__s% ""3'{{&&s++rc|j||j|||jj||Sr)rrr& __setitem__)r)rSr-s rrzEIP712Struct.__setitem__"s7 ""3' ""3.{{&&sE22rctd)Nz5Deleting entries from an EIP712Struct is not allowed.) TypeError)r)_s r __delitem__zEIP712Struct.__delitem__)sOPPrc|sy||uryt|tsy|j|jk(xr!|j|jk(S)NFT)r(rrer9)r)others r__eq__zEIP712Struct.__eq__,s[ 5=%.!U%6%6%88hT=N=N=PTYTfTfTh=hhrc|jjDcgc]\}}t|t|z }}}tjt j |t|jScc}}wr)r&rAhash functoolsreduceoperatorxorr#)r)rDrE value_hashess r__hash__zEIP712Struct.__hash__:s[6:kk6G6G6IJdaQ$q')J J lDrsR  0**#]]DfR:)>fRR *r