IR-ey6ddlZddlZddlZddlmZddlZddlZgdZGdde Z Gdde Z dZ dd Zd Zd Zd Zd ZdZdZdZGddeZdZdS)N) OrderedDict)get_header_from_yamlget_yaml_from_headerget_yaml_from_tablec"eZdZdZfdZxZS)ColumnOrderListzq List of tuples that sorts in a specific order that makes sense for astropy table column attributes. c8tgd}t|}g}|D]#}||vr||||f$|D] \}}||vr|||f!|dd=||dS)N)nameunitdatatypeformat descriptionmeta)supersortdictappendextend) selfargskwargs column_keysin_dictout_listkeyval __class__s 2lib/python3.11/site-packages/astropy/table/meta.pyrzColumnOrderList.sorts  SSS t** 5 5Cg~~gcl 3444 , ,HC+%%c +++ G H)__name__ __module__ __qualname____doc__r __classcell__rs@rrr sB rrc"eZdZdZfdZxZS) 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. c^ttS)zt Return items as a ColumnOrderList, which sorts in the preferred way for column attributes. )rritems)rrs rr)zColumnDict.items-s uww}}///r)r r!r"r#r)r$r%s@rr'r'&sB 000000000rr'c #Kt}|Vt|tjs4tjd|jd|j|j|jD]}t|tj s4tjd|jd|j|jt|jdkrBtjd|jdt|jd|j|jd\}}| |}| |}|||<dS) 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')]) 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 constructorConstructorError start_markidvalue MappingNodelenconstruct_object)loadnodeomapsubnodekey_node value_noderr3s r_construct_odictr=5sh4 ==D JJJ dD- . . // / O 7dg 7 7 O    :'4#344 "333IWZII"   w}   " ""333WS=O=OWWW"   '}Q/*##H--%%j11S )rcPg}tj|||}|j||j|j<d}|D]T\}}|||i} t | tjr| jrd}|| U||j |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@r3r8 best_stylerritems r _repr_pairsrLps E  S%J ? ? ?D ~!37 0JS""C:..411 $* J T  " ."5DOO(DO KrcHt|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)rLr)dumperdatas r _repr_odictrRs v7 F FFrc.|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_mappingrOs r_repr_column_dictrUs  # #$;T B BBrcGddt}tjtjtjtjf} t |dkstd|Ds||dj}|dj dd}|D]K}t|jj |r|j dd|kr|tj ||j}L|dz}n#|$r|j j}d}YnwxYw||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 ceZdZdZdS)6_get_variable_length_array_shape..ConvertErrorz"Local conversion error used below.Nr r!r"r#rr ConvertErrorrXs0000rr[rc3JK|]}t|tjVdSN)r,npndarray).0rs r z3_get_variable_length_array_shape..s.#O#OCJsBJ$?$?#O#O#O#O#O#OrNr]rZ) ValueErrorr^floatingintegerbool_unicode_r5alldtypeshape issubclasstype promote_typesinfo)colr[ np_classesrirjrs r _get_variable_length_array_shaperqs3411111z111+rz28R[AJ s88q==#O#O3#O#O#O O O= A A SbS! 7 7Ccinj99 #SYss^u=T=T""$UCI66EE   %<sB#C''C=<C=c|j}|drd}|dr |dd}|S)zDReturn string version of ``dtype`` for writing to ECSV ``datatype``.)bytesstrstring_Nrb)r startswithendswith)rir s r_get_datatype_from_dtyperysMzH+,,!CRC= Orc(|jj}d}|jdd}|jdkre|dkr,t jt }t |\}}nZt jt }t jt}n'|r%t jt }|jj}t|}t}|jj|d<||d<ddt fd d dfd d dfd ddffD]8\}}}t|j|} || r|r || n| ||<9|r#t||d<|ddkrd|d<|r1|dxxtj t|dz cc<|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+objectrZr r r c |duSr]rZxs rz%_get_col_attributes.. s 1D=rr c |duSr]rZr}s rrz%_get_col_attributes..s Qd]rrc |duSr]rZr}s rrz%_get_col_attributes..s !4-rrc|Sr]rZr}s rrz%_get_col_attributes..s1rsubtypejson),:) separators)rnrirjr r^rtrqr{ryr'getattrrdumpslist) rorirrjr attrsattr nontrivialxformcol_attrs r_get_col_attributesrs HNEG IabbME zX B;;HSMME=cBBNE77HSMMEhv&&GG ! (.'..H LLEHME&M E* ((#. **D1 //6 d# $AAj% 38T** :h   A-2@%%///E$K&3G<<i  x ' '%E)  K iDJtE{{zJJJJ Lrcdt|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)rcolumnsvaluesrr)tableheaders rrr!sGd5=//1122 3F z$v  ' ''rcZddlm}Gdd|}|tt|t t tj|}d|dD|d<|d=tj |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) AstropyDumperceZdZdZddZdS))get_yaml_from_header..TableDumperzP Custom Dumper that represents OrderedDict as an !!omap object. Ncg}tj|||}|j||j|j<d}t |dri|}t |dr|n0t|} t|}n#t$rYnwxYw|D]\}}| |} | |} t| tj r| j rd}t| tj r| j rd}|| | f||j |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-r4rArBhasattrr)rrsorted TypeErrorrCr,rDrErrFr@) rrHmappingr@r3r8rJitem_key item_valuenode_key node_values rrTz;get_yaml_from_header..TableDumper.represent_mappingSs~E#C:FFFD~);?(8Jw(( !--//7F++LLNNNN"7mmG"(//$)0 5 5$*..x88!00<< "8T_=='hn'!&Jz4?;;'DNDT'"'J h 34444!*6&*&=DOO&0DOKs B B('B(r])r r!r"r#rTrZrr TableDumperrNs2  ' ' ' ' ' ' rrc,g|]}t|SrZ)r)r`ros r z(get_yaml_from_header..s!MMMs-c22MMMrrr N)rFDumperwidth) astropy.io.misc.yamlradd_representerrrRr'rUcopyr-rG splitlines)rrrliness rrr6s,322222,,,,,m,,,\ [999 ,=>>> Yv  FMMfVnMMMF:v I4 3   jll  LrceZdZdS)YamlParseErrorN)r r!r"rZrrrrsDrrc$ddlm}Gdd|}|dtt jd|} tj||}n!#t$r}t|d}~wwxYw|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) AstropyLoaderceZdZdZdS))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. NrYrZrr TableLoaderrs    rrrN )LoaderN) rradd_constructorr=textwrapdedentjoinr-r7 Exceptionr)rrr header_yamlrerrs rrrs*322222     m    8:JKKK/$))E"2"233K(;{;;; (((C'( MsA// B 9BB r])rrr collectionsrnumpyr^r-__all__rrrr'r=rLrRrUrqryrrrrrrrZrrrs ###### Q Q Qd4 0 0 0 0 0 0 0 0888v6 G G G CCC222j999x(((*PPPf     Y   &&&&&r