o ;Di@sddlmZmZmZddlZddlmZmZddlm Z m Z ddl m Z m Z   dddZGd d d e Zdd dZddZddZee _dddZee _dS))absolute_importprint_functiondivisionN)PY3 text_type)Tabledata)read_source_from_argwrite_source_from_argFc Ks t|f|||||||d|S)a~ Extract a table from a sheet in an Excel .xlsx file. N.B., the sheet name is case sensitive. The `sheet` argument can be omitted, in which case the first sheet in the workbook is used by default. The `range_string` argument can be used to provide a range string specifying a range of cells to extract. The `min_row`, `min_col`, `max_row` and `max_col` arguments can be used to limit the range of cells to extract. They will be ignored if `range_string` is provided. The `read_only` argument determines how openpyxl returns the loaded workbook. Default is `False` as it prevents some LibreOffice files from getting truncated at 65536 rows. `True` should be faster if the file use is read-only and the files are made with Microsoft Excel. Any other keyword arguments are passed through to :func:`openpyxl.load_workbook()`. )sheet range_stringmin_rowmin_colmax_rowmax_col read_only)XLSXView filenamer r r rrrrkwargsrE/var/www/Datamplify/venv/lib/python3.10/site-packages/petl/io/xlsx.pyfromxlsx s rc@s$eZdZ   dddZddZdS)rNFc Ks:||_||_||_||_||_||_||_||_| |_dSNr) selfrr r r rrrrrrrr__init__.s zXLSXView.__init__c cs$ddl}t|j}|dy}|jd||jd|j}|jdur*||jd}nt |jt r9||j|j}n|t |j}|j durK||j }n |j |j|j|j|jd}|D] }tdd|DVqZz|jWn tywYn wWddSWddS1swYdS)Nrrbrr)r rrrcss|]}|jVqdSr)value).0cellrrr Psz$XLSXView.__iter__..r)openpyxlr ropen load_workbookrrr sheetnames isinstanceintstrr iter_rowsr rrrtuple_archivecloseAttributeError)rr"sourcesource2wbwsrowsrowrrr__iter__;s@       "zXLSXView.__iter__NNNNNNF)__name__ __module__ __qualname__rr4rrrrr,s  rTreplacecCst|||}t|||}|r2t|}zt|}ttt|} t| g|} Wnt y1|} Ynwt |} | D]} | | q8t |} | d} || WddS1sZwYdS)a- Write a table to a new Excel .xlsx file. N.B., the sheet name is case sensitive. The `mode` argument controls how the file and sheet are treated: - `replace`: This is the default. It either replaces or adds a named sheet, or if no sheet name is provided, all sheets (overwrites the entire file). - `overwrite`: Always overwrites the file. This produces a file with a single sheet. - `add`: Adds a new sheet. Raises `ValueError` if a named sheet already exists. The `sheet` argument can be omitted in all cases. The new sheet will then get a default name. If the file does not exist, it will be created, unless `replace` mode is used with a named sheet. In the latter case, the file must exist and be a valid .xlsx file. r0N)_load_or_create_workbook_insert_sheet_on_workbookiternextlistmapr itertoolschain StopIterationrappendr r#save)tblrr write_headermoder0r1ithdrfldsr2r3targettarget2rrrtoxlsxXs$      "rMcCstrt}nt}ddl}d}|dksG|dkr|dusGz"t|}|d}|j|dd}Wdn1s5wYWn |yFd}Ynw|durQ|jdd}|S) Nr overwriter9rFrT) write_only)rFileNotFoundErrorIOErrorr"r r#r$Workbook)rrGr FileNotFoundr"r0r.r/rrrr:s$   r:cCs|dkr%z|t|}|d|jW|Sty$|j|d}Y|Sw|dkr@|j|d}|dur>|j|kr>td||S|dkrL|j|d}|Std|)Nr9)titleaddzSheet %s already exists in filerNzUnknown mode '%s')r( delete_rowsrKeyError create_sheetrU ValueError)rGr r0r1rrrr;s$       r;c Cs@ddl}t|}|d}|j|dd}|dur!||jd}nt|tr.||j|}n|t|}|rZt|} zt | } t t t | } t | g| } WntyY| } Ynwt|} | D]} || q`t|}|d }||Wdn1swYWddSWddS1swYdS)z7 Appends rows to an existing Excel .xlsx file. rNrFrr0)r"r r#r$r%r&r'r(r<r=r>r?rr@rArBrrCr rD)rErr rFr"r.r/r0r1rHrIrJr2r3rKrLrrr appendxlsxs8       "r[r5)NTr9)NF) __future__rrrr@ petl.compatrrpetl.util.baserrpetl.io.sourcesr r rrrMr:r;r[rrrrs ! ,+