li $ddlmZddlmZddZy))utils)RowNc t|j}tj|s|g}t |dk(r"t d|Dr|Dcgc]}|g}}|Dcgc]}|j j|}}|Dcgc]}|jj|}}|D cgc]<} t| D cgc]#\} } || jj| %c} } >}} } } t|}tttt |t|z } | D]} t#|r || }nY| t|}n%dgt |jt |z z}t|| D]\} }|j%| |t|D cgc]-\} } |j | jj| /}} } |j't)||j|j+|gScc}wcc}wcc}wcc} } wcc} } } wcc} } w)aJ Fill in missing rows in a series. This can be used, for instance, to add rows for missing years in a time series. Missing rows are found by comparing the values in the :code:`key` columns with those provided as :code:`compare_values`. Values not found in the table will be used to generate new rows with the given :code:`default_row`. :code:`default_row` should be an array of values or an array-generating function. If not specified, the new rows will have :code:`None` in columns all columns not specified in :code:`key`. If :code:`default_row` is an array of values, its length should be row length minus the number of column names provided in the :code:`key`. If it is an array-generating function, the function should take an array of missing values for each new row and output a full row including those values. :param key: Either a column name or a sequence of such names. :param compare_values: Either an array of column values if key is a single column name or a sequence of arrays of values if key is a sequence of names. It can also be a generator that yields either of the two. A row is created for each value or list of values not found in the rows of the table. :param default_row: An array of values or a function to generate new rows. The length of the input array should be equal to row length minus column_names count. The length of array generated by the function should be the row length. :returns: A new :class:`.Table`. c3HK|]}tj| ywN)r issequence).0 compare_values [/var/www/html/strategist-ai/venv_dbt/lib/python3.12/site-packages/agate/table/homogenize.py zhomogenize..2sW}5##M22Ws "N) row_names)list_rowsrr lenany_columnsget _column_namesindex enumerate data_typecastzipsetmaptuplecallableinsertappendr_fork)selfkeycompare_values default_rowrowsr name column_valuescolumn_indexesvaluesiv differences differencenew_rowds r homogenizer1sN  D   C e 3x1} WW WCQR-}oRNR9<=T]]&&t,=M=ADEd((..t4ENEguvv]ciPVFWXda}Q'1166q9XvNv'Ms3un56]9KKLK! 6 K !*-G&{+&C(:(:$;c#h$FGNJ7 %1q!$ %CLGBTU$!Q4==#--2215UU C!3!345 6 ::db: ))5S=EXv"Vs0 H""H# "H(4H3(H-/H3/2H:-H3r)agater agate.rowsrr1r r6sH*r5