a Of @sUdZddlmZddlZddlmZddlmZddlm Z m Z ddl m Z m Z ddlmZerndd lmZee e Zd d d d dZGdddejdZGdddeZGdddeZeedZded<dS)z) Engine classes for :func:`~pandas.eval` ) annotationsN) TYPE_CHECKING)NumExprClobberingError) align_termsreconstruct_object)MATHOPS REDUCTIONS)printing)Exprr None)exprreturncCs@|j}|t@}|r)z+_check_ne_builtin_clash..zVariables in expression "z" overlap with builtins: ()N)names _ne_builtinsjoinr)r rZoverlapsrrr_check_ne_builtin_clashs rc@s^eZdZdZdZddddZdddd Zd dd d Zed dddZ e j ddZ dS)AbstractEnginez/Object serving as a base class for all engines.Fr r cCs||_d|_d|_dSN)r aligned_axes result_type)selfr rrr__init__4szAbstractEngine.__init__strcCs t|jS)zo Convert an expression for evaluation. Defaults to return the expression as a string. )r Z pprint_thingr r rrrconvert9szAbstractEngine.convertobjectcCs:|jst|jj\|_|_|}t|j||j|jjjS)a* Run the engine on the expression. This method performs alignment which is necessary no matter what engine is being used, thus its implementation is in the base class. Returns ------- object The result of the passed expression. ) _is_alignedrr Ztermsrr _evaluater return_type)r resrrrevaluateAs zAbstractEngine.evaluateboolcCs|jduo|jduSr)rrr#rrrr&VszAbstractEngine._is_alignedcCsdS)a Return an evaluated expression. Parameters ---------- env : Scope The local and global environment in which to evaluate an expression. Notes ----- Must be implemented by subclasses. Nrr#rrrr'ZszAbstractEngine._evaluateN) __name__ __module__ __qualname____doc__ has_neg_fracr!r$r*propertyr&abcabstractmethodr'rrrrr/sr) metaclassc@seZdZdZdZddZdS) NumExprEnginezNumExpr engine classTcCs6ddl}|}|jj}|j}t|j|j||dS)Nr)Z local_dict)numexprr$r envZ full_scoperr*)r nerr7scoperrrr'ps  zNumExprEngine._evaluateN)r,r-r.r/r0r'rrrrr5ksr5c@s*eZdZdZdZddZddddZd S) PythonEnginezS Evaluate an expression in Python space. Mostly for testing purposes. FcCs|Sr)r r#rrrr*szPythonEngine.evaluater rcCsdSrrr#rrrr'szPythonEngine._evaluateN)r,r-r.r/r0r*r'rrrrr:|sr:)r6pythonzdict[str, type[AbstractEngine]]ENGINES)r/ __future__rr2typingrZ pandas.errorsrZpandas.core.computation.alignrrZpandas.core.computation.opsrrZpandas.io.formatsr Zpandas.core.computation.exprr frozensetrrABCMetarr5r:r<__annotations__rrrrs"      <