o XjC@s UddlmZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZmZeejdZed Zed ed ed d Zded<GdddZeZdddZGddde ZdddZedkre edS) ) annotationsN)datetime) HTTPStatus)BaseHTTPRequestHandlerThreadingHTTPServer)Path)parse_qsurlparsewebz index.htmlz styles.csszapp.js)/z/assets/styles.cssz/assets/app.jszdict[str, Path] STATIC_ROUTESc@s>eZdZdddZdddZdd d ZdddZdddZdS)RunStatereturnNonecCs2t|_d|_d|_d|_d|_g|_g|_dS)NF) threadingLocklockrunning started_at finished_at exit_codecommandlogsselfr bin/web_ui.py__init__s  zRunState.__init__dictc CsP|j|j|j|j|j|j|jdddWdS1s!wYdS)Nip)rrrrrr)rrrrrrrrrrrsnapshot$s $zRunState.snapshotcmd list[str]boolcCs|j9|jr WddSd|_tjdd|_d|_d|_||_d|jdd |g|_ WddS1s?wYdS)NFTsecondstimespec[z ] Starting:  ) rrrnow isoformatrrrrjoinr)rr!rrrstart/s$zRunState.startlinestrcCsl|j)|j|dt|jdkr$|jdd|_WddSWddS1s/wYdS)N ii0)rrappendrstriplen)rr-rrr append_log;s"zRunState.append_logcodeintcCsd|j%d|_||_tjdd|_|jd|jd|WddS1s+wYdS)NFr$r%r'z] Finished with exit_code=) rrrrr)r*rrr0)rr4rrrfinishAs "zRunState.finishNrr)rr)r!r"rr#)r-r.rr)r4r5rr)__name__ __module__ __qualname__rr r,r3r6rrrrrs   rr!r"rrc Csftj}tj|tttjtjdd|d}|j dusJ|j D]}t |q| t |jdS)NTr )cwdstdoutstderrtextbufsizeenv)osenvironcopy subprocessPopenr.ROOTPIPESTDOUTr<STATEr3waitr6 returncode)r!r@processr-rrr_runnerLs   rMc@sLeZdZdddd ZddddZdddZd ddZd ddZd!ddZdS)"Handlerpayloadrstatusr5rrcCsRt|d}|||dd|dtt|||j |dS)Nzutf-8 Content-Typezapplication/jsonContent-Length) jsondumpsencode send_response send_headerr.r2 end_headerswfilewrite)rrPrQbodyrrr _send_json_s   zHandler._send_jsonr\bytes content_typer.cCsB|||d||dtt|||j|dS)NrRrS)rWrXr.r2rYrZr[)rr\r_rQrrr _send_bytesgs  zHandler._send_bytespathr#cCs^t|}|dus |sdS|}|dkrd}n tt|\}}|p&d}|||dS)NFr ztext/html; charset=utf-8zapplication/octet-streamT)r getis_file read_bytes mimetypes guess_typer.r`)rra file_pathr\r_guessed_rrr _serve_staticns  zHandler._serve_staticcCsJt|jj}|dkr|tdS||rdS|dditjdS)Nz /api/statuserror Not found)r rar]rIr rjr NOT_FOUND)rrarrrdo_GET}s  zHandler.do_GETcCst|j}|jdkr|dditjdSt|j}|ddgddv}|dd gd}t j d g}|r<| d |rE| d |gt |sU|d ddddStjt|fdd}||dd|ddS)Nz /api/startrkrldry_run0r>1yestrue batch_sizezbin/run_pipeline.pyz --dry-runz --batch-sizeFzA run is already in progress)okmessageiT)targetargsdaemonz Run started)rvrwr)r rar]rrmrqueryrbstripsys executabler0extendrIr,rThreadrM)rparsedqrortr!trrrdo_POSTs$      zHandler.do_POSTfmtcGsdS)Nr)rrryrrr log_messageszHandler.log_messageN)rO)rPrrQr5rr)r\r^r_r.rQr5rr)rar.rr#r7)rr.rr) r8r9r:r]r`rjrnrrrrrrrN^s     rNr5cCsVtstdttjddSd}d}t||ft}td|d||dS) NzMissing web directory: )filer z 127.0.0.1iz#Synapse launcher running at http://:r)WEB_DIRis_dirprintr}r=rrN serve_forever)hostportserverrrrmainsr__main__)r!r"rr)rr5)! __future__rrTrerArDr}rrhttpr http.serverrrpathlibr urllib.parserr __file__resolveparentsrFrr __annotations__rrIrMrNrr8 SystemExitrrrrs4   /  D