3 ?\ @sddlZddlZddlZddlmZmZejjejj e dZ ejjejj e dZ eddZ eddZedd d Zedd d ZdS)N)check handle_errorsz/../examples/matricesz/../examples/datafilescCsddtjtD}|S)z, Return list of available matrices. cSs,g|]$}d|krdj|jdddqS)z.txt.N)joinsplit).0namer ../../logomaker/src/examples.py sz)list_example_matrices..)oslistdir matrix_dir)valid_matricesr r r list_example_matricessrcCsddtjtD}|S)z. Return list of available data files. cSs8g|]0}t|jddkrt|jdddkr|qS)rr)lenr)r r r r r r sz*list_example_datafiles..)rrdata_dir)valid_datafilesr r r list_example_datafilessrTc Cst}t||kd|djdd|Dfdt|f}tjj|sPtd||rtd|t |d.}|j }d d|D}d j|}t|Wd QRXt j |d d ddS)a` Returns an example matrix from which a logo can be made. parameters ---------- name: (None or str) Name of example matrix. print_description: (bool) If true, a description of the example matrix will be printed returns ------- df: (data frame) A data frame containing an example matrix. z3Matrix "%s" not recognized. Please choose from: %s cSsg|] }t|qSr )repr)r xr r r r Csz&get_example_matrix..z %s/%s.txtzFile %s does not exist!z#Description of example matrix "%s":rcSs(g|] }t|dkr|ddkr|qS)r#)r)r lr r r r NsN rr)sep index_colcomment) rrrrrpathisfileAssertionErrorprintopen readlinespdread_csv)r print_descriptionr file_nameflines descriptionr r r get_example_matrix(s    r0c Cst}t||kd|djdd|Dfdt|f}tjj|sPtd||rtd|t |d.}|j }d d|D}d j|}t|Wd QRXt |d kr|dd d krt j |d}n t |d}|S)aG Returns a file handle to an example dataset parameters ---------- name: (None or str) Name of example matrix. print_description: (bool) If true, a description of the example matrix will be printed returns ------- f: (file handle) A handle to the requested file z3Matrix "%s" not recognized. Please choose from: %srcSsg|] }t|qSr )r)r rr r r r qsz)open_example_datafile..z%s/%szFile %s does not exist!z#Description of example matrix "%s":rcSs(g|] }t|dkr|ddkr|qS)rr)r)r rr r r r |srNz.gz) rrrrrr#r$r%r&r'r(rgzip)r r+rr,r-r.r/r r r open_example_datafileVs"     r4)NT)NT)pandasr)rr3logomaker.src.error_handlingrrr#dirnameabspath__file__rrrrr0r4r r r r s  -