tjIdZddlmZddlZddlZddlmZmZmZej dej Z ej dZ dd Z dd ZddZddZddZddZdS)z@Safe arithmetic calculator for chat (no eval of arbitrary code).) annotationsN)DecimalInvalidOperation localcontextzn^(please\s+)?(can\s+you\s+)?(please\s+)?(calculate|compute|solve|evaluate|simplify|what\s+is|whats|what\'s)\s+u^[\d\s+\-*/().%,^=xX÷×]+$textstrreturnboolct|}t|dkstjd|sdStjd|sdStjd|rdSt t |S)zATrue for calculator-style expressions with no accounting wording.z\dFu[+\-*/÷×xX^=]z [a-wyzA-WYZ])_normalize_exprlenresearchr _PURE_MATH_RE fullmatchrss ./var/www/html/CA-Chatbot/app/rag/calculator.py is_pure_mathrsA 1vvzz5!,,zu 9' + +u y!$$u  ''** + ++c|d}td|}|ddkrU|ds.|ddd}|ddddd d }tjd d|}|d d }|S)Nz?.!=×*÷/^z**z(?<=\d)\s*[xX]\s*(?=\d)%z/100) striprstrip_MATH_PREFIX_REsubcount startswithsplitreplacerrs rr r s E""AB""((**Awws||q!5!5c!:!: GGCOOA  $ $ & & $$$T3//77TBBA )322A #vA 7799rnodeast.ASTrcrt|tjrt|jSt|tjrt|jtr.t|jtst|jSt|jtrtdtdt|tj rft|j }t|jtjr|St|jtjr| Stdt|tjrt|j}t|j}t|jtjr||zSt|jtjr||z St|jtjr||zSt|jtjr|dkrt1d||z St|jtjr|dkrt1d||zSt|jtjr|dkrt1d||zSt|jtjrY||ks t;t|dkrtd|t|zStdtd ) Nz Float literals are not supportedzOnly numbers are allowedzUnsupported unary operatorrzDivision by zeroizExponent too largezUnsupported operatorzUnsupported expression) isinstanceast Expression _eval_nodebodyConstantvalueintr rfloat ValueErrorUnaryOpoperandopUAddUSubBinOpleftrightAddSubMultDivZeroDivisionErrorFloorDivModPowto_integral_valueabs)r*valr=r>s rr0r0(s$''%$)$$$$ %%5 dj# & & 'z$*d/K/K '4:&& & dj% ( ( A?@@ @3444$ $$7&& dgsx ( ( J dgsx ( ( 4K5666$ ""1$)$$4:&& dgsw ' ' %<  dgsw ' ' %<  dgsx ( ( %<  dgsw ' ' zz'(:;;;%<  dgs| , , !zz'(:;;;5= dgsw ' ' zz'(:;;;%<  dgsw ' ' &//1111SU__t5K5K !56663u::% %/000 - . ..rnr4c|dkrdnd}tt|}g}|r)||dd|dd}|)|dt |zS)Nr-r,)rrHappendjoinreversed)rJsigndigitspartss r _comma_intrUVsa%%33RD Q[[FE  VBCC[!!!  #((8E??++ ++rr3c||krtt|St|d}d|vr(|dd}|d}|r |dd}d|vr=|dd\}}tt|pdd|}ntt|pd}|rd|dn|S)u@Always fixed-point / full integer — never scientific notation.f.0rLrN)rGrUr4formatr#r'r(lstrip)r3rnegativewholefracouts r_format_resultr``s ''))))#e**%%% %  D d{{{{3&&s++s##HABBx d{{jja(( tC --..7777T[S))**$, 5 szz# #5rcLt|}|sdSt|dkrdSdtjd|D}|rt |dkr dSt d|rt |nd d z} t j|d }t5}||_t|}d d d n #1swxYwYdt|S#t$rYdSttttt t"f$r YdSwxYw)z?Evaluate a pure arithmetic expression and return a short reply.z@I couldn't read that calculation. Try something like 12 + 5 * 3.iz)That expression is too long to calculate.cRg|]$}t|d%S)r)rgroup).0ms r zcalculate..{s(EEEa#aggajj//EEErz\d+z]One of those numbers is too large for a reliable calculator answer. Try a shorter expression.2(eval)modeNzResult: z;That expression divides by zero, so it can't be calculated.udI couldn't calculate that. Use numbers with + − × ÷ and parentheses, for example (100 + 20) / 4.)r rrfinditermaxr.parserprecr0r`rCr6 SyntaxError TypeErrorr OverflowError MemoryError)rexpr digit_lensrptreectxresults r calculaterzss 4 D RQQ 4yy4::EE2;vt+D+DEEEJ c*oo++ (  rz9C OOOrR? @ @D  yF+++ ^^ &sCH%%F & & & & & & & & & & & & & & &3.00222 MMMLLL  Y0@-Q\ ]    *    s<$C))C  C) CC)CC)) D#6)D#"D#)rrr r )rrr r)r*r+r r)rJr4r r)r3rr r)__doc__ __future__rr.rdecimalrrrcompile IGNORECASEr$rrr r0rUr`rzrrrsFF"""""" ;;;;;;;;;;"*NM  9::  , , , ,+/+/+/+/\,,,,6666&      r