o ;Di@szddlmZmZmZddlZddlmZmZddlm Z m Z ddl m Z dddZ Gdd d e Zdd d ZdddZdS))absolute_importprint_functiondivisionN) string_types text_type)Tableiterpeek)construct_dtypecCst|||||dS)a'Extract a table from a bcolz ctable, e.g.:: >>> import petl as etl >>> >>> def example_from_bcolz(): ... import bcolz ... cols = [ ... ['apples', 'oranges', 'pears'], ... [1, 3, 7], ... [2.5, 4.4, .1] ... ] ... names = ('foo', 'bar', 'baz') ... ctbl = bcolz.ctable(cols, names=names) ... return etl.frombcolz(ctbl) >>> >>> example_from_bcolz() # doctest: +SKIP +-----------+-----+-----+ | foo | bar | baz | +===========+=====+=====+ | 'apples' | 1 | 2.5 | +-----------+-----+-----+ | 'oranges' | 3 | 4.4 | +-----------+-----+-----+ | 'pears' | 7 | 0.1 | +-----------+-----+-----+ If `expression` is provided it will be executed by bcolz and only matching rows returned, e.g.:: >>> tbl2 = etl.frombcolz(ctbl, expression='bar > 1') # doctest: +SKIP >>> tbl2 # doctest: +SKIP +-----------+-----+-----+ | foo | bar | baz | +===========+=====+=====+ | 'oranges' | 3 | 4.4 | +-----------+-----+-----+ | 'pears' | 7 | 0.1 | +-----------+-----+-----+ .. versionadded:: 1.1.0 ) expressionoutcolslimitskip) BcolzViewsourcer r r r rF/var/www/Datamplify/venv/lib/python3.10/site-packages/petl/io/bcolz.py frombcolz s,rc@s"eZdZ  dddZddZdS)rNrcCs"||_||_||_||_||_dSNr)selfrr r r r rrr__init__=s  zBcolzView.__init__c#st|jtrddl}|j|jddn|j|jdur"tj}nt|j}tfdd|Ds6Jd|V|j durJj |j|j |j d}n j |j |j|j |j d}|D]}|VqYdS)Nrrmodec3s|]}|jvVqdSr)names).0hctblrr Tsz%BcolzView.__iter__..zinvalid outcols)r r r ) isinstancerrbcolzopenr tuplerallr iterr r where)rr!headeritrowrrr__iter__Es(      zBcolzView.__iter__NNNr)__name__ __module__ __qualname__rr*rrrrr;s  rc  sddl}ddl}t|}t||\}}t|}dd|D}ttt|} t| ||}| dd| dd|j |j g|d fi|t fd dj Dtj } tt|| } |j | |d } | t| | krrnqWS) aLoad data into a bcolz ctable, e.g.:: >>> import petl as etl >>> >>> def example_to_bcolz(): ... table = [('foo', 'bar', 'baz'), ... ('apples', 1, 2.5), ... ('oranges', 3, 4.4), ... ('pears', 7, .1)] ... return etl.tobcolz(table) >>> >>> ctbl = example_to_bcolz() # doctest: +SKIP >>> ctbl # doctest: +SKIP ctable((3,), [('foo', '>> ctbl.names # doctest: +SKIP ['foo', 'bar', 'baz'] >>> ctbl['foo'] # doctest: +SKIP carray((3,), . expectedleni@Brwdtypec3|] }j|jVqdSrcolschunklenrnamerrrr)r!numpyr%rnextlistmaprr setdefaultctablearraysumrlen itertoolsisliceappendflush) tabler3samplekwargsr!npr(peekhdrfldsr7datarrrtobcolzcs2!      rPTc sddl}ddl}t|tr|j|ddnt|drt|ds%Jd||j}t|}t|}t t t |}|rHt |t j ksHJdtfd d j Dtj } t t|| } |j| |d } | t| | krunqZS) zAppend data into a bcolz ctable. The `obj` argument can be either an existing ctable or the name of a directory were an on-disk ctable is stored. .. versionadded:: 1.1.0 rNarrFrz$expected rootdir or ctable, found %rzcolumn names do not matchc3r4rr5r8rrrrr:zappendbcolz..Tr2)r!r;r rr"hasattrr3r%r<r=r>rr#rrBrCrDrErArFrG) rHobj check_namesr!rKr3r(rMrNr7rOrrr appendbcolzs8     rUr+)Nr/)T) __future__rrrrD petl.compatrrpetl.util.baserr petl.io.numpyr rrrPrUrrrrs  0 (?