lidZddlmZddlmZddlmZddlmZGddeZ ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1e e _ e e _ ee _ee _ee _ee _e e _ e!e _!e"e _"ee _ee _e#e _#ee _e$e _$e%e _%e&e _&ee _ee _e'e _'e(e _(e)e _)ee _e-e _-e*e _*e/e _/e1e _1e+e _+y)a The :class:`.TableSet` class collects a set of related tables in a single data structure. The most common way of creating a :class:`.TableSet` is using the :meth:`.Table.group_by` method, which is similar to SQL's ``GROUP BY`` keyword. The resulting set of tables will all have identical columns structure. :class:`.TableSet` functions as a dictionary. Individual tables in the set can be accessed by using their name as a key. If the table set was created using :meth:`.Table.group_by` then the names of the tables will be the grouping factors found in the original data. :class:`.TableSet` replicates the majority of the features of :class:`.Table`. When methods such as :meth:`.TableSet.select`, :meth:`.TableSet.where` or :meth:`.TableSet.order_by` are used, the operation is applied to *each* table in the set and the result is a new :class:`TableSet` instance made up of entirely new :class:`.Table` instances. :class:`.TableSet` instances can also contain other TableSet's. This means you can chain calls to :meth:`.Table.group_by` and :meth:`.TableSet.group_by` and end up with data grouped across multiple dimensions. :meth:`.TableSet.aggregate` on nested TableSets will then group across multiple dimensions. )StringIO) zip_longest)Text)MappedSequencecleZdZdZd dZdZedZedZedZ edZ d d Z d Z y) TableSeta^ An group of named tables with identical column definitions. Supports (almost) all the same operations as :class:`.Table`. When executed on a :class:`TableSet`, any operation that would have returned a new :class:`.Table` instead returns a new :class:`TableSet`. Any operation that would have returned a single value instead returns a dictionary of values. TableSet is implemented as a subclass of :class:`.MappedSequence` :param tables: A sequence :class:`Table` instances. :param keys: A sequence of keys corresponding to the tables. These may be any type except :class:`int`. :param key_name: A name that describes the grouping properties. Used as the column header when the groups are aggregated. Defaults to the column name that was grouped on. :param key_type: An instance some subclass of :class:`.DataType`. If not provided it will default to a :class`.Text`. :param _is_fork: Used internally to skip certain validation steps when data is propagated from an existing tablset. Nctt|}t|}||_|xs t|_|d|_t |jt r/|jd|_t |jt r/|jj|_|jj|_ |sf|D]a}tdt|j|jDr td|j|jk7sXtdtj|||y)Nrc3NK|]\}}t|t| yw)N) isinstancetype).0abs \/var/www/html/strategist-ai/venv_dbt/lib/python3.12/site-packages/agate/tableset/__init__.py z$TableSet.__init__..Ks"rda:aa11rs#%z*Not all tables have the same column types!z*Not all tables have the same column names!)tuple _key_namer _key_type _sample_tabler r column_types _column_types column_names _column_namesanyr ValueErrorr__init__)selftableskeyskey_namekey_type_is_forktables rrzTableSet.__init__;s vT{!!+TV#AY++X6!%!3!3A!6D ++X6"//<<!//<< Srk%J\J\^b^p^p>qrr$%QRR%%););;$%QRR  S fd3cZt}|j||jS)zV Print the tableset's structure via :meth:`TableSet.print_structure`. )output)rprint_structuregetvalue)r structures r__str__zTableSet.__str__Ss+J  I.!!##r$c|jS)z Get the name of the key this TableSet is grouped by. (If created using :meth:`.Table.group_by` then this is the original column name.) )rrs rr zTableSet.key_name] ~~r$c|jS)z Get the :class:`.DataType` this TableSet is grouped by. (If created using :meth:`.Table.group_by` then this is the original column type.) )rr,s rr!zTableSet.key_typeer-r$c|jS)z Get an ordered list of this :class:`.TableSet`'s column types. :returns: A :class:`tuple` of :class:`.DataType` instances. )rr,s rrzTableSet.column_typesm!!!r$c|jS)z Get an ordered list of this :class:`TableSet`'s column names. :returns: A :class:`tuple` of strings. )rr,s rrzTableSet.column_nameswr0r$cZ| |j}| |j}t||||dS)z Create a new :class:`.TableSet` using the metadata from this one. This method is used internally by functions like :meth:`.TableSet.having`. T)r")rrr)rrrr r!s r_forkzTableSet._forks5  ~~H  ~~Hh4HHr$cg}|jD]&\}}|jt|||i|(|j||j S)zJ Calls a method on each table in this :class:`.TableSet`. )itemsappendgetattrr3r)r method_nameargskwargsrkeyr#s r_proxyzTableSet._proxysc**, HJC MM5'%5tFvF G Hzz  IIK  r$)groupNF)NN) __name__ __module__ __qualname____doc__rr*propertyr r!rrr3r<r$rrr sl440$"""" I  r$r) aggregate) bar_chart) column_chart)from_csv) from_json)having) line_chart)merge)r')binscompute denormalizedistinctexcludefindgroup_by homogenizejoinlimit normalizeorder_bypivotselectwhere) scatterplot)to_csv)to_jsonN)2rAior itertoolsragate.data_typesragate.mapped_sequencerragate.tableset.aggregaterDagate.tableset.bar_chartrEagate.tableset.column_chartrFagate.tableset.from_csvrGagate.tableset.from_jsonrHagate.tableset.havingrIagate.tableset.line_chartrJagate.tableset.mergerKagate.tableset.print_structurer'agate.tableset.proxy_methodsrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZagate.tableset.scatterplotr[agate.tableset.to_csvr\agate.tableset.to_jsonr]rCr$rros90!!0| ~| ~/.4,.(0&:bbbbb2(* $"    *"r$