B d<@sddlZddlZddlZddlmZddlmZmZddl m Z ddl Z ddl m Z ddlmZddlmZmZdd Zd d Zd d ZddZddZddZddZddZdS)N) ConfigParser) EnvironmentFileSystemLoader)unescape)unescape_table)save_html_with_metadata)SegFilemakedirc Kstj|}t|ydd|D}Wntk r>g}YnXtjjdd}tt |d}|j j t d|j j tjj d|j j td||}||d }| |||} d d |d } | |x| D]} t| | | | <qWtt| |d | d S)z Writes a template given inputs from the workflow generator. Takes a list of tuples. Each tuple is a pycbc File object. Also the name of the subtemplate to render and the filename of the output. cSs$g|]}|D]}|dk r |jq qS)N)name).0filelistfra/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/results/render.py )sz1render_workflow_html_template..rz/templates/wells)loader)get_embedded_config) path_exists)len) filelistsdir render_tmplt,)zrender-function filenamesN)ospathdirnamer TypeErrorpycbcresults__path__rrglobalsupdaterexistsr get_templaterenderjoinstrr) filenameZ subtemplaterkwargsZdirnamrZsubtemplate_direnvcontextoutputkwdskeyrrrrender_workflow_html_templates,       r/cCsFdd}ytj|}Wntk r2t}YnXt|||_|S)z2 Attempt to load config data attached to file cSs"||o |||p ||kS)N) has_section has_optiondefaults)selfsectionr rrr check_optionFs z)get_embedded_config..check_option)rrZload_metadata_from_filerrtypes MethodTyper5)r(r5cprrrrCs rcCst|}d}tj|}||drD||d}t|}|||}nXtj|r||||dr||d}t|}|||}qt ||}n t ||}|S)zG This function is the gateway for rendering a template for a file. zrender-function) rrrbasenamer1getevalr#readrender_default)rZ config_pathr8r,r(Zrender_function_nameZrender_functionrrrsetup_template_renderSs            r?c Cstj|}|dd}d}|ds0|drzyt|}|Wn2tk rx}zt d||fWdd}~XYnXt j j dd}t t|d }|jjtd |jjtd |jjtjjd |d } |||||d} | | } | S)a% This is the default function that will render a template to a string of HTML. The string will be for a drop-down tab that contains a link to the file. If the file extension requires information to be read, then that is passed to the content variable (eg. a segmentlistdict). ._Nz.xmlz.xml.gzz!No segment table found in %s : %srz/templates/files)r)abs)open)rzfile_default.html)rr(slugr8content)rrr:replaceendswithrZfrom_segment_xmlZreturn_union_seglist Exceptionprintrrr rrr!r"rBrCr#r$r%) rr8r(rDrEZwf_filee template_dirr*templater+r,rrrr>us,    "  r>c Csptj|}|dd}tjjdd}tt|d}|j j t d| | |d}|||d}||}|S) z( Render a glitchgram file template. r@rArz/templates/files)r)rBrL)r(rDr8)rrr:rFrrr rrr!r"rBr$r;r%) rr8r(rDrKr*rLr+r,rrrrender_glitchgrams   rMc Cstj|}|dd}d}tj|dddd}|}WdQRXt|t}t j j dd }t t |d }|jjtd |jjtjjd |d }||||d} || } | S)z Render a file as text. r@rANrzutf-8rF)encodingerrorsrz/templates/files)r)rB)rz file_pre.html)r(rDr8rE)rrr:rFcodecsrCr=rrrrr rrr!r"rBr#r$r%) rr8r(rDrEfprKr*rLr+r,rrr render_texts"     rScCsdS)z Does not render anything. r9r)rr8rrr render_ignoresrTc Cstj|}|dd}d}t|d}|}WdQRXt|t}d| ddd}t t |d}|j j td|j j td |j j tjjd ||}||||d } || } | S) z Render a file as text. r@rANrN/)r)r?)r)r)r(rDr8rE)rrr:rFrCr=rrr&splitrrr!r"r?rr#r$r%) rr8r(rDrErRrKr*rLr+r,rrrrs$      r)os.pathrr6rQ configparserrZjinja2rrZxml.sax.saxutilsrZ pycbc.resultsrrZpycbc.results.metadatarZpycbc.workflow.corerr r/rr?r>rMrSrTrrrrrs     $")