B d7@sddlZddlZddlZddlmZddlZdddgZGdddeZ Gdd d e Z d d Z dd d Z ddZddZddZddZddZddZddZGdddeZddZdS)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___/work/yifan.wang/ringdown/master-ringdown-env/lib/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|jD]}t||j st|jd|jd|j|jt |jdkr|jd|jdt |jd|j|jd\}}| |}| |}|||<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 mapzexpected a sequence, but found z*expected a mapping of length 1, but found z*expected a single mapping item, but found z items) yamlr isinstance SequenceNode constructorZConstructorErrorZ start_markidvalue MappingNodelenZconstruct_object) loadnoder"ZomapZsubnodeZkey_nodeZ value_noderr'rrr_construct_odict4s,      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)r9r )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)r:r;rrr_repr_column_dictsr>cCsGdddt}tjtjtjtjf}yt|dksDtdd|DsH||dj}|dj dd}x>|D]6}t |jj |r|j dd|kr|t ||j}qjW|d}Wn |k r|j j}d }YnX||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 belowN)rrrrrrrr ConvertErrorsr?rcss|]}t|tjVqdS)N)r#npZndarray).0rrrr sz3_get_variable_length_array_shape..N)Nr) ValueErrorr@ZfloatingintegerZbool_Zunicode_r)alldtypeshape issubclasstypeZ promote_typesinfo)colr?Z np_classesrGrHrrrr _get_variable_length_array_shapes      rMcCs.|j}|drd}|dr*|dd}|S)zCReturn string version of ``dtype`` for writing to ECSV ``datatype``)bytesstrstring_NrC)r startswithendswith)rGr rrr_get_datatype_from_dtypes    rTc CsN|jj}d}|jdd}|jdkrZ|dkrDtt}t|\}}qptt}tt}n|rptt}|jj}t|}t }|jj|d<||d<xjddd tfd d d dfd d d dfddd dffD]2\}}}t |j|} || r|r|| n| ||<qW|r&t||d<|ddkr&d|d<|rJ|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|dk S)Nr)xrrrz%_get_col_attributes..r cSs|dk S)Nr)rVrrrrWrXr cSs|dk S)Nr)rVrrrrW rXr cSs|S)Nr)rVrrrrW rXsubtypejson),:) separators)rKrGrHrr@rOrMrUrTrgetattrrZdumpslist) rLrGrYrHr attrsattrZ nontrivialZxformZcol_attrrrr_get_col_attributess<          rccCs*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 )r`columnsvaluesr r)tableheaderrrrrs csy ddlWntk r(tdYnXddlm}Gfddd|}|tt|ttt |}dd|d D|d <|d =j |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 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 rr`sorted TypeErrorr0r#r1r2rr3r-) rr5mappingr-r'r+r7Zitem_keyZ item_valueZnode_keyZ node_value)r"rrr=Ss6           z;get_yaml_from_header..TableDumper.represent_mapping)N)rrrrr=r)r"rr TableDumperNsrncSsg|] }t|qSr)rc)rArLrrr ~sz(get_yaml_from_header..rdr )r3ZDumperwidth) r" ImportErrorastropy.io.misc.yamlriZadd_representerrr<rr>copyr4 splitlines)rhrirnlinesr)r"rr0s  ,   c@s eZdZdS)YamlParseErrorN)rrrrrrrrwsrwc 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 TableLoadersryztag:yaml.org,2002:omap )Loader) r"rrrsrxZadd_constructorr,textwrapdedentjoinr* ExceptionrwrO)rvr"rxryZ header_yamlrherrrrrrs   )N)rZr|rt collectionsrnumpyr@__all__r`rrrr,r9r<r>rMrTrcrrrrwrrrrrs$  6  4 :V