AHjt=dZddlmZddlmZmZmZmZmZm Z m Z ddl Z ddl mZddlmZddlmZmZddlmZdd lmZdd lmZmZer dd lmZdd lmZGd deZ Gdde Z!Gdde Z"Gdde Z#Gdde Z$eddZ% edZ& e jNjQdGddee%Z)y)a This module contains the `DOMQuery` class and related objects. A DOMQuery is a set of DOM nodes returned by [query][textual.dom.DOMNode.query]. The set of nodes may be further refined with [filter][textual.css.query.DOMQuery.filter] and [exclude][textual.css.query.DOMQuery.exclude]. Additional methods apply actions to all nodes in the query. !!! info If this sounds like JQuery, a (once) popular JS library, it is no coincidence. ) annotations) TYPE_CHECKINGGenericIterableIteratorTypeVarcastoverloadN) active_app) AwaitRemove)DeclarationError TokenErrormatch) SelectorSet)parse_declarationsparse_selectors)DOMNodeWidgetceZdZdZy) QueryErrorz%Base class for a query related error.N__name__ __module__ __qualname____doc__I/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/css/query.pyrr s/rrceZdZdZy)InvalidQueryFormatzQuery did not parse correctly.Nrrrr r"r"$s(rr"ceZdZdZy) NoMatcheszNo nodes matched the query.Nrrrr r$r$(s%rr$ceZdZdZy)TooManyMatchesz!Too many nodes matched the query.Nrrrr r&r&,s+rr&ceZdZdZy) WrongTypez)Query result was not of the correct type.Nrrrr r(r(0s3rr( QueryTyper)bound ExpectTypeT)angularcHeZdZgdZddddd d+dZed,dZed-dZd.dZd/d Z d0d Z d0d Z e re d1d Ze d2d Zd3dZd4dZd5dZd5dZe re d6dZe d7dZ d8 d9dZe re d6dZe d7dZ d8 d9dZe re d6dZe d7dZ d8 d9dZe re d0dZe d:dZ d8 d;dZdd Zd>d!Zd>d"Zd?d#Z d8 d@d$Zdd%d%d& dAd'ZdBd(ZdBd)Z dC dDd*Z y)EDOMQuery)_node_nodes_filters _excludes_deepNT)filterexcludedeepparentcd}||_d|_|r|jjng|_|r|jjng|_||_|% |jj t||& |jj t|yy#t$rtd|dwxYw#t$rtd|dwxYw)aEInitialize a query object. !!! warning You won't need to construct this manually, as `DOMQuery` objects are returned by [query][textual.dom.DOMNode.query]. Args: node: A DOM node. filter: Query to filter children in the node. exclude: Query to exclude children in the node. deep: Query should be deep, i.e. recursive. parent: The parent query, if this is the result of filtering another query. Raises: InvalidQueryFormat: If the format of the query is invalid. TNzUnable to parse filter z as query) r/r0r1copyr2r3appendrrr")selfnoder4r5r6r7_rich_traceback_omits r __init__zDOMQuery.__init__>s2 $ .2 &,FOO "" (.F   ! ! #2    X $$_V%<=   X%%og&>?   X(+B6*I)VWW X X(+B6*I)VWW Xs!$B.$C .C C#c|jS)zThe node being queried.)r/r;s r r<z DOMQuery.nodenszzrcddlm}|jt|jr|j j |n|j j}|Dcgc]"tfd|jDr$}}|Dcgc]"tfd|jDs$}}td||_|jScc}wcc}w)zLazily evaluate nodes.rrc36K|]}t|ywNr.0 selector_setr<s r z!DOMQuery.nodes..sS\u\40Sc36K|]}t|ywrCrrDs r rGz!DOMQuery.nodes..sX5t4XrHlist[QueryType]) textual.widgetrr0listr3r/ walk_childrenallr1anyr2r )r;r initial_nodesr<nodess ` r rQzDOMQuery.nodesss * ;;  48JJ ((0DJJDUDUM *ST]]SSE"XXXE 0%8DK{{ s 'C 'Cc,t|jSrC)lenrQr@s r __len__zDOMQuery.__len__s4::rc,t|jS)z#True if non-empty, otherwise False.)boolrQr@s r __bool__zDOMQuery.__bool__sDJJrc,t|jSrC)iterrQr@s r __iter__zDOMQuery.__iter__sDJJrc,t|jSrC)reversedrQr@s r __reversed__zDOMQuery.__reversed__s ##rcyrCrr;indexs r __getitem__zDOMQuery.__getitem__s8;rcyrCrr_s r razDOMQuery.__getitem__s@Crc |j|SrC)rQr_s r razDOMQuery.__getitem__szz%  rc#K |jr&ddjd|jDf|jr'ddjd|jDfyy#t$rYywxYww)Nqueryz AND c3LK|]}djd|Dyw),c34K|]}|jywrCcssrEselectors r rGz3DOMQuery.__rich_repr__... H( HNjoinrE selectorss r rGz)DOMQuery.__rich_repr__..s'!% Hi HH!"$r5z OR c3LK|]}djd|Dyw)rgc34K|]}|jywrCrirks r rGz3DOMQuery.__rich_repr__...rmrnNrorqs r rGz)DOMQuery.__rich_repr__..s' % Hi HH rs)r1rpr2AttributeErrorr@s r __rich_repr__zDOMQuery.__rich_repr__s }}LL!)-!~~KK )-    s)A9A$A*(A9* A63A95A66A9cHt|j||j|S)zFilter this set by the given CSS selector. Args: selector: A CSS selector. Returns: New DOM Query. )r4r6r7r.r<r3r;rls r r4zDOMQuery.filters% II   rcHt|j||j|S)zExclude nodes that match a given selector. Args: selector: A CSS selector. Returns: New DOM query. )r5r6r7ryrzs r r5zDOMQuery.excludes% II   rcyrCrr@s r firstzDOMQuery.firsts&)rcyrCrr; expect_types r r}zDOMQuery.firstsFIrcd}|jr:|jd}|'t||std|jd||St d|d|j )aWGet the *first* matching node. Args: expect_type: Require matched node is of this type, or None for any type. Raises: WrongType: If the wrong type was found. NoMatches: If there are no matching nodes in the query. Returns: The matching Widget. Tr-Query value is the wrong type; expected type , found No nodes match z on )rQ isinstancer(rr$r<)r;rr=r}s r r}zDOMQuery.firstsx $ ::JJqME&!%5#G H\H\G__ghmgnoLodXT$))GH HrcyrCrr@s r only_onezDOMQuery.only_ones),rcyrCrrs r rzDOMQuery.only_onesILrcd}||j|n|j} |jd}td#t$rY|SwxYw)aGet the *only* matching node. Args: expect_type: Require matched node is of this type, or None for any type. Raises: WrongType: If the wrong type was found. NoMatches: If no node matches the query. TooManyMatches: If there is more than one matching node in the query. Returns: The matching Widget. Tz7Call to only_one resulted in more than one matched node)r}rQr& IndexError)r;rr=the_one_s r rzDOMQuery.only_onesi" $(3'>DJJ{ #DJJL    1 A I     sA A AcyrCrr@s r lastz DOMQuery.last,s%(rcyrCrrs r rz DOMQuery.last/sEHrc|jstd|d|j|jd}|'t||st d|j d||S)aVGet the *last* matching node. Args: expect_type: Require matched node is of this type, or None for any type. Raises: WrongType: If the wrong type was found. NoMatches: If there are no matching nodes in the query. Returns: The matching Widget. rz on domrr)rQr$r<rr(r)r;rrs r rz DOMQuery.last2sq zzodXWTYYMJK Kzz"~  ":dK+H? @T@T?WW_`d_ef  rcyrCrr@s r resultszDOMQuery.resultsMs25rcyrCr)r; filter_types r rzDOMQuery.resultsPsRUrc#ZK| |Ed{y|D]}t||s|y7w)aGet query results, optionally filtered by a given type. Args: filter_type: A Widget class to filter results, or None for no filter. Yields: Iterator[Widget | ExpectType]: An iterator of Widget instances. N)r)r;rr<s r rzDOMQuery.resultsSs8  OO dK0J  s+)++c:|D]}|j|g||S)zSet the given class name(s) according to a condition. Args: add: Add the classes if True, otherwise remove them. Returns: Self. ) set_class)r;add class_namesr<s r rzDOMQuery.set_classfs+ .D DNN3 - - . rct|tr|D]}|j||St|}|D]}|j||S)zSet the classes on nodes to exactly the given set. Args: classes: A string of space separated classes, or an iterable of class names. Returns: Self. )rstr set_classesrL)r;classesr<rs r rzDOMQuery.set_classesss_ gs # *  ) *  w-K .  - . rc2|D]}|j||S)z%Add the given class name(s) to nodes.) add_classr;rr<s r rzDOMQuery.add_classs$ )D DNNK ( ) rc2|D]}|j||S)z,Remove the given class names from the nodes.) remove_classrs r rzDOMQuery.remove_class& ,D D  { + , rc2|D]}|j||S)z0Toggle the given class names from matched nodes.) toggle_classrs r rzDOMQuery.toggle_classrrcvtj}|j|jd|jiS)zRemove matched nodes from the DOM. Returns: An awaitable object that waits for the widgets to be removed. r7)r get_prunerQr/)r;apps r removezDOMQuery.removes. nnszz4::9djj99rc >d}|D]}|jdi||B t|d}|D]/}|j j||jd1|S#t$r1}t|j|j|j dd}~wwxYw)zjSet styles on matched nodes. Args: css: CSS declarations to parser, or None. TN) set_styles) read_from)layoutr) rrr nametokenmessage_inline_stylesmergerefresh)r;rj update_stylesr=r< new_styleserrors r rzDOMQuery.set_styless $ -D DOO ,m , - ? Y/?QR  *##))*5 D ) * $ Y&uzz5;; NTXX Ys A"" B+,BBFrepaintr recomposec<|D]}|j||||S)zRefresh matched nodes. Args: repaint: Repaint node(s). layout: Layout node(s). recompose: Recompose node(s). Returns: Query for chaining. r)r)r;rrrr<s r rzDOMQuery.refreshs, ND LL9L M N rcZ|D]%}|js|j|S|S)zlFocus the first matching node that permits focus. Returns: Query for chaining. ) allow_focusfocus)r;r<s r rzDOMQuery.focuss7  D!    rc|jjj}|6t|}||vr'|jjj |||S)zhBlur the first matching node that is focused. Returns: Query for chaining. )avoiding)r/screenfocusedrL _reset_focus)r;rrQs r blurz DOMQuery.blursR **##++  "&t*E% !!..w.G rc^|D]'}|||_|||_|||_|!||_)|S)aSets common attributes on matched nodes. Args: display: Set `display` attribute on nodes, or `None` for no change. visible: Set `visible` attribute on nodes, or `None` for no change. disabled: Set `disabled` attribute on nodes, or `None` for no change. loading: Set `loading` attribute on nodes, or `None` for no change. Returns: Query for chaining. )displayvisibledisabledloading)r;rrrrr<s r setz DOMQuery.setsN$ 'D"& "& # ( "&  ' r) r<rr4 str | Noner5rr6rVr7zDOMQuery | NonereturnNone)rr)rrJ)rint)rrV)rzIterator[QueryType])r`rrr))r`slicerrJ)r`z int | slicerzQueryType | list[QueryType])rzrich.repr.Result)rlrrDOMQuery[QueryType])rr))rtype[ExpectType]rr+rC)rtype[ExpectType] | NonerzQueryType | ExpectType)rrrzIterator[ExpectType])rrrz Iterator[QueryType | ExpectType])rrVrrrr)rzstr | Iterable[str]rr)rrrr)rr )rjrrr)rrVrrVrrVrr)rr)NNNN) r bool | Nonerrrrrrrr)!rrr __slots__r>propertyr<rQrTrWrZr]rr rarwr4r5r}rrrrrrrrrrrrrrrrr r.r.:spEI """&.X.X .X  .X  .X .X .X`*  $ ; ; C C!* "  ) ) I I6:I2I I8 , , L L6:%2% %N ( ( H H6:2 2 5 5 U U6:2 )& &   :!% ."&eu/3HL "   $# $#      rr.)*r __future__rtypingrrrrrr r rich.reprrichtextual._contextr textual.await_remover textual.css.errorsr rtextual.css.matchrtextual.css.modelrtextual.css.parserr textual.domrrKr Exceptionrr"r$r&r(r)r+reprautor.rrr rs #VVV',;#)A#%00))& &,Z,4 4 Kx 0 1 \ " 5Awy!AAr