ó į?Æ\c@sĘddlZddlZddlZddlmZmZejjejj e ƒƒdZ ejjejj e ƒƒdZ ed„ƒZ ed„ƒZeded„ƒZeded„ƒZdS( iĸĸĸĸN(tcheckt handle_errorss/../examples/matricess/../examples/datafilescCsHgtjtƒD].}d|krdj|jdƒd ƒ^q}|S(s, Return list of available matrices. s.txtt.iĸĸĸĸ(tostlistdirt matrix_dirtjointsplit(tnametvalid_matrices((s../../logomaker/src/examples.pytlist_example_matricessAcCs`gtjtƒD]F}t|jdƒƒdkrt|jdƒdƒdkr|^q}|S(s. Return list of available data files. Rii(RRtdata_dirtlenR(Rtvalid_datafiles((s../../logomaker/src/examples.pytlist_example_datafiless +c Cstƒ}t||kd|djg|D]}t|ƒ^q(ƒfƒdt|f}tjj|ƒsztd|ƒ‚|rĸd|GHt |dƒb}|j ƒ}g|D].}t |ƒdkrŪ|ddkrŪ|^qŪ}d j|ƒ}|GHWd QXnt j |d d d dddƒS(s` 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. s3Matrix "%s" not recognized. Please choose from: %ss s %s/%s.txtsFile %s does not exist!s#Description of example matrix "%s":trit#tNtseps t index_coltcomment(R RRtreprRRtpathtisfiletAssertionErrortopent readlinesR tpdtread_csv( Rtprint_descriptionR txt file_nametftlinestlt description((s../../logomaker/src/examples.pytget_example_matrix(s  0"  ;c CsItƒ}t||kd|djg|D]}t|ƒ^q(ƒfƒdt|f}tjj|ƒsztd|ƒ‚|rĸd|GHt |dƒb}|j ƒ}g|D].}t |ƒdkrŪ|ddkrŪ|^qŪ}d j|ƒ}|GHWd QXnt |ƒd kr6|d d kr6t j |dƒ}nt |dƒ}|S(sG 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 s3Matrix "%s" not recognized. Please choose from: %ss s%s/%ssFile %s does not exist!s#Description of example matrix "%s":RiRRNiiýĸĸĸs.gz( RRRRR RRRRRRR tgzip( RRR RRR R!R"R#((s../../logomaker/src/examples.pytopen_example_datafileVs"  0"  ;"(tpandasRRR%tlogomaker.src.error_handlingRRRtdirnametabspatht__file__RR R RtNonetTrueR$R&(((s../../logomaker/src/examples.pyts    -