g&dgZGddeZdZy)OrigincReZdZdZdZedZdZdZdZ dZ d dZ d Z d Z y ) raGThis represents the origin of some object in some string. For example, if we have an object ``x1_obj`` that was produced by parsing the ``x1`` in the formula ``"y ~ x1:x2"``, then we conventionally keep track of that relationship by doing:: x1_obj.origin = Origin("y ~ x1:x2", 4, 6) Then later if we run into a problem, we can do:: raise PatsyError("invalid factor", x1_obj) and we'll produce a nice error message like:: PatsyError: invalid factor y ~ x1:x2 ^^ Origins are compared by value, and hashable. c.||_||_||_yNcodestartend)selfrrr s I/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/patsy/origin.py__init__zOrigin.__init__$s  cg}|D]4}|t|ts |j}|$|j|6|syt |Dcgc]}|j c}}t |dk(sJt|Dcgc]}|jc}}t|Dcgc]}|jc}}||j||Scc}wcc}wcc}w)aClass method for combining a set of Origins into one large Origin that spans them. Example usage: if we wanted to represent the origin of the "x1:x2" term, we could do ``Origin.combine([x1_obj, x2_obj])``. Single argument is an iterable, and each element in the iterable should be either: * An Origin object * ``None`` * An object that has a ``.origin`` attribute which fulfills the above criteria. Returns either an Origin object, or None. N) isinstanceroriginappendsetrlenminrmaxr pop)cls origin_objsoriginsobjocodesrr s r combinezOrigin.combine)s$ Cz#v'>jj{ NN3   W-QVV-.5zQg.QWW./'*Q155*+599;s++ ..*sC;CCcH|j|j|jS)zgExtracts and returns the span of the original code represented by this Origin. Example: ``x1``.rr s r relevant_codezOrigin.relevant_codeJsyydhh//r ct|txrO|j|jk(xr4|j|jk(xr|j|jk(Sr)rrrrr r others r __eq__z Origin.__eq__OsR uf % & UZZ' & ekk) &EII%  r c||k( Srr#s r __ne__z Origin.__ne__Ws5=  r cdtt|j|j|jfSr)hashrrrr r s r __hash__zOrigin.__hash__Zs"VTYY DHH=>>r cd|z|jdd|zd|jzd|j|jz zS)a3Produces a user-readable two line string indicating the origin of some code. Example:: y ~ x1:x2 ^^ If optional argument 'indent' is given, then both lines will be indented by this much. The returned string does not have a trailing newline.   ^r)r indents r caretizezOrigin.caretize]sD &L II &L $**  488djj( )   r c d|jd|jd|j|j|jd|j|jdd|jd|jd S)Nzz<-z (-z)>rr s r __repr__zOrigin.__repr__psT IIl # IIdjj488 , IIdhhj ! JJ HH   r ctr)NotImplementedErrorr s r __getstate__zOrigin.__getstate__zs!!r N))__name__ __module__ __qualname____doc__r classmethodrr!r%r(r+r1r4r7r'r r rrsD* ,,@0  !? & "r ctddd}tddd}|jdk(sJ|jdk(sJtj||g}|jdk(sJ|jdk(sJ|j dk(sJ|jddk(sJ|tdddk(sJGd d t }tj|||dg}||k(sJtj|||g}||k(sJtj||gJd d lm}|td d d y)N012345z 012345 ^^z 012345 ^)r0z 012345 ^^^ceZdZddZy)"test_Origin..ObjWithOriginNc||_yr)r)r rs r r z+test_Origin..ObjWithOrigin.__init__s  DKr r)r9r:r;r r'r r ObjWithOriginrDs !r rFr8)assert_no_pickling) rr1rrrr object patsy.utilrG)o1o2o3rFo4o5rGs r test_OriginrP~sE !Q B !Q B ;;=N ** * ;;=O ++ + R !B 77h   88q== 66Q;; ;;a; $7 77 7 !Q' '' '!! r*MOTB CB 8O8 r*B/ 0B 8O8 >>=?MO< = EE E-vb!Q'(r N)__all__rIrrPr'r r rRs" *m"Vm"`)r