o i@sddlmZddlZddlZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZd8d9ddZd:ddZd;d d!Zdd0d1Zd2Zd3Zd?d6d7ZdS)@) annotationsN)List)proto)_normalize_value)InArgs) InStatement)InValue) LibsqlError) Statement) ResultSet)Row)Valuein_stmtr in_argsrreturn proto.StmtcCsbt||}g}g}|jdurnt|jtr!dd|jD}ndd|jD}|j||ddS)NcSsg|] \}}|t|dqS))namevalue_value_to_proto).0keyrrO/home/ubuntu/.local/lib/python3.10/site-packages/libsql_client/hrana/convert.py s z"_stmt_to_proto..cSsg|]}t|qSrr)rrrrrrsT)sqlargs named_args want_rows)r convertr isinstancedictitemsr)rrstmtrrrrr_stmt_to_protos   r& proto_resproto.StmtResultr c Cstdd|dD}ddt|D}g}|dD]}tdd|D}|t||q|d}|d }|dur?t|nd}t||||S) Ncss|] }|dp dVqdS)rNr)r proto_colrrr $sz)_result_set_from_proto..colscSsi|]\}}||qSrr)ridxcolumnrrr %sz*_result_set_from_proto..rowscss|]}t|VqdSN)_value_from_proto)r proto_valrrrr+(saffected_row_countlast_insert_rowid)tuple enumerateappendrgetintr ) r'columns column_idxsr0 proto_rowvalues rows_affectedlast_insert_rowid_strr5rrr_result_set_from_proto#s  rAin_stmtsList[InStatement] proto.BatchcCsg}|ddddi|D]}|dt|ddt|dq|dt|ddd ddd|d dt|ddd d dddd |iS)Nr%BEGINF)rr okr)typestep) conditionr%COMMITnot)rGcondROLLBACKsteps)r8lenr&)rBrNrrrr_batch_to_proto2s<       rPproto.BatchResult stmt_countr:List[ResultSet]cCst|d|dkrtddt|d|dkrtdd|dD] }|dur,t|q"g}|dddD]}|durBtd d|t|q7|S) N step_resultsz:Server did not return the expected number of batch resultsHRANA_PROTO_ERROR step_errorsz9Server did not return the expected number of batch errorsrz'Server did not return a result in batch)rOr _error_from_protor8rA)r'rR proto_err result_setsstmt_resrrr_batch_results_from_proto^s, r]rZ proto.Errorr cCs |d}|dp d}t||S)NmessagecodeUNKNOWN)r9r )rZr_r`rrrrYzs rYin_valuer proto.ValuecCst|}|dur ddiSt|trd|dSt|tr.|tks#|tkr'tddt|dSt|trAt |sInteger exceeds the range of SQLite integers (64 bits, signed)integerz6Only finite floats (not Infinity or NaN) are supportedfloatblob)rGbase64zUnsupported value of type )rr"strr: _MIN_INTEGER _MAX_INTEGER OverflowErrorrgmathisfinite ValueErrorri b64encodedecode TypeErrorrG)rbrdatarrrrs*        rllrrcCs|ddkrdS|ddkrt|dS|ddkr t|dS|ddkr,t|dS|ddkr;t|dd Std |dd ) NrGrdrerrfrgrhriz====zUnknown value type rV)rjr:rgri b64decoder )rrrrr2s        r2r1)rr rrrr)r'r(rr )rBrCrrD)r'rQrRr:rrS)rZr^rr )rbr rrc)rrcrr) __future__rrirntypingrr)rclientrrr r r r resultr rrr&rArPr]rYrrkrlr2rrrrs.                ,