B q\+@sddlZddlZddlmZdddgZGdddeZGdd d eZd d Z dd d Z ddZ ddZ ddZ ddZddZGdddeZddZdS)N) OrderedDictget_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. cstddddddg}t|}g}x&|D]}||kr,||||fq,Wx&|D]\}}||krT|||fqTW|dd=||dS)Nnameunitdatatypeformat descriptionmeta)supersortdictappendextend)selfargskwargsZ column_keysZin_dictZout_listkeyval) __class__1lib/python3.7/site-packages/astropy/table/meta.pyrs   zColumnOrderList.sort)__name__ __module__ __qualname____doc__r __classcell__rr)rrr 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)r)rrrr *szColumnDict.items)rrrrr rrr)rrr#src csddl}t}|Vt||js>|jd|jd|j|jx|j D]}t||j st|jd|jd|j|jt |j dkr|jd|jdt |j |j|j d\}}| |}| |}|||<qFWdS)a Construct OrderedDict from !!omap in yaml safe load. Source: https://gist.github.com/weaver/317164 License: Unspecified This is the same as SafeConstructor.construct_yaml_omap(), except the data type is changed to OrderedDict() and setitem is used instead of append in the loop Examples -------- :: >>> 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')]) rNz!while constructing an ordered mapz!expected a sequence, but found {}z,expected a mapping of length 1, but found {}z2expected a single mapping item, but found {} items) yamlr isinstance SequenceNode constructorZConstructorErrorZ start_markr idvalue MappingNodelenZconstruct_object) loadnoder"ZomapZsubnodeZkey_nodeZ value_noderr'rrr_construct_odict2s,      r,c Csddl}g}|j|||d}|jdk r2||j|j<d}x>|D]6\}} ||| i} t| |jrd| jrhd}|| q>> 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' ztag:yaml.org,2002:omap)r8r )dumperdatarrr _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_mapping)r9r:rrr_repr_column_dictsr=cCst}|jj|d<|jjjj}|dr,d}|drB|dd}||d<xjdd d tfd d d dfd dd dfddd dffD]2\}}}t |j|}||r|r||n|||<qW|S)zw Extract information from a column (apart from the values) that is required to fully serialize the column. r)bytesstrstring_Nr rcSs|dk S)Nr)xrrrsz%_get_col_attributes..r cSs|dk S)Nr)rCrrrrDsr cSs|dk S)Nr)rCrrrrDsr cSs|S)Nr)rCrrrrDs) rinforZdtypetyper startswithendswithr?getattr)colZattrsZ type_nameattrZ nontrivialZxformZcol_attrrrr_get_col_attributess         rLcCs*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 )listcolumnsvaluesr r)tableheaderrrrrs csy ddlWntk r(tdYnXddlm}Gfddd|}|tt|ttt |}dd|d D|d <|d =j ||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 rNzN`import yaml` failed, PyYAML package is required for serializing mixin columns) AstropyDumpercseZdZdZdfdd ZdS)z)get_yaml_from_header..TableDumperzP Custom Dumper that represents OrderedDict as an !!omap object. Nc sg}j|||d}|jdk r*||j|j<d}t|dr~|}t|drT|n*t|}y t|}Wntk r|YnXx^|D]V\}}| |} | |} t | j r| j rd}t | j r| j rd}| | | fqW|dkr|jdk r|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.r/hasattrr rrNsorted TypeErrorr0r#r1r2rr3r-) rr5mappingr-r'r+r6Zitem_keyZ item_valueZnode_keyZ node_value)r"rrr<s6           z;get_yaml_from_header..TableDumper.represent_mapping)N)rrrrr<r)r"rr TableDumpersrXcSsg|] }t|qSr)rL).0rJrrr sz(get_yaml_from_header..rMr )ZDumperwidth) r" ImportErrorastropy.io.misc.yamlrSZadd_representerrr;rr=copyr4 splitlines)rRrSrXlinesr)r"rrs  ,   c@s eZdZdS)YamlParseErrorN)rrrrrrrrb&srbc Csy ddl}Wntk r(tdYnXddlm}Gddd|}|dttd|}y|j ||d }Wn.t k r}zt t |Wdd}~XYnX|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 rNzN`import yaml` failed, PyYAML package is required for serializing mixin columns) 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. N)rrrrrrrr TableLoaderHsrdztag:yaml.org,2002:omap )Loader) r"r]r^rcZadd_constructorr,textwrapdedentjoinr* Exceptionrbr?)rar"rcrdZ header_yamlrRerrrrrr*s   )N)rgr_ collectionsr__all__rNrrrr,r8r;r=rLrrrjrbrrrrrs  6  U