B q\7*@sddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z mZdd lmZd ZGd d d eZdddZdddZddZGdddeZddZddZdS)) import_moduleN)deepcopy) OrderedDict) MixinInfo)Column)TableQTablehas_info_class) QuantityInfo) zastropy.time.core.Timezastropy.time.core.TimeDeltazastropy.units.quantity.Quantityz#astropy.coordinates.angles.Latitudez$astropy.coordinates.angles.Longitudez astropy.coordinates.angles.Anglez&astropy.coordinates.distances.Distancez'astropy.coordinates.earth.EarthLocationz+astropy.coordinates.sky_coordinate.SkyCoordz astropy.table.table.NdarrayMixinz!astropy.table.column.MaskedColumnc@seZdZdZdS)SerializedColumnaF Subclass of dict that is a used in the representation to contain the name (and possible other info) for a mixin attribute (either primary data or an array-like attribute) that is serialized as a column in the table. Normally contains the single key ``name`` with the name of the column in the table. N)__name__ __module__ __qualname____doc__rr6lib/python3.7/site-packages/astropy/table/serialize.pyr sr rcsjjj}r j|kr.|dSi}xjdddtfddddfddddfd d ddffD]2\}}} tj|} || rh| r| | n| ||<qhWfd d |D} x| D]} | } | jjkr|}n |d | }t| t s|t | fd|i|t d|i| <n t | ||t  || <x"jjD]}||krD||=qDW|r|d<qWjd jjd<||<dS)aCarry out processing needed to serialize ``col`` in an output table consisting purely of plain ``Column`` or ``MaskedColumn`` columns. This relies on the object determine if any transformation is required and may depend on the ``serialize_method`` and ``serialize_context`` context variables. For instance a ``MaskedColumn`` may be stored directly to FITS, but can also be serialized as separate data and mask columns. This function builds up a list of plain columns in the ``new_cols`` arg (which is passed as a persistent list). This includes both plain columns from the original table and plain columns that represent data from serialized columns (e.g. ``jd1`` and ``jd2`` arrays from a ``Time`` column). For serialized columns the ``mixin_cols`` dict is updated with required attributes and information to subsequently reconstruct the table. Table mixin columns are always serialized and get represented by one or more data columns. In earlier versions of the code *only* mixin columns were serialized, hence the use within this code of "mixin" to imply serialization. Starting with version 3.1, the non-mixin ``MaskedColumn`` can also be serialized. NunitcSs|dk o|dkS)Nr)xrrrNsz,_represent_mixin_as_column..formatcSs|dk S)Nr)rrrrrOs descriptioncSs|dk S)Nr)rrrrrPsmetacSs|S)Nr)rrrrrQscs>g|]6}|krt|ddddjddkr|qS)shaperNr)getattrr).0key)col obj_attrsrr Vs z._represent_mixin_as_column...name__info__ __class__)infoZ_represent_as_dictZ_represent_as_dict_attrsr$appendstrrZ_represent_as_dict_primary_datar rrr _represent_mixin_as_columnpopattrs_from_parentrr )rr"new_cols mixin_colsexclude_classesZ ordered_keysr%attr nontrivialZxformcol_attrZ data_attrs data_attrdatanew_namer)rrrr(%s>             r(cCs\i}g}x&|D]}t||jj|||dqW|s8|St|j}||d<t||dd}|S)z Convert any mixin columns to plain Column or MaskedColumn and return a new table. Exclude any mixin columns in ``exclude_classes``, which must be a tuple of classes. )r-__serialized_columns__F)rcopy)itercolsr(r%r"rrr)tblr-r,r+rroutrrr_represent_mixins_as_columns{s  r9c Cs|d}|tkr td|td|\}}t|}t||}x(| D]\}}||j j krP|||<qPW|j |} x*| D]\}}||krt | j ||qW| S)Nr$z"unsupported class for construct {}z (.+)\.(\w+))r)__construct_mixin_classes ValueErrorrrematchgroupsrritemsr%r*Z_construct_from_dictsetattr) rr%Z cls_full_namemod_nameZcls_namemoduleclsr.valueZmixinrrr(_construct_mixin_from_obj_attrs_and_infos     rEc@s.eZdZdZd ddZeddZddZd S) _TableLitea Minimal table-like object for _construct_mixin_from_columns. This allows manipulating the object like a Table but without the actual overhead for a full Table. More pressing, there is an issue with constructing MaskedColumn, where the encoded Column components (data, mask) are turned into a MaskedColumn. When this happens in a real table then all other columns are immediately Masked and a warning is issued. This is not desirable. rcCs>|j}|||jj<x&t|D]\}}||kr||qWdS)N)colnamesr%r" enumerate move_to_end)selfrindexrGZiir"rrr add_columns  z_TableLite.add_columncCs t|S)N)listkeys)rJrrrrGsz_TableLite.colnamescCs|S)N)values)rJrrrr6sz_TableLite.itercolsN)r)r rrrrLpropertyrGr6rrrrrFs   rFc s<i}xH|D]<\}}t|trd|kr6|||d<qt|||||<qWx|D] }||=qXWtfdd|D}x*|D]\}}|}|||<|=qW|di} t|dkrxTdddfd d dfd d dfd ddffD]$\} } t|j | } | | r| | | <qW|| d<t || }j ||ddS)Nr"c3s|]}j|VqdS)N)rGrK)rr")r8rr sz0_construct_mixin_from_columns..r#rrcSs|dkS)N)Nrr)rrrrrsz/_construct_mixin_from_columns..rcSs|dk S)Nr)rrrrrsrcSs|dk S)Nr)rrrrrsrcSs|S)Nr)rrrrrs)rK) r? isinstancer _construct_mixin_from_columnsrOminr)lenrr%rErL) r3rr8Zdata_attrs_mapr"validxr1rr%r.r/r0r)r8rrSs4            rScCsd|jkr|S|j}|d}t|j}x |D]\}}t|||q6Wtdd|D}|rlt nt }|t | |j d|dS)Nr4css|]}t|jtVqdS)N)rRr%r )rrrrrrQsz1_construct_mixins_from_columns..F)namesr5r)rr5r)rFcolumnsr?rSanyr6r rrMrOrG)r7rr,r8r3rZhas_quantitiesZout_clsrrr_construct_mixins_from_columnss     r[)r)r) importlibrr<r5r collectionsrZastropy.utils.data_inforcolumnrtablerr r Zastropy.units.quantityr r:dictr r(r9rErFrSr[rrrrs        U *