U e{jC@sdZddlmZmZddlmZmZGdddejZGdddeZGdd d ej Z Gd d d ej Z Gd d d ej ZGddde ZGddde ZGdddeZGddde ZGdddeZGddde ZGdddeZdS)zAn extensible ASCII table reader and writer. fixedwidth.py: Read or write a table with fixed width columns. :Copyright: Smithsonian Astrophysical Observatory (2011) :Author: Tom Aldcroft (aldcroft@head.cfa.harvard.edu) )basiccore)DefaultSplitterInconsistentTableErrorc@s,eZdZdZdZdZdZddZddZd S) FixedWidthSplittera Split line based on fixed start and end positions for each ``col`` in ``self.cols``. This class requires that the Header class will have defined ``col.start`` and ``col.end`` for each column. The reference to the ``header.cols`` gets put in the splitter object by the base Reader.read() function just in time for splitting data lines by a ``data`` object. Note that the ``start`` and ``end`` positions are defined in the pythonic style so line[start:end] is the desired substring for a column. This splitter class does not have a hook for ``process_lines`` since that is generally not useful for fixed-width input. F|c#sD|D]:fddjD}jr8fdd|DVq|VqdS)Ncsg|]}|j|jqS)startend.0xliner Z/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/ascii/fixedwidth.py 'sz/FixedWidthSplitter.__call__..csg|]}|qSr ) process_valr selfr rr)s)colsr)rlinesvalsr )rrr__call__%s zFixedWidthSplitter.__call__cCsf|jpd}|jpd}|||}|jr8||}||}nd}d}ddt||D}||||S)NrcSs$g|]\}}d|t||qS) len)r valwidthr r rr7sz+FixedWidthSplitter.join..) delimiter_pad delimiterbookendzipjoin)rrwidthspadr Z padded_delimZ bookend_leftZ bookend_rightr r rr#-s    zFixedWidthSplitter.joinN) __name__ __module__ __qualname____doc__rr!r rr#r r r rrs rc@seZdZdZdZdS)FixedWidthHeaderSplitterz$Splitter class that splits on ``|``.rNr&r'r(r)r r r r rr*;sr*c@s@eZdZdZeZdZedZddZ ddZ dd Z d d Z dS) FixedWidthHeaderz* Fixed width table header reader. Nz`~!#$%^&*-_+=\|":'cCs0t||D]\}}||krq,qtd|S)NzNo header line found in table) enumerate process_linesr)rrindexirr r rget_lineLs zFixedWidthHeader.get_linecstddg}tj|}tj|}|dkr|dk rNtdjj}|sbt d |d\}}}fddt d t |d D_ n|dk r(||} t t| jjd hd krt d jjjd h} t| | st d |  | \}_} d d| D_|||d} | \_ }}t|D]\\} } | dkr`||| } | d}tj|D]\}}|rt|| |qq`tjD]\}}|||_|||_qdS)a: Initialize the header Column objects from the table ``lines``. Based on the previously set Header attributes find or create the column names. Sets ``self.cols`` with the list of Columns. Parameters ---------- lines : list List of table lines header_rowsnameNz:Cannot set position_line without also setting header_startz7No data lines found so cannot autogenerate column namescsg|]}j|qSr )Z auto_formatformat)r r0rr rr|sz-FixedWidthHeader.get_cols..rrz]Position line should only contain delimiters and one other character, e.g. "--- ------- ---".z,Characters in position line must be part of cSs g|]}|dk r|dndqSNrr r r r rrs)getattrrZ_get_line_index start_liner. position_line ValueErrordata data_linesrget_fixedwidth_paramsrangernamesr1setsplitterr set_of_position_line_charactersunionissubset col_startscol_endsr/Z_set_cols_from_namesr-r"rsetattrr r )rrr2r8r9r<rstartsendsrcharsetrFiiattrcolrr0r rrget_colsTs\ $      zFixedWidthHeader.get_colscs|jdk rb|jdk rbt|j}dd|jD}t|t|krFtdfddt||D}n$|jdkr"|jdkr"|jj}dg}g}|D]B}|r| |dt|| |ddq|dd7<q|dd}d d|D}t|t|kst|t|krt d nd|jdk rLt|j}|dddg}n"d d|jD}dg|dd}fd dt||D}|||fS) a Split ``line`` on the delimiter and determine column values and column start and end positions. This might include null columns with zero length (e.g. for ``header row = "| col1 || col2 | col3 |"`` or ``header2_row = "----- ------- -----"``). The null columns are stripped out. Returns the values between delimiters and the corresponding start and end positions. Parameters ---------- line : str Input line Returns ------- vals : list List of values. starts : list List of starting indices. ends : list List of ending indices. NcSs g|]}|dk r|dndqSr6r r r r rrsz:FixedWidthHeader.get_fixedwidth_params..z=Fixed width col_starts and col_ends must have the same lengthcs g|]\}}||qSr stripr r r rr rrsr4rcSsg|]}|r|qSr rOr r r rrsz Error parsing fixed width headercSsg|] }|dqS)rr r r r rrscs g|]\}}||qSr rOrQrr rrs) rErFlistrr:r"splitrAr appendr)rrrHrIrrr rrr=s8  $   z&FixedWidthHeader.get_fixedwidth_paramscCsdSNr )rrr r rwriteszFixedWidthHeader.write) r&r'r(r)r*splitter_classr9r@rBr1rNr=rWr r r rr,AsaAr,c@s eZdZdZeZdZddZdS)FixedWidthDataz! Base table data reader. NcsBjjdkrgndg}td|}g}t|D]}|q6g}|D]$fddjD}||qNfddttjD}|rttjD](t |t fdd|D|<q|D]}|j ||qЈjj dk rfdd|D}|j ||D]}|j ||q"|S) Nr3r2cs,g|]$}t|jdkr dntqS)Nr)r7infostr)r rM)col_attrrr rrsz(FixedWidthData.write..cs"g|]tfddDqS)c3s|]}t|VqdSrVrr ri_colr r sz2FixedWidthData.write...)max)r ) vals_listr^rrsc3s|]}t|VqdSrVrr]r^r rr`!sz'FixedWidthData.write..csg|]}jj|qSr )header position_char)r rrr rr)s) rcr8r7Zstr_valsr"rUrr>rrarAr#r9)rrZdefault_header_rowsr2Z col_str_itersrZ hdrs_listr$r )r\r_rrrbrrWs:        zFixedWidthData.write)r&r'r(r)rrXr8rWr r r rrYsrYcs2eZdZdZdZdZeZeZ d fdd Z Z S) FixedWidtha_Fixed width table with single header line defining column names and positions. Examples:: # Bar delimiter in header and data | Col1 | Col2 | Col3 | | 1.2 | hello there | 3 | | 2.4 | many words | 7 | # Bar delimiter in header only Col1 | Col2 | Col3 1.2 hello there 3 2.4 many words 7 # No delimiter with column positions specified as input Col1 Col2Col3 1.2hello there 3 2.4many words 7 See the :ref:`astropy:fixed_width_gallery` for specific usage examples. Z fixed_widthz Fixed widthNrTcsh|dkrdg}t||jj_||jj_||j_||j_||j_ ||j_ |jj dkrdt ||j_ dS)Nr3) super__init__r;rArr!rcrErFr2r8r)rrErFrr!r2 __class__r rrgSs    zFixedWidth.__init__)NNrTN) r&r'r(r) _format_name _descriptionr, header_classrY data_classrg __classcell__r r rhrre2srec@seZdZdZdZdS)FixedWidthNoHeaderHeaderz7Header reader for fixed with tables with no header lineNr&r'r(r)r8r r r rrohsroc@seZdZdZdZdS)FixedWidthNoHeaderDataz6Data reader for fixed width tables with no header liner4Nrpr r r rrqnsrqcs2eZdZdZdZdZeZeZ d fdd Z Z S) FixedWidthNoHeadera?Fixed width table which has no header line. When reading, column names are either input (``names`` keyword) or auto-generated. Column positions are determined either by input (``col_starts`` and ``col_stops`` keywords) or by splitting the first data line. In the latter case a ``delimiter`` is required to split the data line. Examples:: # Bar delimiter in header and data | 1.2 | hello there | 3 | | 2.4 | many words | 7 | # Compact table having no delimiter and column positions specified as input 1.2hello there3 2.4many words 7 This class is just a convenience wrapper around the ``FixedWidth`` reader but with ``header_start=None`` and ``data_start=0``. See the :ref:`astropy:fixed_width_gallery` for specific usage examples. Zfixed_width_no_headerzFixed width with no headerNrTcstj||||gddS)Nrr!r2)rfrg)rrErFrr!rhr rrgszFixedWidthNoHeader.__init__)NNrT) r&r'r(r)rjrkrorlrqrmrgrnr r rhrrrts rrc@seZdZdZeZdS)FixedWidthTwoLineHeaderzHeader reader for fixed width tables splitting on whitespace. For fixed width tables with several header lines, there is typically a white-space delimited format line, so splitting on white space is needed. N)r&r'r(r)rrXr r r rrtsrtc@seZdZdZdZdS)FixedWidthTwoLineDataSplitterz5Splitter for fixed width tables splitting on ``' '``.rNr+r r r rrusruc@seZdZdZeZdS)FixedWidthTwoLineDataz8Data reader for fixed with tables with two header lines.N)r&r'r(r)rurXr r r rrvsrvcs2eZdZdZdZdZeZeZ d fdd Z Z S) FixedWidthTwoLineaFixed width table which has two header lines. The first header line defines the column names and the second implicitly defines the column positions. Examples:: # Typical case with column extent defined by ---- under column names. col1 col2 <== header_start = 0 ----- ------------ <== position_line = 1, position_char = "-" 1 bee flies <== data_start = 2 2 fish swims # Pretty-printed table +------+------------+ | Col1 | Col2 | +------+------------+ | 1.2 | "hello" | | 2.4 | there world| +------+------------+ See the :ref:`astropy:fixed_width_gallery` for specific usage examples. Zfixed_width_two_linez#Fixed width with second header lineN-Fcsbt|dkrtd|dtj|||d|dkrBt|jj}||j_||j_|d|j_ dS)NrzPosition_char="z" must be a single characterrs) rr:rfrgrcr2r9rdr;r8)rr9rdrr!r2rhr rrgs   zFixedWidthTwoLine.__init__)NrxNFN) r&r'r(r)rjrkrvrmrtrlrgrnr r rhrrwsrwN)r)rrrrrZ BaseSplitterrr*Z BasicHeaderr,Z BasicDatarYZBasicrerorqrrrtrurvrwr r r rs +<66+