3
���h� � @ s\ d dl mZmZ d dlZd dlmZmZ d dlm Z d dl
Z d dl m
Z
G dd� de
�ZdS )� )�List�castN)�FilePathOrBuffer�Scalar)�import_optional_dependency)�_BaseExcelReaderc s� e Zd ZdZed�� fdd� Zedd� �Zed�dd�Zee d �d
d ��Z
ee e d �d d
��Z e
d�dd�Ze d�dd�Zee e e d�dd�Ze
d �dd�Ze
d �dd�Zed �dd �Zeed�d d �Ze d �d d �Z� ZS )!�
_ODFReaderz�
Read tables out of OpenDocument formatted files.
Parameters
----------
filepath_or_buffer: string, path to be parsed or
an open readable stream.
)�filepath_or_bufferc s t d� t� j|� d S )NZodf)r �super�__init__)�selfr )� __class__� �</tmp/pip-build-5_djhm0z/pandas/pandas/io/excel/_odfreader.pyr s z_ODFReader.__init__c C s ddl m} |S )Nr )� OpenDocument)�odf.opendocumentr )r r r r r �_workbook_class s z_ODFReader._workbook_classc C s ddl m} ||�S )Nr )�load)r r )r r r r r r �
load_workbook! s z_ODFReader.load_workbook)�returnc C s dS )z'Property for compat with other readers.� r )r r r r � empty_value& s z_ODFReader.empty_valuec C s& ddl m} | jj|�}dd� |D �S )z4Return a list of sheet names present in the documentr )�Tablec S s g | ]}|j d ��qS )�name)� getAttribute)�.0�tr r r �
<listcomp>1 s z*_ODFReader.sheet_names.<locals>.<listcomp>)� odf.tabler �book�getElementsByType)r r �tablesr r r � sheet_names+ s z_ODFReader.sheet_names)�indexc C s ddl m} | jj|�}|| S )Nr )r )r r r r )r r# r r! r r r �get_sheet_by_index3 s z _ODFReader.get_sheet_by_index)r c C sL ddl m} | jj|�}x |D ]}|jd�|kr |S q W td|� d���d S )Nr )r r zsheet z
not found)r r r r r �
ValueError)r r r r! �tabler r r �get_sheet_by_name9 s
z _ODFReader.get_sheet_by_name)�
convert_floatr c s� ddl m}m}m} |� j}|� j}||h� |j|�}d} d}
g } �xt|�D �]\} }
� fdd�|
jD �}d}g }xvt|�D ]j\}}|j|kr�| j||�}n| j }| j
|�}|| j kr�||7 }q~|j | j g| � d}|j |g| � q~W |
t |�k �rt |�}
| j
|
�}| j|
��r"| |7 } qN| j | j gg| � d} x t|�D ]}| j|� �qDW qNW x6| D ].}t |�|
k �rb|j | j g|
t |� � �qbW | S )z9
Parse an ODF Table into a list of lists
r )�CoveredTableCell� TableCell�TableRowc s g | ]}|j � kr|�qS r )�qname)r �x)�
cell_namesr r r U s z-_ODFReader.get_sheet_data.<locals>.<listcomp>)r r) r* r+ r, r � enumerate�
childNodes�_get_cell_valuer �_get_column_repeat�extend�len�_get_row_repeat�
_is_empty_row�range�append)r Zsheetr( r) r* r+ Zcovered_cell_nameZtable_cell_nameZ
sheet_rowsZ
empty_rowsZ max_row_lenr& �iZ sheet_rowZ sheet_cellsZ empty_cellsZ table_row�jZ
sheet_cell�valueZ
column_repeatZ
row_repeat�_�rowr )r. r �get_sheet_dataD sF
z_ODFReader.get_sheet_datac C s" ddl m} t|jj|dfd��S )z�
Return number of times this row was repeated
Repeating an empty row appeared to be a common way
of representing sparse rows in the table.
r )�TABLENSznumber-rows-repeated� )�odf.namespacesr? �int�
attributes�get)r r= r? r r r r5 } s z_ODFReader._get_row_repeatc C s" ddl m} t|jj|dfd��S )Nr )r? znumber-columns-repeatedr@ )rA r? rB rC rD )r �cellr? r r r r2 � s z _ODFReader._get_column_repeatc C s& x |j D ]}t|j �dkrdS qW dS )z4
Helper function to find empty rows
r FT)r0 r4 )r r= �columnr r r r6 � s z_ODFReader._is_empty_rowc C sD ddl m} t|�dkr tjS |jj|df�}|dkrJt|�dkrFdS dS |d krX| jS |d kr�t|jj|d
f��}|r�t |�}||kr�|S |S |d kr�|jj|d
f�}t|�S |d kr�| j
|�S |d
kr�|jj|d
f�}t|�S |dk�r|jj|df�}t j |�S |dk�r2t j t|��}t
t j|�}|j� S td|� ���d S )Nr )�OFFICENSz#N/Az
value-type�boolean�TRUETF�floatr; �
percentage�string�currency�datez
date-value�timezUnrecognized type )rA rG �str�np�nanrC rD r rJ rB �_get_cell_string_value�pdZ to_datetimer Z TimestamprO r% )r rE r( rG Z cell_typeZ
cell_value�val�resultr r r r1 � sB
z_ODFReader._get_cell_valuec C s� ddl m} ddlm} ddlm} |� j}g }xh|jD ]^}t||�r�|j|krvt |j
j |dfd��}|j d| � q�|j | j
|�� q8|j t|�� q8W dj|�S ) z�
Find and decode OpenDocument text:s tags that represent
a run length encoded sequence of space characters.
r )�Element)�TEXTNS)�S�cr@ � r )Z odf.elementrW rA rX Zodf.textrY r, r0 �
isinstancerB rC rD r8 rS rP �join) r rE rW rX rY Ztext_sr; �fragment�spacesr r r rS � s
z!_ODFReader._get_cell_string_value)�__name__�
__module__� __qualname__�__doc__r r �propertyr r rP r r r" rB r$ r' �boolr r> r5 r2 r6 r1 rS �
__classcell__r r )r
r r
s 9
'r )Ztypingr r ZnumpyrQ Zpandas._typingr r Zpandas.compat._optionalr ZpandasrT Zpandas.io.excel._baser r r r r r �<module> s |