U d$@sldZdddZdddZdgZddlZddlZddlZddlZd d lm Z d d l m Z Gd ddZ dS) zDBF header definition. TODO: - handle encoding of the character fields (encoding information stored in the DBF header) z$Revision: 1.6 $ z$Date: 2010/09/16 05:06:39 $ DbfHeaderN)fields)getDatec@seZdZdZdZd$ddZd d ZeeZd d ZeeZe d dZ e ddZ e ddZ ddZ e dde ddZ ddZddZddZddZddZd d!Zd"d#ZdS)%ra|Dbf header definition. For more information about dbf header format visit `http://www.clicketyclick.dk/databases/xbase/format/dbf.html#DBF_STRUCT` Examples: Create an empty dbf header and add some field definitions: dbfh = DbfHeader() dbfh.addField(("name", "C", 10)) dbfh.addField(("date", "D")) dbfh.addField(DbfNumericFieldDef("price", 5, 2)) Create a dbf header with field definitions: dbfh = DbfHeader([ ("name", "C", 10), ("date", "D"), DbfNumericFieldDef("price", 5, 2), ]) ) signaturer lastUpdate recordLength recordCount headerLengthchanged_ignore_errorsNrFcCsR||_|dkrg|_n t||_t||_||_||_||_||_t |j|_ dS)aInitialize instance. Arguments: fields: a list of field definitions; recordLength: size of the records; headerLength: size of the header; recordCount: number of records stored in DBF; signature: version number (aka signature). using 0x03 as a default meaning "File without DBT". for more information about this field visit ``http://www.clicketyclick.dk/databases/xbase/format/dbf.html#DBF_NOTE_1_TARGET`` lastUpdate: date of the DBF's update. this could be a string ('yymmdd' or 'yyyymmdd'), timestamp (int or float), datetime/date value, a sequence (assuming (yyyy, mm, dd, ...)) or an object having callable ``ticks`` field. ignoreErrors: error processing mode for DBF fields (boolean) N) r rlistrr r r r ignoreErrorsboolr)selfrr r r r r rr@/tmp/pip-unpacked-wheel-_3ph8cfy/tablib/packages/dbfpy/header.py__init__9s  zDbfHeader.__init__cCs|tt|S)z.Return header instance from the string object.) fromStreamioStringIOstr)clsstringrrr fromStringbszDbfHeader.fromStringc Cs|d|d}t|tkr.t|t}|}td|dd\}}}|d}|dkrf|d7}n|d 7}|d ||||d||d |d f}d} |d}|d kr||d7}t |d || } | | | j } |d}q|S)z%Return header object from the stream.r zzDbfHeader.cCs|jjSr7)r monthr9rrrr:r;cCs|jjSr7)r dayr9rrrr:r;cCs$t||_}|jD] }||_qdS)z1Update `ignoreErrors` flag on self and all fieldsN)rrrr)rvalue_fieldrrrrs zDbfHeader.ignoreErrorscCs|jSr7)rr9rrrr:r;zError processing mode for DBF field value conversion if set, failing field value conversion will return ``INVALID_VALUE`` instead of raising conversion error. )doccCs:d|j|j|j|j|jf}|ddd|jD7}|S)NzVersion (signature): 0x%02x Last update: %s Header length: %d Record length: %d Record count: %d FieldName Type Len Dec  cSsg|]}d|qS)z%10s %4s %3s %3s)Z fieldInfo.0r6rrr sz&DbfHeader.__repr__..)r r r r r joinr)rZ_rvrrr__repr__s  zDbfHeader.__repr__c Gsg}d}|D]b}t|tjr"|}n8t|ddd\}}}} t|} | ||| |jd}||j7}||q |j|7_|S)ahInternal variant of the `addField` method. This method doesn't set `self.changed` field to True. Return value is a length of the appended records. Note: this method doesn't modify ``recordLength`` and ``headerLength`` fields. Use `addField` instead of this method if you don't exactly know what you're doing. r)NNNNNr )r) isinstancerZ DbfFieldDeftupler/rlengthappend) rdefsZ_defsZ _recordLengthZ_defr4_name_type_lenZ_dec_clsrrrr0s    zDbfHeader._addFieldcGsN|j}|j|j|7_|s,|jd7_ddt|jd|_d|_dS)aAdd field definition to the header. Examples: dbfh.addField( ("name", "C", 20), dbf.DbfCharacterFieldDef("surname", 20), dbf.DbfDateFieldDef("birthdate"), ("member", "L"), ) dbfh.addField(("price", "N", 5, 2)) dbfh.addField(dbf.DbfNumericFieldDef("origprice", 5, 2)) rrTN)r r0lenrr r)rrKZ_oldLenrrraddFields zDbfHeader.addFieldcCsV|d||dd|jD}|d|t|dd|_dS)z&Encode and write header to the stream.rcSsg|] }|qSr)toStringrBrrrrDsz#DbfHeader.write..r%FN) r'writerRrrEencoder+r,r)rr2rrrrrTs   zDbfHeader.writec Cs.td|j|jd|j|j|j|j|jdS)z4Returned 32 chars length string with encoded header.z<4BI2Hr#s) r-packr r8r<r=r r r r9rrrrRszDbfHeader.toStringcCstj|_dS)z9Update ``self.lastUpdate`` field with current date value.N)datetimedatetodayr r9rrrsetCurrentDateszDbfHeader.setCurrentDatecCsHt|tr:|}|jD]}|j|kr|Sqt|n |j|SdS)z9Return a field definition by numeric index or name stringN)rGrupperrnameKeyError)ritemrLr?rrr __getitem__s     zDbfHeader.__getitem__)NrrrrNF)__name__ __module__ __qualname____doc__ __slots__rr classmethodrpropertyr8r<r=rrFr0rQrTrRrZr_rrrrrs8 )       ) rc __version____date____all__rWrr-r+rSrutilsrrrrrrs