U ZfG@s$ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZmZmZmZddlmZeeZdddZGdddZGd d d ejZGd d d eZGd ddeZGdddeZGdddeZ GdddeZ!dS)N)NoAppFoundErrorTestingTimeoutErrorServerCloseErrorDashAppLoadingError)waitappc CsXzt|}||}Wn<tk rR}ztdtd||W5d}~XYnX|S)a Import a dash application from a module. The import path is in dot notation to the module. The variable named app will be returned. :Example: >>> app = import_app("my_app.app") Will import the application in module `app` of the package `my_app`. :param app_file: Path to the app (dot-separated). :type app_file: str :param application_name: The name of the dash application instance. :raise: dash_tests.errors.NoAppFoundError :return: App from module. :rtype: dash.Dash zthe app name cannot be foundz$No dash `app` instance was found in N)runpyZ run_moduleKeyErrorlogger exceptionr)Zapp_fileapplication_name app_modulerZapp_name_missingrD/tmp/pip-unpacked-wheel-47crqvv_/dash/testing/application_runners.py import_apps   rc@steZdZdZdZddZddZddZed d Z d d Z d dZ ddZ e ddZe ddZe ddZdS)BaseDashRunnerz4Base context manager class for running applications.icCs"d|_d|_||_||_d|_dS)Nr)portstarted keep_open stop_timeout _tmp_app_pathselfrrrrr__init__@s zBaseDashRunner.__init__cOstdSNNotImplementedErrorrargskwargsrrrstartGszBaseDashRunner.startcCstdSrrrrrrstopJszBaseDashRunner.stopcCs.zt|Wntjjk r(YdSXdS)NFT)requestsget exceptionsRequestException)urlrrr accessibleMs zBaseDashRunner.accessiblecOs |j||Sr)r!rrrr__call__UszBaseDashRunner.__call__cCs|Srrr"rrr __enter__XszBaseDashRunner.__enter__c Csh|jrZ|jsZztd|Wn6tk rX}ztd|jd|W5d}~XYnXtddS)Nzkilling the app runnerzCannot stop server within z s timeoutz__exit__ complete)rrr infor#rrr)rexc_typeexc_val tracebackZcannot_stop_serverrrr__exit__[s    zBaseDashRunner.__exit__cCs d|jS)zThe default server url.zhttp://localhost:)rr"rrrr(fszBaseDashRunner.urlcCs tjdkS)Nwin32)sysplatformr"rrr is_windowskszBaseDashRunner.is_windowscCs|jSr)rr"rrr tmp_app_pathoszBaseDashRunner.tmp_app_pathN)__name__ __module__ __qualname____doc__ _next_portrr!r# staticmethodr)r*r+r0propertyr(r4r5rrrrr;s     rcs$eZdZfddZddZZS) KillerThreadc s"tjf|ttj|_dSr)superrlist threading_activekeys _old_threads)rr  __class__rrruszKillerThread.__init__cCstttjD]d}||jkrq tjt|tt }|dkrJt d||dkr tjt|dt dq dS)NrzInvalid thread id: zStopping thread failure) r?r@rArCctypes pythonapiZPyThreadState_SetAsyncExcc_long py_object SystemExit ValueError)rZ thread_idresrrrkillys zKillerThread.kill)r6r7r8rrN __classcell__rrrDrr=ts r=cs<eZdZdZd fdd ZddZd dd Zd d ZZS)ThreadedRunnerzkRuns a dash application in a thread. This is the default flavor to use in dash integration tests. Fcstj||dd|_dSN)rr)r>rthreadrrDrrrszThreadedRunner.__init__cCs |jr||StddS)NzThread is not alive.)rSis_aliver)r)rr(rrrrunning_and_accessibles  z%ThreadedRunner.running_and_accessiblec sfdd}d}js|dkrzhjrHjr>n jt|d_dj_jtj fdd|d j_Wqt k r}z&t |d _|d 7}t d W5d }~XYqXqj_jstd d S)z)Start the app server in threading flavor.c sdjj_djj_}dkrDtj|d<_tjd7_n |d_zjfddi|WnLt k rt dYn0t k r}zt ||W5d}~XYnXdS)NTrrFthreadedServer stopped)scriptsconfig serve_locallycsscopyrr:rrunrKr r, Exceptionr optionserrorrr rrrr]s    z!ThreadedRunner.start..runrrQtargetTcs jSr)rUr(rr"rrz&ThreadedRunner.start..timeoutFrFNzthreaded server failed to start)rrSrTr#rNr=daemonr!runtilr^r r timesleepr)rr start_timeoutr r]retrieserrrrbrr!s0        zThreadedRunner.startcCs0|j|jt|jj|jd|_dS)NF)rSrNjoinrZ until_notrTrrr"rrrr#s  zThreadedRunner.stop)FrQ)rQ) r6r7r8r9rrUr!r#rOrrrDrrPs  3rPcs0eZdZd fdd Zd ddZddZZS) MultiProcessRunnerFrQcst||d|_dSrr>rprocrrDrrrszMultiProcessRunner.__init__c sTdd_fdd}tj|d_jtjfdd|dd _dS) Nrrc sdjj_djj_}zjfddi|WnNtk rTtdYn0t k r}zt ||W5d}~XYnXdS)NTrVrW) rXrYrZr[r\r]rKr r,r^r r_)rr rrrds    z(MultiProcessRunner.start..targetrccs jSrr)r(rr"rrrerfz*MultiProcessRunner.start..rgT) r%r multiprocessProcessrsr!rrjr)rrrmr rdrrbrr!s  zMultiProcessRunner.startc Cst|jj}|jddD](}z |Wqtjk r@YqXqz |Wntjk rfYnXz|dWntjtjfk rYnXdS)NT) recursiverF) psutilrvrspidchildrenrNZ NoSuchProcessrTimeoutExpired)rprocessrsrrrr#s  zMultiProcessRunner.stop)FrQ)rQ)r6r7r8rr!r#rOrrrDrrqs rqcs4eZdZdZd fdd Zdd d Zd d ZZS) ProcessRunnerz}Runs a dash application in a waitress-serve subprocess. This flavor is closer to production environment but slower. FrQcstj||dd|_dSrRrrrrDrrr szProcessRunner.__init__Nrrc s|s|stddS|_tj|r(|nd|d|d|dj d}td|z0tj |tj tj d _ t j fd d |d Wn2ttfk rtd d_YdSXd_dS)z7Start the server with waitress-serve in process flavor.zAthe process runner needs to start with at least one valid commandNz waitress-serve --listen=0.0.0.0: :z.serverposixstart dash process with %s)stdoutstderrcs jSrrtrr"rrre/rfz%ProcessRunner.start..rg'process server has encountered an errorFT)loggingrarshlexsplitr4r debug subprocessPopenPIPErsrrjOSErrorrLr rr#)rr r Z raw_commandrrmrrr"rr!s4   zProcessRunner.startcCs|jrzbtd|jj|j|jrPtj|jrPt d|jt |jt j }|jj|jdWn2|k rtd|j|jYnXtddS)Nzproc.terminate with pid %szremoving temporary app path %srgzPsubprocess terminate not success, trying to kill the subprocess in a safe mannerzprocess stop completes!)rsr r,ry terminater5ospathexistsrshutilrmtreerr{ communicaterr rN)rZ_exceptrrrr#9s$    zProcessRunner.stop)FrQ)NrNrrQ)r6r7r8r9rr!r#rOrrrDrr}s 'r}cs(eZdZd fdd Zd ddZZS) RRunnerFrQcstj||dd|_dSrRrrrrDrrrQszRRunner.__init__Nc stj|r8tj|r8s4tj|tdntjjsHdnt dt j _ ztjWn"tk rtdjYnXtjjd}tdtd|tdtd |t|d d d }||W5QRX|}sLtD]F}d |dddkrtjtj|dtdqLqrtdfddtD}|D]p}tjjtj|} tj| rtd| t| td|jt|| tdt| qzn tdtd|tj dtj|dj d} td| z@t!j"| t!j#t!j#jrPjnd_$t%j&fdd |d!Wn,tt'fk rtd"d#_(Yd$SXd%_(d$S)&z-Start the server with subprocess and Rscript.z&RRunner inferred cwd from app path: %s/tmpTEMPcannot make temporary folder %szapp.Rz$RRunner start => app is R code chunkz+make a temporary R file for execution => %szcontent of the dashR app%swutf-8encoding/dash/testing/rF\/get cwd from inspect => %sz3RRunner inferred cwd from the Python call stack: %scs:g|]2}|dstjtj|rtj|qS__ startswithrrisdirrp.0_cwdrr s z!RRunner.start..delete existing target %scopying %s => %scopied with %szRRunner found no cwd in the Python call stack. You may wish to specify an explicit working directory using something like: dashr.run_server(app, cwd=os.path.dirname(__file__))z Run dashR app with Rscript => %szRscript -e 'source("z")'rrrrrcs jSrrtrr"rrrerfzRRunner.start..rgrFNT)rrisfilerdirnamer r,rpr4getenvuuidZuuid4hexrmkdirr5rr ropenwriteinspectstackreplacerealpathwarninglistdirbasenamerrcopytreerrrrrrsrrjrLr rrrmrrfpentryZassetsZassetrdrrrrrr!Vsz                z RRunner.start)FrQ)rNr6r7r8rr!rOrrrDrrPsrcs(eZdZd fdd Zd ddZZS) JuliaRunnerFrQcstj||dd|_dSrRrrrrDrrrszJuliaRunner.__init__Nc stj|r8tj|r8s4tj|tdntjjsHdnt dt j _ ztjWn"tk rtdjYnXtjjd}tdtd|tdtd |t|d d d }||W5QRX|}sLtD]F}d |dddkrtjtj|dtdqLqrtdfddtD}|D]p}tjjtj|} tj| rtd| t| td|jt|| tdt| qzn tdtd|tj dtj|j d} td| z@t!j"| t!j#t!j#jrNjnd_$t%j&fdd|d Wn,tt'fk rtd!d"_(Yd#SXd$_(d#S)%z+Start the server with subprocess and julia.z*JuliaRunner inferred cwd from app path: %srrrzapp.jlz,JuliaRunner start => app is Julia code chunkz/make a temporary Julia file for execution => %szcontent of the Dash.jl apprrrrrrFrrrz7JuliaRunner inferred cwd from the Python call stack: %scs:g|]2}|dstjtj|rtj|qSrrrrrrrs z%JuliaRunner.start..rrrzJuliaRunner found no cwd in the Python call stack. You may wish to specify an explicit working directory using something like: dashjl.run_server(app, cwd=os.path.dirname(__file__))z Run Dash.jl app with julia => %szjulia --project rzstart Dash.jl process with %srcs jSrrtrr"rrre rfz#JuliaRunner.start..rgrFNTrrrrrr!s               zJuliaRunner.start)FrQ)rNrrrrDrrsr)r)"r2rrkrrr@rrrrrGrr$rxruZdash.testing.errorsrrrrZ dash.testingr getLoggerr6r rrThreadr=rPrqr}rrrrrrs2   9J2Ia