a ߙfbB6@sddlZddlZddlZddlmZddlZddlZgdZGddde Z Gddde Z dd Z dd d Zd d ZddZddZddZddZddZddZGdddeZddZdS)N) OrderedDict)get_header_from_yamlget_yaml_from_headerget_yaml_from_tablecs eZdZdZfddZZS)ColumnOrderListzq List of tuples that sorts in a specific order that makes sense for astropy table column attributes. cs~tgd}t|}g}|D]}||vr"||||fq"|D]\}}||vrF|||fqF|dd=||dS)N)nameunitdatatypeformat descriptionmeta)supersortdictappendextend)selfargskwargsZ column_keysZin_dictZout_listkeyval __class__1lib/python3.9/site-packages/astropy/table/meta.pyrs   zColumnOrderList.sort)__name__ __module__ __qualname____doc__r __classcell__rrrrr srcs eZdZdZfddZZS) ColumnDictz Specialized dict subclass to represent attributes of a Column and return items() in a preferred order. This is only for use in generating a YAML map representation that has a fixed order. csttS)zt Return items as a ColumnOrderList, which sorts in the preferred way for column attributes. )rr items)rrrrr!-szColumnDict.items)rrrrr!rrrrrr &sr ccst}|Vt|tjs6tjd|jd|j|j|jD]}t|tj sjtjd|jd|j|jt |jdkrtjd|jdt |jd|j|jd\}}| |}| |}|||<q>> yaml.load(''' # doctest: +SKIP ... !!omap ... - foo: bar ... - mumble: quux ... - baz: gorp ... ''') OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')]) >>> yaml.load('''!!omap [ foo: bar, mumble: quux, baz : gorp ]''') # doctest: +SKIP OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')]) z!while constructing an ordered mapzexpected a sequence, but found z*expected a mapping of length 1, but found z*expected a single mapping item, but found z itemsrN) r isinstanceyaml SequenceNode constructorZConstructorErrorZ start_markidvalue MappingNodelenZconstruct_object)loadnodeZomapZsubnodeZkey_nodeZ value_noderr(rrr_construct_odict5s0      r-c Csg}tj|||d}|jdur*||j|j<d}|D]6\}}|||i} t| tjrZ| jr^d}|| q2|dur|j dur|j |_ n||_ |S)z This is the same code as BaseRepresenter.represent_sequence(), but the value passed to dump.represent_data() in the loop is a dictionary instead of a tuple. Source: https://gist.github.com/weaver/317164 License: Unspecified  flow_styleNTF) r$r% alias_keyrepresented_objectsrepresent_datar# ScalarNodestylerdefault_flow_styler/) dumptagZsequencer/r(r, best_stylerritemrrr _repr_pairsis       r:cCst|d|S)a Represent OrderedDict in yaml dump. Source: https://gist.github.com/weaver/317164 License: Unspecified >>> data = OrderedDict([('foo', 'bar'), ('mumble', 'quux'), ('baz', 'gorp')]) >>> yaml.dump(data, default_flow_style=False) # doctest: +SKIP '!!omap\n- foo: bar\n- mumble: quux\n- baz: gorp\n' >>> yaml.dump(data, default_flow_style=True) # doctest: +SKIP '!!omap [foo: bar, mumble: quux, baz: gorp]\n' tag:yaml.org,2002:omap)r:r!Zdumperdatarrr _repr_odicts r>cCs |d|S)z Represent ColumnDict in yaml dump. This is the same as an ordinary mapping except that the keys are written in a fixed order that makes sense for astropy table columns. ztag:yaml.org,2002:map)represent_mappingr<rrr_repr_column_dictsr@cCsGdddt}tjtjtjtjf}zt|dksDtdd|DsH||dj}|dj dd}|D]6}t |jj |r|j dd|kr|t ||j}qh|d}Wn|y|j j}d }Yn0||fS) aLCheck if object-type ``col`` is really a variable length list. That is true if the object consists purely of list of nested lists, where the shape of every item can be represented as (m, n, ..., *) where the (m, n, ...) are constant and only the lists in the last axis have variable shape. If so the returned value of shape will be a tuple in the form (m, n, ..., None). If ``col`` is a variable length array then the return ``dtype`` corresponds to the type found by numpy for all the individual values. Otherwise it will be ``np.dtype(object)``. Parameters ========== col : column-like Input table column, assumed to be object-type Returns ======= shape : tuple Inferred variable length shape or None dtype : np.dtype Numpy dtype that applies to col c@seZdZdZdS)z6_get_variable_length_array_shape..ConvertErrorz!Local conversion error used belowNrrrrrrrr ConvertErrorsrBrcss|]}t|tjVqdSN)r#npZndarray).0rrrr z3_get_variable_length_array_shape..NrCr) ValueErrorrDZfloatingZintegerZbool_Zunicode_r*alldtypeshape issubclasstypeZ promote_typesinfo)colrBZ np_classesrKrLrrrr _get_variable_length_array_shapes      rQcCs.|j}|drd}|dr*|dd}|S)zCReturn string version of ``dtype`` for writing to ECSV ``datatype``)bytesstrstring_NrH)r startswithendswith)rKr rrr_get_datatype_from_dtypes    rXc CsJ|jj}d}|jdd}|jdkrZ|dkrDtt}t|\}}qptt}tt}n|rptt}|jj}t|}t }|jj|d<||d<ddd tfd d d dfd d d dfddd dffD]2\}}}t |j|} || r|r|| n| ||<q|r"t||d<|ddkr"d|d<|rF|dt j t |dd7<|S)a Extract information from a column (apart from the values) that is required to fully serialize the column. Parameters ---------- col : column-like Input Table column Returns ------- attrs : dict Dict of ECSV attributes for ``col`` Nr"objectrrr rcSs|duSrCrxrrrrGz%_get_col_attributes..r cSs|duSrCrrZrrrr\rGr cSs|duSrCrrZrrrr\rGr cSs|SrCrrZrrrr\rGsubtypejson),:)Z separators)rOrKrLrrDrSrQrYrXr getattrr^dumpslist) rPrKr]rLr attrsattrZ nontrivialZxformZcol_attrrrr_get_col_attributess>            rfcCs*dt|ji}|jr"|j|d<t|S)a/ Return lines with a YAML representation of header content from the ``table``. Parameters ---------- table : `~astropy.table.Table` object Table for which header content is output Returns ------- lines : list List of text lines with YAML header content colsr )rccolumnsvaluesr r)tableheaderrrrrs rcCstddlm}Gddd|}|tt|ttt|}dd|dD|d<|d=tj |d |d d  }|S) a Return lines with a YAML representation of header content from a Table. The ``header`` dict must contain these keys: - 'cols' : list of table column objects (required) - 'meta' : table 'meta' attribute (optional) Other keys included in ``header`` will be serialized in the output YAML representation. Parameters ---------- header : dict Table header content Returns ------- lines : list List of text lines with YAML header content r) AstropyDumperc@seZdZdZdddZdS)z)get_yaml_from_header..TableDumperzP Custom Dumper that represents OrderedDict as an !!omap object. Nc Ssg}tj|||d}|jdur*||j|j<d}t|dr||}t|drT|n(t|}z t|}Wnt yzYn0|D]V\}}| |} | |} t | tj r| j rd}t | tj r| j rd}|| | fq|dur|jdur|j|_n||_|S)a| This is a combination of the Python 2 and 3 versions of this method in the PyYAML library to allow the required key ordering via the ColumnOrderList object. The Python 3 version insists on turning the items() mapping into a list object and sorting, which results in alphabetical order for the column keys. r.NTr!rF)r$r)r0r1hasattrr!rrcsorted TypeErrorr2r#r3r4rr5r/) rr7mappingr/r(r,r8Zitem_keyZ item_valueZnode_keyZ node_valuerrrr?Js6            z;get_yaml_from_header..TableDumper.represent_mapping)N)rrrrr?rrrr TableDumperEsrqcSsg|] }t|qSr)rf)rErPrrr urGz(get_yaml_from_header..rgr N)r5ZDumperwidth) astropy.io.misc.yamlrlZadd_representerrr>r r@copyr$r6 splitlines)rkrlrqlinesrrrr-s ,    rc@s eZdZdS)YamlParseErrorN)rrrrrrrry}sryc Cs|ddlm}Gddd|}|dttd|}ztj||d}Wn,t yv}zt |WYd}~n d}~00|S) aS Get a header dict from input ``lines`` which should be valid YAML. This input will typically be created by get_yaml_from_header. The output is a dictionary which describes all the table and column meta. The get_cols() method in the io/ascii/ecsv.py file should be used as a guide to using the information when constructing a table using this header dict information. Parameters ---------- lines : list List of text lines with YAML header content Returns ------- header : dict Dictionary describing table and column meta r) AstropyLoaderc@seZdZdZdS)z)get_header_from_yaml..TableLoaderz Custom Loader that constructs OrderedDict from an !!omap object. This does nothing but provide a namespace for adding the custom odict constructor. NrArrrr TableLoadersr{r; )LoaderN) rurzZadd_constructorr-textwrapdedentjoinr$r+ Exceptionry)rxrzr{Z header_yamlrkerrrrrrs  r)N)r^r~rv collectionsrZnumpyrDr$__all__rcrrr r-r:r>r@rQrXrfrrrryrrrrrs& 4  4 :P