o ;Di~@sddlmZmZmZddlZddlmZddlmZddl m Z m Z m Z ddl mZmZddZee_ee_d d Zee_ee_d d Zee_ee_d dZee_ee_dddZee_dddZee_dddZee_GdddeZdS))absolute_importprint_functiondivisionN) OrderedDict)islice) izip_longest text_typenext) asindicesTablecCdd|DS)NcSg|]}t|qSlist.0rowrrN/var/www/Datamplify/venv/lib/python3.10/site-packages/petl/util/materialise.py zlistoflists..rtblrrr listoflists rcCtdd|DS)Ncs|]}t|VqdSNtuplerrrr z tupleoftuples..rrrrr tupleoftuplesr"cCr )NcSr rrrrrrrrz listoftuples..rrrrr listoftuplesrr$cCr)Ncsrrrrrrrr &r!ztupleoflists..rrrrr tupleoflists%r#r%c Cst}t|}zt|}Wn tyg}Ynwttt|}|D]}t||<q"|D]}t|||dD]\}}||vrD|||q5q,|S)ap Construct a :class:`dict` mapping field names to lists of values. E.g.:: >>> import petl as etl >>> table = [['foo', 'bar'], ['a', 1], ['b', 2], ['b', 3]] >>> cols = etl.columns(table) >>> cols['foo'] ['a', 'b', 'b'] >>> cols['bar'] [1, 2, 3] See also :func:`petl.util.materialise.facetcolumns`.  fillvalue) riterr StopIterationrmaprrappend) tablemissingcolsithdrfldsfrvrrrcolumns-s"   r4cCst}t|}zt|}Wn tyg}Ynwttt|}t||}t|dks/Jdt j |}|D]7} || } | |vrRt} |D]} t| | <qE| || <n|| } t || |dD]\} } | | vrl| |  | q]q6|S)a Like :func:`petl.util.materialise.columns` but stratified by values of the given key field. E.g.:: >>> import petl as etl >>> table = [['foo', 'bar', 'baz'], ... ['a', 1, True], ... ['b', 2, True], ... ['b', 3]] >>> fc = etl.facetcolumns(table, 'foo') >>> fc['a'] {'foo': ['a'], 'bar': [1], 'baz': [True]} >>> fc['b'] {'foo': ['b', 'b'], 'bar': [2, 3], 'baz': [True, None]} rzno key field selectedr&) dictr(r r)rr*rr lenoperator itemgetterrr+)r,keyr-fctr/r0r1indicesgetkeyrkvr.r2r3rrr facetcolumnsPs2      r>cCs t||dS)z Wrap the table with a cache that caches up to `n` rows as they are initially requested via iteration (cache all rows be default). )n) CacheView)r,r?rrrcaches rAc@s&eZdZdddZddZddZdS) r@NcCs||_||_t|_d|_dSNF)innerr?rrA cachecomplete)selfrCr?rrr__init__s zCacheView.__init__cCst|_d|_dSrB)rrArD)rErrr clearcaches zCacheView.clearcacheccs|jD]}|Vq|jsBt|j}t|t|jdD]}|jr(t|j|jkr.|j||Vq|jr=t|j|jkrDd|_dSdSdS)NT)rArDr(rCrr6r?r+)rErr/rrr__iter__s     zCacheView.__iter__r)__name__ __module__ __qualname__rFrGrHrrrrr@s  r@r) __future__rrrr7 collectionsr itertoolsr petl.compatrrr petl.util.baser r rlolr"totr$lotr%tolr4r>rAr@rrrrs2    -