li2dZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZGd d ZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9dd l:m;Z;dd!lm?Z?dd#l@mAZAdd$lBmCZCdd%lDmEZEdd&lFmGZGdd'lHmIZIdd(lJmKZKdd)lLmMZMdd*lNmOZOdd+lPmQZQdd,lRmSZSdd-lTmUZUee_ee_ee_ee_ee_e!e_!e#e_#e%e_%e'e_'e)e_)e+e_+e-e_-e/e_/e1e_1e3e_3e5e_5e7e_7e9e_9e;e_;e=e_=e?e_?eAe_AeCe_CeEe_EeGe_GeIe_IeKe_KeMe_MeOe_OeQe_QeSe_SeUe_Uy).a The :class:`.Table` object is the most important class in agate. Tables are created by supplying row data, column names and subclasses of :class:`.DataType` to the constructor. Once created, the data in a table **can not be changed**. This concept is central to agate. Instead of modifying the data, various methods can be used to create new, derivative tables. For example, the :meth:`.Table.select` method creates a new table with only the specified columns. The :meth:`.Table.where` method creates a new table with only those rows that pass a test. And :meth:`.Table.order_by` creates a sorted table. In all of these cases the output is a new :class:`.Table` and the existing table remains unmodified. Tables are not themselves iterable, but the columns of the table can be accessed via :attr:`.Table.columns` and the rows via :attr:`.Table.rows`. Both sequences can be accessed either by numeric index or by name. (In the case of rows, row names are optional.) N)StringIO)chain)utils)Column)DataType) CastError)MappedSequence)Row) TypeTesterceZdZdZddZdZdZdZdZe dZ e d Z e d Z e d Z e d Zdd ZdZdZy)Tablea1 A dataset consisting of rows and columns. Columns refer to "vertical" slices of data that must all be of the same type. Rows refer to "horizontal" slices of data that may (and usually do) contain mixed types. The sequence of :class:`.Column` instances are retrieved via the :attr:`.Table.columns` property. They may be accessed by either numeric index or by unique column name. The sequence of :class:`.Row` instances are retrieved via the :attr:`.Table.rows` property. They may be accessed by either numeric index or, if specified, unique row names. :param rows: The data as a sequence of any sequences: tuples, lists, etc. If any row has fewer values than the number of columns, it will be filled out with nulls. No row may have more values than the number of columns. :param column_names: A sequence of string names for each column or `None`, in which case column names will be automatically assigned using :func:`.letter_name`. :param column_types: A sequence of instances of :class:`.DataType` or an instance of :class:`.TypeTester` or `None` in which case a generic TypeTester will be used. Alternatively, a dictionary with column names as keys and instances of :class:`.DataType` as values to specify some types. :param row_names: Specifies unique names for each row. This parameter is optional. If specified it may be 1) the name of a single column that contains a unique identifier for each row, 2) a key function that takes a :class:`.Row` and returns a unique identifier or 3) a sequence of unique identifiers of the same length as the sequence of rows. The uniqueness of resulting identifiers is not validated, so be certain the values you provide are truly unique. :param _is_fork: Used internally to skip certain validation steps when data is propagated from an existing table. When :code:`True`, rows are assumed to be :class:`.Row` instances, rather than raw data. Nc >t|tr td|rtj|d|_np|r_t dtt|dD|_tjdt|j ztdnt |_t|j }| t}nt|tr=|jD]}t|trtd t| }n2t|ts"|D]}t|trtd t|tr"|j!||j |_nt ||_|t|j"k7r td |sg} |j"D cgc]} | j$} } t'|D]\} } t| }||kDrtd | ||fzt| |krt)| dg||z z} g}t'| D] \}} |j+| ||"| j+t/||j n|} |rg}t|tr| D]} | |}|j+|nPt1|dr!| D]} || }|j+|n#tj2|r|}n td|D]}t5|t6ustdt ||_nd|_t;| |j8|_g}t|D]U} |j | }|j"| }t?| |||j<|j8}|j+|Wt;||j |_ ycc} w#t,$r0}t-t|d | d|j |dzd}~wwxYw)NzqWhen created directly, the first argument to Table must be a sequence of rows. Did you want agate.Table.from_csv?T) column_namesc3FK|]}tj|yw)N)r letter_name).0is Y/var/www/html/strategist-ai/venv_dbt/lib/python3.12/site-packages/agate/table/__init__.py z!Table.__init__..Rs&Yu'8'8';&Ys!rz7Column names not specified. "%s" will be used as names.) stacklevelz+Column types must be instances of DataType.)forcez6column_names and column_types must be the same length.z4Row %i has %i values, but Table only has %i columns.z Error at row z column .__call__z5row_names must be a column name, function or sequencezDRow names cannot be of type int. Use Decimal for numbered row names.) row_names)! isinstancestr ValueErrorr deduplicate _column_namestuplerangelenwarningswarnRuntimeWarningr dictvaluesrrun _column_typescast enumeraterappendrr hasattr issequencetypeint _row_namesr _rowsr_columns)selfrowsr column_typesr_is_forklen_column_namesv column_typenew_rowsc cast_funcsrrowlen_row row_valuesjdecomputed_row_namesnamerow_name new_columns data_typecolumns r__init__zTable.__init__Is dC BC C !&!2!2D&--d34J/#4C$S>D&--d34!!),%." !XYY. m>S($%kll m$$67DO"DO#Hdoo>  '( 'A%%a(D**1-IAtY dooVF   v &  ''{D4F4FG y> %g'A>!HTM_M_`aMbLccd1e(effgsO?O## P,+PPcZt}|j||jS)zS Print the table's structure using :meth:`.Table.print_structure`. )output)rprint_structuregetvalue)r5 structures r__str__z Table.__str__s+J  I.!!##c6|jjS)z8 Shorthand for :code:`len(table.rows)`. )r3__len__r5s rrTz Table.__len__szz!!##rRc6|jjS)z9 Shorthand for :code:`iter(table.rows)`. )r3__iter__rUs rrWzTable.__iter__szz""$$rRc8|jj|S)z8 Shorthand for :code:`table.rows[foo]`. )r3 __getitem__)r5keys rrYzTable.__getitem__szz%%c**rRc|jS)z8 An tuple :class:`.DataType` instances. )r*rUs rr7zTable.column_types !!!rRc|jS)z& An tuple of strings. )r rUs rrzTable.column_namesr\rRc|jS)z An tuple of strings, if this table has row names. If this table does not have row names, then :code:`None`. )r2rUs rrzTable.row_namessrRc|jS)zz A :class:`.MappedSequence` with column names for keys and :class:`.Column` instances for values. )r4rUs rcolumnsz Table.columnss }}rRc|jS)z A :class:`.MappedSeqeuence` with row names for keys (if specified) and :class:`.Row` instances for values. )r3rUs rr6z Table.rowss zzrRcv| |j}| |j}| |j}t||||dS)a Create a new table using the metadata from this one. This method is used internally by functions like :meth:`.Table.order_by`. :param rows: Row data for the forked table. :param column_names: Column names for the forked table. If not specified, fork will use this table's column names. :param column_types: Column types for the forked table. If not specified, fork will use this table's column names. :param row_names: Row names for the forked table. If not specified, fork will use this table's row names. T)rr8)r r*r2r )r5r6rr7rs r_forkz Table._forksI&  --L  --L  IT<UYZZrRc F|jtjfi|y)z Print this table as a CSV. This is the same as passing :code:`sys.stdout` to :meth:`.Table.to_csv`. :code:`kwargs` will be passed on to :meth:`.Table.to_csv`. N)to_csvsysstdoutr5kwargss r print_csvzTable.print_csvs  CJJ)&)rRc F|jtjfi|y)z Print this table as JSON. This is the same as passing :code:`sys.stdout` to :meth:`.Table.to_json`. :code:`kwargs` will be passed on to :meth:`.Table.to_json`. N)to_jsonrfrgrhs r print_jsonzTable.print_jsons  SZZ*6*rR)NNNF)NNN)__name__ __module__ __qualname____doc__rKrQrTrWrYpropertyr7rrr`r6rcrjrmrRrr r "s%LeHN$$ % + "" "" [<* +rRr ) aggregate) bar_chart)bins) column_chart)compute) denormalize)distinct)exclude)find)from_csv) from_fixed) from_json) from_object)group_by) homogenize)join)limit) line_chart)merge) normalize)order_by)pivot) print_bars) print_html)rN) print_table)rename) scatterplot)select)re)rl)where)Vrqrfr$ior itertoolsragater agate.columnsragate.data_typesragate.exceptionsragate.mapped_sequencer agate.rowsr agate.type_testerr r agate.table.aggregatertagate.table.bar_chartruagate.table.binsrvagate.table.column_chartrwagate.table.computerxagate.table.denormalizeryagate.table.distinctrzagate.table.excluder{agate.table.findr|agate.table.from_csvr}agate.table.from_fixedr~agate.table.from_jsonragate.table.from_objectragate.table.group_byragate.table.homogenizeragate.table.joinragate.table.limitragate.table.line_chartragate.table.mergeragate.table.normalizeragate.table.order_byragate.table.pivotragate.table.print_barsragate.table.print_htmlragate.table.print_structurerNagate.table.print_tableragate.table.renameragate.table.scatterplotragate.table.selectragate.table.to_csvreagate.table.to_jsonrlagate.table.whererrsrRrrs&  %&0(B+B+J,+!1'/)'!)-+/)-!#-#+)#--7/%/%%'#  !        '     rR