Your IP : 216.73.216.213
B
�d{T} � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl m
Z
G dd� d�Z dd� Z e
d kr�e d
�Zx,eD ]$Zed ed �� ed
�� f � qxW dS )a? This is a DBF reader which reads Visual Fox Pro DBF format with Memo field
Usage:
>>> rec = readDbf('test.dbf')
>>> for line in rec:
>>> print line['name']
:date: 13/07/2007
http://www.physics.ox.ac.uk/users/santoso/Software.Repository.html
page says code is "available as is without any warranty or support".
� )�print_functionN)�rangec @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�Dbasec C sF d | _ d | _d | _d | _d | _d| _d | _d| _d | _d| _ d| _
d S )Nr � ) �fdb�fmemo�db_data� memo_data�fields� num_records�header� memo_file� memo_header�memo_block_size�memo_header_len)�self� r �C/opt/alt/python37/lib/python3.7/site-packages/logilab/common/dbf.py�__init__, s zDbase.__init__c C sD x>t dt|��D ],}tt�d|| �d �dkr|d |� S qW |S )Nr �c)r �len�ord�struct�unpack)r Ztxt�ir r r �_drop_after_NULL9 s zDbase._drop_after_NULLc C s< t |�s dS t�d|�}t�d|d �}t�d|�}|d S )Nr z<Lz>L)r r r Zpack)r �num�valr r r �_reverse_endian? s zDbase._reverse_endianc C s2 i }d}x$|D ] }|| }|||< |d7 }qW |S )Nr � r )r Zlst�ids�result�idx�item�idr r r � _assign_idsG s
zDbase._assign_idsc
C s� t j�|�}|dkr td��t|d�| _d| _t j�|dd� d �rZ|dd� d | _n*t j�|dd� d �r�|dd� d | _| jr�t| jd�| _| j� � | _
| � t �
d
| j
d d � �d g�| _| jd }|s�d
}|| _|| _t j�| j�}| j� d�}| � t �
d|�dddddddg�| _t| jd �| jd< | jd | _| j� | jd d �}i | _d}d}dddd g}d} x�tdt|�d�D ]�}
t||
�d k�r�P |d 7 }||
|
| � } t| �| k �r�| d | t| � 7 } | � t �
|| �|�| j|< | �| j| d �| j| d< �q�W | j� d � | jd �rX| jd | jd d } | j� | �| _nd| _d!| _g | _d| _ xR| jD ]H}
| j|
}| jd"|d 7 _| j� |d � | j |d 7 _ �qxW d S )#N�D z-The file is not large enough to be a dbf file�rbr r ����t���Zfptz>6x1H� z
Block sizei � z<B 3B L 2H 20xr$ ZYearZMonthZDayz # of Recordsz Header Sizez Record Size�" z<11s c 4x B B 14xz
Field Namez
Field Typez Field LengthzField Precision�
r � � �<z%ds ) �os�path�getsize�IOError�openr r
�isfiler �readr r% r r r r r r �hexr r
r r r r r �
row_format�row_ids�row_len�append)r Zdb_nameZfilesize�
block_sizeZ memo_size�data�xZheader_patternr Z pattern_len�offsetZ data_subsetZ data_size�key�fieldr r r r6 P sf "
$ $
z
Dbase.openc C s$ | j r| j �� | jr | j�� d S )N)r �closer )r r r r rD � s
z Dbase.closec C s | j S )N)r )r r r r �get_numrecords� s zDbase.get_numrecordsc
C s� |dk s|| j krtd��| jd | }| j||| j � }| �t�| j|�| j �}| j
r�x�| j D ]x}| j | }|d }|d }|| } |dks�|dks�|dks�|d kr�| � | �} | r�| �
| d
��� ||< qb| �� ||< qbW |S ) zB
This function accept record number from 0 to N-1
r z(Unable to extract data outside the rangez Record Sizez
Field Typez
Field Name�M�G�B�Pr )r � Exceptionr r r< r% r r r: r; r
r
r � read_memo�strip)
r Zrec_norA r? �recordrB rC Zf_typeZf_nameZc_datar r r �get_record_with_names� s"
zDbase.get_record_with_namesc C sR |dk r| j }| j|| j }| j�|� |dk r:| }|dkrFdS | j�|�S )z�
Read the record of given number. The second parameter is the length of
the record to read. It can be undefined, meaning read the whole record,
and it can be negative, meaning at most the length
r r )r r r �seekr8 )r r Z in_lengthrA r r r �read_memo_record� s zDbase.read_memo_recordc C s� d}| � |d�}t|�dkr dS t�d|dd� �d d }| j}||k rV|d|� S || }| � |d |�}t|�dkr~dS |dd � | S )Nr r( r z>L� r+ r )rP r r r r )r r r! �buffer�lengthr> Z rest_lengthZ rest_datar r r rK � s zDbase.read_memoN)
�__name__�
__module__� __qualname__r r r r% r6 rD rE rN rP rK r r r r r + s
>r c C sP t � }|�| � |�� }g }x&td|�D ]}|�|�}|�|� q(W |�� |S )z�
Read the DBF file specified by the filename and
return the records as a list of dictionary.
:param: filename File name of the DBF
:return: List of rows
r )r r6 rE r rN r= rD )�filenameZdbr �recr rM r r r �readDbf� s
rY �__main__zdbf/sptable.dbfz%s %sZGENUSZSPECIES)�__doc__Z
__future__r r r2 Zos.path�sysZcsv�tempfileZ six.movesr r rY rT rX �line�printrL r r r r �<module> s (