a GGb @sUdZddlmZddlZddlmZmZddlmZddl m Z m Z ddl m mmZee e ZGdddeZd d d d d ZGdddejdZGdddeZGdddeZeedZded<dS)z) Engine classes for :func:`~pandas.eval` ) annotationsN) align_termsreconstruct_object)Expr)MATHOPS REDUCTIONSc@s eZdZdS)NumExprClobberingErrorN)__name__ __module__ __qualname__r r >lib/python3.9/site-packages/pandas/core/computation/engines.pyrsrrNone)exprreturncCs@|j}|t@}|r(z+_check_ne_builtin_clash..zVariables in expression "z" overlap with builtins: ()N)names _ne_builtinsjoinr)rrZoverlapsr r r _check_ne_builtin_clashs rc@sXeZdZdZdZddZddddZd dd d Zed dd dZ e j ddZ dS)AbstractEnginez/Object serving as a base class for all engines.FcCs||_d|_d|_dSN)r aligned_axes result_type)selfrr r r __init__3szAbstractEngine.__init__strrcCs t|jS)zo Convert an expression for evaluation. Defaults to return the expression as a string. )printingZ pprint_thingrr r r r convert8szAbstractEngine.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_alignedrrZtermsrr _evaluaterZ return_type)r resr r r evaluate@s zAbstractEngine.evaluateboolcCs|jduo|jduSr)rrr%r r r r(UszAbstractEngine._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. Nr r%r r r r)YszAbstractEngine._evaluateN) r r r __doc__ has_neg_fracr!r&r+propertyr(abcabstractmethodr)r r r r r.sr) metaclassc@seZdZdZdZddZdS) NumExprEnginezNumExpr engine classTcCs6ddl}|}|jj}|j}t|j|j||dS)Nr)Z local_dict)numexprr&renvZ full_scoperr+)r nerr5Zscoper r r r)ps  zNumExprEngine._evaluateN)r r r r-r.r)r r r r r3ksr3c@s*eZdZdZdZddZddddZd S) PythonEnginezS Evaluate an expression in Python space. Mostly for testing purposes. FcCs|Sr)rr%r r r r+szPythonEngine.evaluaterr#cCsdSrr r%r r r r)szPythonEngine._evaluateN)r r r r-r.r+r)r r r r r7|sr7)r4Zpythonzdict[str, type[AbstractEngine]]ENGINES)r-Z __future__rr0Zpandas.core.computation.alignrrZpandas.core.computation.exprrZpandas.core.computation.opsrrZpandas.io.formats.printingioZformatsr$ frozensetr NameErrorrrABCMetarr3r7r8__annotations__r r r r s   =