AHjZ(ddlmZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZmZmZmZddlmZmZmZddlmZe rdd lmZGd d eZy) ) annotationsN)ABCabstractmethod)contextmanager)Path) TYPE_CHECKINGAnyBinaryIOIteratorLiteralTextIO)eventslogmessages)MouseUp)AppceZdZdZdddd ddZeddZeddZedd Zedd Z d d Z d d Z e d!d Z d"dZe d"dZe d"dZe d"dZd"dZd"dZGddej*Zed#dZd"dZd$d%dZddddd d&dZ d'dZ d(dZy))DriverzA base class for drivers.FTN)debugmousesizec||_||_||_||_t j |_g|_d|_d|_ d|_ y)zInitialize a driver. Args: app: The App instance. debug: Enable debug mode. mouse: Enable mouse support, size: Initial size of the terminal or `None` to detect. NT) _app_debug_mouse_sizeasyncioget_running_loop_loop _down_buttons_last_move_event _auto_restart cursor_origin)selfapprrrs F/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/driver.py__init__zDriver.__init__sS     --/ (*9=!F59cy)z%Is the driver 'headless' (no output)?Fr$s r& is_headlesszDriver.is_headless/r(cy)z)Is the driver 'inline' (not full-screen)?Fr*r+s r& is_inlinezDriver.is_inline4r-r(cy)z,Is the driver 'web' (running via a browser)?Fr*r+s r&is_webz Driver.is_web9r-r(cy)zCan this driver be suspended?Fr*r+s r& can_suspendzDriver.can_suspend>r-r(cxtj|jj||jy)zYSend a message to the target app. Args: message: A message. )loopN)rrun_coroutine_threadsafer _post_messager)r$messages r& send_messagezDriver.send_messageCs* (( II # #G ,4:: r(c4|j|j|jd}d}n|j\}}t|tj rT|xj |zc_|xj|zc_|xj|zc_|xj|zc_ t|tjr4|jr|jj|jnt|tjrM|jri|j|jvrP|jj|jn)t|tj r|jr|js|j"t%t&j)|jj+}|jj-|j"}|D]}|j/t|j0|j2|j4dd||j6|j8|j:|j<|j>|j@ ||_|j/|y)z~Perform additional processing on a message, prior to sending. Args: event: A message to process. Nr) xydelta_xdelta_ybuttonshiftmetactrlscreen_xscreen_ystyle)! set_senderrr# isinstancer MouseEvent_x_y _screen_x _screen_y MouseDownr?r appendrremove MouseMover!listdictfromkeyskeysclearr9widgetr;r<r@rArBrCrDrE)r$r8offset_xoffset_ybuttons move_eventr?s r&process_messagezDriver.process_messageMs 499%    %HH!%!3!3 Hh gv00 1 JJ( "J JJ( "J    )     )  gv// 0~~""))'..9  0~~'..D4F4F"F""))'..9 !1!1 2""))5t}}T-?-?@EEGH""((*!22 %F%%#NN(ll(ll$%$%#)")--!(!(%/%8%8%/%8%8")-- "%,D ! '"r(cy)zTWrite data to the output device. Args: data: Raw data. Nr*)r$datas r&writez Driver.writer(cy)zFlush any buffered data.Nr*r+s r&flushz Driver.flushr_r(cy)zStart application mode.Nr*r+s r&start_application_modezDriver.start_application_moder_r(cy)zDisable further input.Nr*r+s r& disable_inputzDriver.disable_inputr_r(cy)z%Stop application mode, restore state.Nr*r+s r&stop_application_modezDriver.stop_application_moder_r(cD|j|jy)zSuspend application mode. Used to suspend application mode and allow uninhibited access to the terminal. N)rgcloser+s r&suspend_application_modezDriver.suspend_application_modes ""$ r(c$|jy)zzResume application mode. Used to resume application mode after it has been previously suspended. N)rcr+s r&resume_application_modezDriver.resume_application_modes ##%r(ceZdZdZy)Driver.SignalResumez?Event sent to the app when a resume signal should be published.N)__name__ __module__ __qualname____doc__r*r(r& SignalResumernsMr(rsc#bK|j}d|_ d||_y#||_wxYww)aA context manager used to tell the driver to not auto-restart. For drivers that support the application being suspended by the operating system, this context manager is used to mark a body of code as one that will manage its own stop and start. FN)r")r$ auto_restarts r&no_automatic_restartzDriver.no_automatic_restarts3)) " . !-D D s/#/ ,/cy)zPerform any final cleanup.Nr*r+s r&riz Driver.closer_r(c.ddl}|j|y)a0Open a URL in the default web browser. Args: url: The URL to open. new_tab: Whether to open the URL in a new tab. This is only relevant when running via the WebDriver, and is ignored when called while running through the terminal. rN) webbrowseropen)r$urlnew_tabrys r&open_urlzDriver.open_urls r(download) open_methodencoding mime_typenamecdfd }t|trd} nd} tj||| f} | j y)aSave the file `path_or_file` to `save_path`. If running via web through Textual Web or Textual Serve, this will initiate a download in the web browser. Args: binary: The binary file to save. delivery_key: The unique key that was used to deliver the file. save_path: The location to save the file to. open_method: *web only* Whether to open the file in the browser or to prompt the user to download it. When running via a standard (non-web) terminal, this is ignored. encoding: *web only* The text encoding to use when saving the file. This will be passed to Python's `open()` built-in function. When running via web, this will be used to set the charset in the `Content-Type` header. mime_type: *web only* The MIME type of the file. This will be used to set the `Content-Type` header in the HTTP response. name: A user-defined named which will be returned in [`DeliveryComplete`][textual.events.DeliveryComplete] and [`DeliveryComplete`][textual.events.DeliveryComplete]. c@ t | xsd5}|j}|j}d} ||}|s j n ||( ddd|js|jyy#1swY'xYw#t$rf}t j d|ddl}t j t|j j | Yd}~d}~wwxYw#|js|jwwxYw)Nzutf-8)ri) save_pathrzFailed to deliver file: r) exceptionr) rzreadr^_delivery_complete Exceptionrerror tracebackstr format_exc_delivery_failedclosedri)binarymodedestination_filerr^ chunk_sizer]rr delivery_keyrrrr$s r&save_file_threadz/Driver.deliver_binary..save_file_threads #th.A'$%!;;D,22E!*J#J/# 33 , 4"d "$0}}LLN%1$$ P 4UG<=  #i22456%%le$%OO P}}LLN%sBB ABB B B C;AC61C>6C;;C>>Dwbw)targetargsN)rBinaryIO | TextIOrrreturnNone)rGr threadingThreadstart) r$rrrrrrrrrthreads ` `` ` ` r&deliver_binaryzDriver.deliver_binarysDD # #: fh 'DD!!)9O r(c|jj|jjtj|||y)zsCalled when a file has been delivered successfully. Delivers a DeliveryComplete event to the app. )keypathrN)rcall_from_thread post_messagerDeliveryComplete)r$rrrs r&rzDriver._delivery_completes5 "" II " "  # # 94 P r(c|jj|jjtj|||y)z`Called when a file delivery fails. Delivers a DeliveryFailed event to the app. )rrrN)rrrrDeliveryFailed)r$rrrs r&rzDriver._delivery_failed#s5 "" II " "  ! !lid S r() r%zApp[Any]rboolrrrztuple[int, int] | Nonerr)rr)r8zmessages.Messagerr)r]rrr)rr)rzIterator[None])T)r{rr|rrr)rrrrrrrzLiteral['browser', 'download']r str | Nonerrrrrr)rrrz Path | Nonerrrr)rrr BaseExceptionrrrr)rorprqrrr'propertyr,r/r1r3r9r[rr^rarcrergrjrlrEventrsrrvrir}rrrr*r(r&rrs# '+ : : :  : % : :6 7#r  '&&%%44&Nv||N . .) &7A# $E!E E  E 4 EEEE EN    ,7  ?I        ,9  AK    r(r) __future__rrrabcrr contextlibrpathlibrtypingrr r r r r textualrrrtextual.eventsr textual.apprrr*r(r&rs:"#%JJ))"\ S\ r(