tjmUddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd l mZdd lmZdd lmZej4eZedZedZedZda de!d<ddZ"GddZ#y)) annotationsN)Literal)utils) Controller) HttpHeader) CoreServer)State)VirtualNodeManager)share)ArgumentParser) Translator)vue2vue3)aiohttpgenerictornadojupyter)maindesktoptask coroutiner ClientTypeDEFAULT_CLIENT_TYPEc|ayN)r)values >/DATA/.local/lib/python3.12/site-packages/trame_server/core.pyset_default_client_typersc&eZdZdZ d$ d%dZd&d'dZdZedZdZ d&d(dZ ed Z ed Z ed Z ed)d Zed'd ZedZedZed*dZej&d+dZedZed,dZed,dZed-dZed.dZed/dZedZedZdZdZdZedZdZ dZ! d0 d1d Z"d%d!Z#ed2d"Z$ed#Z%y)3Servera] Server implementation for trame. This is the core object that manage client/server communication but also holds a state and controller instance. With trame a server instance should be retrieved by using **trame.app.get_server()** Known options: - log_network: False (path to log file) - ws_max_msg_size: 10000000 (bytes) - ws_heart_beat: 30 - desktop_debug: False :param name: A name identifier for a given server :type name: str, optional (default: trame) :param **options: Gather any keyword arguments into options :type options: Dict Nc ||_|r|n t|_t|d||_t|d||_t|dd|_t|dt|_d|_ d|_ d|_ d|_ d|_ i|_t|_g|_d|_d|_g|_dt*j,vxst/t1j2dd|_|/|j j7dt0j8j7d d |j d<|j j7d t0j8j7d d |j d <|j j7dt0j8j7dd|j d<|j j7dt0j8j7dd |j d<dt0j8d<|t;|j<|j>|j4|_ dD]}g|j@d|<dg|j@_!d|j@_"d|j@_#d|j@_$n|j4|_ |'tK|j<|j4|_&n1tK|j<|jL|j4|_&|'t;|j<|j4|_'n1t;|j<|jN|j4|_'t|dtQ|||_)y)N_name_options _client_type _http_headerr --hot-reloadTRAME_HOT_RELOAD log_networkTRAME_LOG_NETWORKFws_max_msg_sizeTRAME_WS_MAX_MSG_SIZEi ws_heart_beatTRAME_WS_HEART_BEAT desktop_debugTRAME_DESKTOP_DEBUGWSLINK_READY_MSG) commit_fn hot_reload)scriptsmodule_scriptsstylesvue_use mousetraptrame__ trame__busyrTrame)internalr5r6)r6)r?r6_ui)*_parent_serverr _translatorr r$r%r&rr'_server_running_stage _running_port_running_future_wwwserveset_loaded_modules_loaded_module_dicts _cli_parser_root_protocol_protocols_to_configuresysargvboolosgetenvr6getenvironr translator _push_state_statetrame__client_onlyr=trame__favicon trame__titler _controller_contextr r@)selfnamevn_constructorrV parent_serveroptionskeys r__init__zServer.__init__6s,)3:='48 mZA !-F!-N #  "u$&!"')$)CHH4  II($ /9   +/==+<+<rzz~~.A5I,DMM- (04}}/@/@!2::>>2I8#T0DMM+ ,.2]]->->0Er!J.DMM/ *.2]]->->0Eu!M.DMM/ *.0BJJ) *  4+;+;DKW/1 gcUO,W.;_DKK *&'DKK #)-DKK &'.DKK $&--**?? DK  )$//dooVD )&22?? D   !$//dooNDM!&//??DM/A$/WXr c<|r|n t|}t||S)N)prefix)rVra)rr")r^rVrfs rcreate_child_serverzServer.create_child_servers#-Z:V3L 4@@r cT|jr|jj|yyr)protocolpush_state_change)r^states rrWzServer._push_states == MM + +E 2 r c|jS)zIReturn http header helper so they can be applied before the server start.)r'r^s r http_headerszServer.http_headerss   r c |j|k7r|jj|fi|S|}t|tr+||jvry|jj |n6||j vry|j j||j}d|vr |d|fi|dD]$}||vs|jd|xx||z cc<&d|vr|jj|dd|vr|jj|dd|vr |d|_ tj|jy) an Expend server using a module definition which can be used to serve custom client code or assets, load/initialize resources (js, css, vue), register custom protocols and even execute custom code. Any previously seem module will be automatically skipped. The attributes that are getting processed in a module are the following: - setup(server, **kwargs): Function called first - scripts = [] : List all JavaScript URL that should be loaded - module_scripts = [] : List all JavaScript URL as type=module to load - styles = [] : List all CSS URL that should be loaded - vue_use = ['libName', ('libName2', { **options })]: List Vue plugin to load - state = {} : Set of variable to add to state - serve = { data: '/path/on/fs' }: Set of endpoints to serve static content - www = '/path/on/fs' : Path served as main web content :param module: A module to enable or a dict() :param kwargs: Any optional parameters needed for your module setup() function. Fsetup)r7r8r9r:r<rkrHwwwT) root_server enable_module isinstancedictrKappendrJadd__dict__rkupdaterHrGrreduce_vue_use)r^modulekwargs definitionsrcs rrszServer.enable_modulesO*   t #14##11&CFC C k4 (d777  % % , ,[ 9 D00 0  $ $[ 1%..K k ! K  0 0ECk! WSE?+{3/??+F k ! JJ  k'2 3 k ! JJ  k'2 3 K #E*DI TZZ(r cr|jr+|jjd||t|dgyy)a Python call method on JS element. :param ref: ref name of the widget element :type ref: str :param method: name of the method that should be called :type method: str :param *args: set of parameters needed for the function method)typerefrargsN)ri push_actionslist)r^rrrs rjs_callzServer.js_calls> == MM & &!)""( $T   r c.|jjS)a; Use as decorator `@server.change(key1, key2, ...)` so the decorated function will be called like so `_fn(**state)` when any of the listed key name is getting modified from either client or server. :param *_args: A list of variable name to monitor :type *_args: str )rXchangerms rrz Server.changes{{!!!r c.|jjS)z Use as decorator `@server.trigger(name)` so the decorated function will be able to be called from the client by doing `click="trigger(name)"`. :param name: A name to use for that trigger :type name: str )r\triggerrms rrzServer.triggers'''r c.|jjS)z Given a function this method will register a trigger and returned its name. If manually registered, the given name at the time will be returned. :return: The trigger name for that function :rtype: str )r\ trigger_namerms rrzServer.trigger_names,,,r c|jS)zName of server)r$rms rr_z Server.name,szzr cJ|jr|jjS|S)zRoot server to start)rArrrms rrrzServer.root_server1s$   &&22 2 r c|jS)zTranslator of the server)rBrms rrVzServer.translator8sr c|jS)z-Server options provided at instantiation time)r%rms rrbzServer.options=s}}r c>|jtS|jS)z9Specify the client type. Either 'vue2' or 'vue3' for now.)r&rrms r client_typezServer.client_typeBs"    $& &   r c|j||_|j|k7rd|jd|d}t|y)z;Should only be called once before any widget initializationNz"Trying to switch client_type from z to z&.The client_type can only be set once.)r&r TypeError)r^rmsgs rrzServer.client_typeIs[    $ %D    u $4T5F5F4GtE7S88 C. %r c|j|k7r|jjS|jr |jStdd|_|jj ddd|jj dd d|jj d d d|jj d d dd|jj dd|jj ddd|jj dd|jj ddddt j |j|jS)zargparse parserz Kitware trameF) description allow_abbrevz--serverz,Prevent your browser from opening at startup store_true)helpactionz--bannerzPrint trame bannerz--appzUse OS built-in browserz --no-httpzDo not serve anything over httpno_http)rdestrz --authKeyFilezPath to a File that contains the Authentication key for clients to connect to the WebSocket. This takes precedence over '-a, --authKey' from wslink.)rr(zAutomatically reload state/controller callback functions for every function call. This allows live editing of the functions. Functions located in the site-packages directories are skipped.z --trame-argsa#If specified, trame will ignore all other arguments, and only the contents of the `--trame-args` will be used. For example: `--trame-args="-p 8081 --server"`. Alternatively, the environment variable `TRAME_ARGS` may be set instead.z--follow-symlinksstatic_follow_symlinksaflag for allowing to follow symlinks that lead outside the static root directory, by default it's not allowed and HTTP/404 will be returned on access. Enabling follow_symlinks can be a security risk, and may lead to a directory transversal attack. You do NOT need this option to follow symlinks which point to somewhere else within the static directory, this option is only used to break out of the security sandbox. Enabling this option is highly discouraged, and only expected to be used for edge cases in a local development setting where remote users do not have access to the server.)rrr)rrclirLr add_argumentr add_argumentsrms rrz Server.cliVs   t ###'' '   ## #)'  %% ? & %% % & %% * & %% 2 & %% O & %% M &  %% 8 &  %% ) T  & "   !1!12r c|jS)z[ :return: The server shared state :rtype: trame_server.state.State )rXrms rrkz Server.states {{r c|jS)z The server-only context (not shared with the client). :return: The server context state :rtype: trame_server.state.State )r]rms rcontextzServer.contexts}}r c|jS)zc :return: The server controller :rtype: trame_server.controller.Controller )r\rms r controllerzServer.controllers r c|jS)zl :return: The server VirtualNode manager :rtype: trame_server.ui.VirtualNodeManager )r@rms ruiz Server.uis xxr cj|j|k7r|jjS|jdkDS)z;Return True if the server is currently starting or running.r)rrrunningrDrms rrzServer.runnings5   t ###++ +""Q&&r cptj|jjj S)zReturn a future to await if you want to ensure that any pending network call have been issued before locking the server)asyncio ensure_futurernetwork_monitor completionrms rnetwork_completionzServer.network_completions($$T\\%A%A%L%L%NOOr c|j|k7r|jjS|j'tjj |_|jS)z:Return a future that will resolve once the server is ready)rrreadyrFrget_running_loop create_futurerms rrz Server.readysV   t ###)) )    '#*#;#;#=#K#K#MD ###r cH|j|jjdS)zReturn the current server state)r_rk)r$rkinitialrms rget_server_statezServer.get_server_states"JJZZ''  r c@|j}|r|j|yyr)riclear_state_client_cache)r^ state_namesris rrzServer.clear_state_client_caches#==  -H - -{ ; r c|j|k7r|jj|y|jj|y)a9 Register function that will be called with a wslink.ServerProtocol when the server start and is ready for registering new wslink.Protocol. :param configure_protocol_fn: A function to be called later with a wslink.ServerProtocol as argument. N)rradd_protocol_to_configurerNrv)r^configure_protocol_fns rrz Server.add_protocol_to_configures?   t #    6 67L M  $$++,ABr cd|j|k7r|jjS|jS)zReturn the server root protocol)rrrirMrms rrizServer.protocols0   t ###,, ,"""r c|jr/|jj|}|r|\}}||g|i|Syd}t|)aQ Call a registered protocol method :param method: Method registration name :type method: str :param *args: Set of args to use for that method call :param **kwargs: Set of keyword arguments to use for that method call :return: transparently return what the called function returns NzProtocol does not exist yet)ri getRPCMethod ValueError)r^rrr|pairobjfuncerrors r protocol_callzServer.protocol_callsS ====--f5D TC1$1&11-r c |jdg|Dcgc]}|jj|c}ycc}w)z Should only be needed when client corrupted its data and need the server need to send it again. :param *args: Set of key names to be send again to the client. ztrame.force.pushN)rrB translate_key)r^ key_namesks rforce_state_pushzServer.force_state_pushsB   MV!WY$"2"2"@"@"CY!W !Ws"< c p  jk7r'jjd||||||| | d yjryjddlm} j | jjjjjrjjtjjjd} | t j"j%dd}||| _!t j"j%dd | j(d k(r)| t j"j%d d } | | _| | | _||| _| j.sj| _|rd | _| j2r+d d lm}jj8j;|| j<rd}|dk(r=d dlm d| _d\}}jj8j; fdtC| dd}|s6|r4|dk7r/d dl"m#jj8j;fd|sBr@|dk7r;| jHs/d dl%m&jj8j;fdtOjPr~g}jPD]W}jP|}tS|tTtVfr|jY|d|dB|jY|d|Ydj[|| _.| j^rd| _d| _.| _0tjb| d _tjd| fii |||d}|dk(rd_jjfjrtijj}jjfdijljoD]c}tqjr|r|ju|*tw|s6|}tqjr|sS|ju|e|Sty|drd fd }|j{||S)!a Start the server by listening to the provided port or using the `--port, -p` command line argument. If the server is already starting or started, any further call will be skipped. When the exec_mode="main" or "desktop", the method will be blocking. If exec_mode="task", the method will return a scheduled task. If exec_mode="coroutine", the method will return a coroutine which will need to be scheduled by the user. :param port: A port number to listen to. When 0 is provided the system will use a random open port. :param thread: If the server run in a thread which means we should disable interuption listeners :param open_browser: Should we open the system browser with app url. Using the `--server` command line argument is similar to setting it to False. :param show_connection_info: Should we print connection URL at startup? :param disable_logging: Ask wslink to disable logging :param backend: aiohttp by default but could be generic or tornado. This can also be set with the environment variable ``TRAME_BACKEND``. Defaults to ``'aiohttp'``. :param exec_mode: main/desktop/task/coroutine specify how the start function should work :param timeout: How much second should we wait before automatically stopping the server when no client is connected. Setting it to 0 will disable such auto-shutdown. :param host: The hostname used to bind the server. This can also be set with the environment variable ``TRAME_DEFAULT_HOST``. Defaults to ``'localhost'``. :param **kwargs: Keyword arguments for capturing optional parameters for wslink server and/or desktop browser ) portthread open_browsershow_connection_infodisable_loggingbackend exec_modetimeouthostNr)r{ TRAME_BACKENDr TRAME_SERVERF localhostTRAME_DEFAULT_HOSTTr) print_bannerr) start_browser)rFFcfiSr)_r|r^rs rzServer.start..sM$9&9r reverse_urlr)print_informationscSrr)rrr^s rrzServer.start..s .s L.on_donesMKKM*+D'77>>@888P4::;M;M;OPA-- M%%&EtLMsA/A44B* B*)B*r)rz asyncio.TaskreturnNone)>rrstartrDrG trame_clientr{rsr'applyron_server_startrr bind_serverrparse_known_argsrRrUrTrrrrcontentnosignalhandlersbanner utils.bannerron_server_readyrwapp utils.desktoprgetattr utils.serverrserver utils.browserrlenrHrtrtuplervjoin fsEndpointsr_server_options configure server_startrrget_event_looprkrinspect isawaitablerun_until_completecallablehasattrr)r^rrrrrrfollow_symlinksrrrr|r{rbrr endpointsrcrrloop exit_taskrrrrs` ` ` @@rrz Server.start*s*^   t # "D   " " )%9 /#      99  +   v & ! ?? * * 1 1 3 OO + +D 1t$((++-a0 ?jjnn_i@G  &-> 2 OO + + / / = ;;!I  ! 4GL

r1s" " "1'   8 $ ^ $ @A => "(Z( g $g $r