r5i~dZddlmZddlZddlZddlmZddlmZm Z Gdde Z Gd d Z Gd d Z y) z4KPI calculation engine with safe formula evaluation.) annotationsN)Any)KPITreeKPINodeceZdZdZy) FormulaErrorz'Error in formula parsing or evaluation.N)__name__ __module__ __qualname____doc__g/var/www/tkim.planitai.co.jp/gemegg/20251207-make-pdf-report/project/planitai-kpi/src/kpi/calculator.pyr r s1rr c  eZdZdZej ej ejejejejejejejejf Zd dZd dZd dZd dZd dZddZy) FormulaParserzSafe formula parser using AST. Only allows basic arithmetic operations: +, -, *, / Variables are referenced as {variable_name} c8tjd|_y)zInitialize parser.z \{(\w+)\}N)recompile_variable_pattern)selfs r__init__zFormulaParser.__init__'s!#L!9rc8|jj|S)zExtract variable names from formula. Args: formula: Formula string with {variable} syntax. Returns: List of variable names. )rfindall)rformulas rparse_variableszFormulaParser.parse_variables+s%%--g66rc|}|j|D]6}||vrtd||jd|dt||}8|S)aAReplace variables with values. Args: formula: Formula string with {variable} syntax. values: Dictionary mapping variable names to values. Returns: Formula with variables replaced by values. Raises: FormulaError: If a variable is missing. zMissing variable: {})rr replacestr)rrvaluesresultvars rsubstitute_variablesz"FormulaParser.substitute_variables6sh''0 DC& "%7u#=>>^^bRL#fSk2BCF D  rc|j||} tj|d}|j ||j |jS#t$r}t d|d}~wwxYw)a?Evaluate formula with given values. Args: formula: Formula string with {variable} syntax. values: Dictionary mapping variable names to values. Returns: Calculated result. Raises: FormulaError: If formula is invalid or evaluation fails. eval)modezInvalid formula syntax: N)r%astparse SyntaxErrorr _validate_ast _eval_nodebody)rrr" expressiontreees revaluatezFormulaParser.evaluateLsu..w?  ?99Zf5D 4 tyy)) ?!9!=> > ?sA A3 A..A3ctj|D]:}t||jrt dt |j dy)zValidate that AST only contains allowed node types. Args: tree: AST to validate. Raises: FormulaError: If tree contains disallowed nodes. zInvalid expression: z is not allowedN)r)walk isinstance ALLOWED_NODESr typer )rr0nodes rr,zFormulaParser._validate_asthsOHHTN DdD$6$67"*4:+>+>*?O rct|tjrt|jSt|tj r|j |j}|j |j}t|jtjr||zSt|jtjr||z St|jtjr||zSt|jtjr |dk(ry||z St|tjrh|j |j}t|jtj r| St|jtj"r|St%dt'|j()zEvaluate a single AST node. Args: node: AST node to evaluate. Returns: Evaluated value. Raises: FormulaError: If node type is not supported. rzUnsupported node type: )r5r)ConstantfloatvalueBinOpr-leftrightopAddSubMultDivUnaryOpoperandUSubUAddr r7r )rr8r?r@rGs rr-zFormulaParser._eval_nodews; dCLL )$ $ dCII &??499-DOODJJ/E$''377+e|#$''377+e|#$''388,e|#$''377+A:e|# dCKK (oodll3G$''388,x$''388,4T$Z5H5H4IJKKrN)returnNone)rr!rJ list[str])rr!r"dict[str, float]rJr!)rr!r"rMrJr<)r0ast.ASTrJrK)r8rNrJr<)r r r r r) Expressionr>rFr;rBrCrDrErHrIr6rrr%r2r,r-rrrrrsu           M: 7,*8 %Lrrc8eZdZdZddZd dZd dZd dZd dZy) KPICalculatorzKPI calculation engine.c0||_t|_y)zZInitialize with KPI tree. Args: tree: KPI tree to calculate. N)r0rparser)rr0s rrzKPICalculator.__init__s  #o rc|j}i}|D]|}|jj|}|jr4 |jj |j|}||_|||<\|j |j ||<xd||<~|S#t$r }td|d|d||<Yd}~d}~wwxYw)zCalculate all KPIs in the tree. Uses topological sort to ensure dependencies are calculated first. Returns: Dictionary mapping node ID to calculated value. zWarning: Failed to calculate z: r:N) _get_calculation_orderr0nodesrrSr2r=r print)rorderresultsnode_idr8r=r1s r calculatezKPICalculator.calculates++-$& 'G99??7+D||+ KK00wGE!&DJ',GG$ '#':: $' % '($+9'"QCHI'*GG$+s2B B?B::B?clgtdfd |jjS)zGet nodes in topological order for calculation. Returns nodes from leaves to root, ensuring all dependencies are calculated before dependents. Returns: List of node IDs in calculation order. c|jvryj|j|jD] }| j|jy)N)idaddchildrenappend)r8childrXvisitvisiteds rrcz3KPICalculator._get_calculation_order..visitsPww'! KK  e   LL !r)r8rrJrK)setr0root)rrXrcrds @@@rrUz$KPICalculator._get_calculation_orders-E " diinn rc|jjjDcic]}|j|j}}|j D]<\}}||jjvs||jj|_>|j }|j D]#\}}||jj|_%|Scc}w)zSimulate KPI changes without modifying the tree. Args: changes: Dictionary of node ID to new value. Returns: Dictionary mapping node ID to simulated value. )r0rVr"r^r=itemsr[)rchangesn saved_valuesrZr=rYs rsimulatezKPICalculator.simulates04yy/E/E/GH!agg H H&mmo 7NGU$))//)16 (. 7 .."+002 3NGU-2DIIOOG $ * 3IsCc|jj|}|r |jsgS|jj |jS)zGet all dependencies for a node's formula. Args: node_id: Node ID to check. Returns: List of node IDs that this node depends on. )r0get_noderrSr)rrZr8s rget_formula_dependenciesz&KPICalculator.get_formula_dependenciess?yy!!'*4<<I{{**4<<88rN)r0r)rJrM)rJrL)rirMrJrM)rZr!rJrL) r r r r rr[rUrlrorrrrQrQs!&!F44 9rrQ)r __future__rr)rtypingrr0rr Exceptionr rrQrrrrss>:" " 9 JLJLZp9p9r