U e{jx6@sddlZddlZddlZddlmZddlZddlZdddgZGddde Z Gdd d e Z d d Z dd d ZddZddZddZddZddZddZddZGdddeZddZdS)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}|D]}||kr*||||fq*|D]\}}||krN|||fqN|dd=||dS)Nnameunitdatatypeformat descriptionmeta)supersortdictappendextend)selfargskwargsZ column_keysZin_dictZout_listkeyval __class__Q/var/www/html/bwave/python/venv/lib/python3.8/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_odict5s8       r-c Csg}tj|||d}|jdk r*||j|j<d}|D]6\}}|||i} t| tjrZ| jr^d}|| q2|dkr|j dk r|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/) dumptagsequencer/r(r, best_stylerritemrrr _repr_pairsps       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_dictsrAcCsGdddt}tjtjtjtjf}zt|dksDtdd|DsH||dj}|dj dd}|D]6}t |jj |r|j dd|kr|t ||j}qh|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 belowNrrrrrrrr ConvertErrorsrCrcss|]}t|tjVqdSN)r#npZndarray).0rrrr sz3_get_variable_length_array_shape..NrDr) ValueErrorrEZfloatingintegerZbool_Zunicode_r*alldtypeshape issubclasstypeZ promote_typesinfo)colrCZ np_classesrLrMrrrr _get_variable_length_array_shapes     rRcCs.|j}|drd}|dr*|dd}|S)zCReturn string version of ``dtype`` for writing to ECSV ``datatype``)bytesstrstring_NrH)r startswithendswith)rLr rrr_get_datatype_from_dtypes    rYc 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|dk SrDrxrrr z%_get_col_attributes..r cSs|dk SrDrr[rrrr]r^r cSs|dk SrDrr[rrrr]r^r cSs|SrDrr[rrrr]r^subtypejson),:) separators)rPrLrMrrErTrRrZrYr getattrr`dumpslist) rQrLr_rMr attrsattrZ nontrivialZxformZcol_attrrrr_get_col_attributess>            ricCs*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 )rfcolumnsvaluesr r)tableheaderrrrr!s cCstddlm}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}|jdk r*||j|j<d}t|dr~|}t|drT|n*t|}z t|}Wnt k r|YnX|D]V\}}| |} | |} t | tj r| j rd}t | tj r| j rd}|| | fq|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)r0r1hasattrr!rrfsorted TypeErrorr2r#r3r4rr5r/) rr7mappingr/r(r,r9Zitem_keyZ item_valueZnode_keyZ node_valuerrrr@Ts<            z;get_yaml_from_header..TableDumper.represent_mapping)N)rrrrr@rrrr TableDumperOsrtcSsg|] }t|qSr)ri)rFrQrrr sz(get_yaml_from_header..rjr N)r5ZDumperwidth) astropy.io.misc.yamlroZadd_representerrr?r rAcopyr$r6 splitlines)rnrortlinesrrrr7s .    c@s eZdZdS)YamlParseErrorN)rrrrrrrr|sr|c Cszddlm}Gddd|}|dttd|}ztj||d}Wn*t k rt}z t |W5d}~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 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. NrBrrrr TableLoadersr~r< )LoaderN) rxr}Zadd_constructorr-textwrapdedentjoinr$r+ Exceptionr|)r{r}r~Z header_yamlrnerrrrrrs  )N)ryr`r collectionsrnumpyrEr$__all__rfrrr r-r;r?rArRrYrirrrr|rrrrrs&  ;  5 <S