U f@sddlZddlZddlZddlmZmZejej e dZ ejej e dZ eddZ eddZedd d Zedd d ZdS)N)check handle_errorsz/../examples/matricesz/../examples/datafilescCsddttD}|S)z, Return list of available matrices. cSs,g|]$}d|krd|dddqS)z.txt.N)joinsplit.0namer =/tmp/pip-target-lpfmz8o1/lib/python/logomaker/src/examples.py sz)list_example_matrices..)oslistdir matrix_dir)valid_matricesr r r list_example_matricessrcCsddttD}|S)z. Return list of available data files. cSs8g|]0}t|ddkrt|dddkr|qS)rr)lenrrr r r r sz*list_example_datafiles..)rrdata_dir)valid_datafilesr r r list_example_datafilessrTc Cst}t||kd|ddd|Dftt|tdt|dt|f}tj |sht d||rt d|t |d .}| }d d|D}d |}t |W5QRXtj|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. 3Matrix "%s" not recognized. Please choose from: %s cSsg|] }t|qSr reprr xr r r r Csz&get_example_matrix..3type(print_description) = %s; must be of type bool z %s/%s.txtFile %s does not exist!#Description of example matrix "%s":rcSs(g|] }t|dkr|ddkr|qSr#rr lr r r r Ss  rr#)sepZ index_colcomment)rrr isinstancebooltyperrpathisfileAssertionErrorprintopen readlinespdZread_csv)r print_descriptionr file_nameflines descriptionr r r get_example_matrix(s*     r:c Cst}t||kd|ddd|Dftt|tdt|dt|f}tj |sht d||rt d|t |d .}| }d d|D}d |}t |W5QRXt|d kr|d ddkrt |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 rrcSsg|] }t|qSr rrr r r r vsz)open_example_datafile..rz%s/%srr r!cSs(g|] }t|dkr|ddkr|qSr"r$r%r r r r s r'Nz.gz)rrrr+r,r-rrr.r/r0r1r2r3rgzip)r r5rr6r7r8r9r r r open_example_datafile[s0      r>)NT)NT)Zpandasr4rr=Zlogomaker.src.error_handlingrrr.dirnameabspath__file__rrrrr:r>r r r r s"   2