B jV@sddlZddlZddlmZddddddd d d d d dddgZddddgeZxeD]Zeeeee<qRWddZ d"ddZ ddZ Gddde Z e ZGddde ZeZGd ddejZGd!ddejZdS)#N)izipZregister_dialectZunregister_dialectZ get_dialectZ list_dialectsZfield_size_limitZDialectexcelZ excel_tabZSnifferZ QUOTE_ALLZ QUOTE_MINIMALQUOTE_NONNUMERICZ QUOTE_NONEErrorreaderwriter DictReader DictWritercCsF|dkr dSt|tr"|||St|tjr0nt|tsBt|}|S)N) isinstanceunicodeencodenumbersNumberstr)sencodingerrorsr-lib/python3.7/site-packages/unicodecsv/py2.py _stringify"s    rstrictc sNyfddt|DStk rH}ztt|Wdd}~XYnXdS)Ncsg|]}t|qSr)r).0r)rrrr 0sz#_stringify_list..)iter TypeErrorcsvrr)lrrer)rrr_stringify_list.srcCs8|dkr dSt|tttfr |St|tr4||S|S)N)r r intfloatrdecode)rrrrr _unicodify5s  r#c@s>eZdZdZejddfddZddZdd Ze d d Z d S) UnicodeWriteruQ >>> import unicodecsv >>> from cStringIO import StringIO >>> f = StringIO() >>> w = unicodecsv.writer(f, encoding='utf-8') >>> w.writerow((u'é', u'ñ')) >>> f.seek(0) >>> r = unicodecsv.reader(f, encoding='utf-8') >>> row = r.next() >>> row[0] == u'é' True >>> row[1] == u'ñ' True zutf-8rcOs&||_tj||f|||_||_dS)N)rrrencoding_errors)selffdialectrrargskwdsrrr__init__NszUnicodeWriter.__init__cCs|jt||j|jS)N)rwriterowrrr%)r&rowrrrr,TszUnicodeWriter.writerowcCsx|D]}||qWdS)N)r,)r&Zrowsr-rrr writerowsXs zUnicodeWriter.writerowscCs|jjS)N)rr()r&rrrr(\szUnicodeWriter.dialectN) __name__ __module__ __qualname____doc__rrr+r,r.propertyr(rrrrr$?s r$c@s>eZdZdddZddZdd Zed d Zed d ZdS) UnicodeReaderNutf-8rc spdddddddg|dkr:tfdd |Ds:tj}tj||f||_||_||_t|jj tj @|_ dS) NZ delimiterZ doublequoteZ escapecharZlineterminatorZ quotecharquotingZskipinitialspacecsg|] }|kqSrr)rZkwd_name) format_paramsrrrksz*UnicodeReader.__init__..) anykeysrrrrr%boolr(r6r_parse_numerics)r&r'r(rrr*r)r7rr+cs zUnicodeReader.__init__csV|j}|j|jt|jr.csg|]}|qSrr)rr<)rr%r>rrrs)rnextrr%r r;r!)r&r-r)rr%r=r>rr?ts zUnicodeReader.nextcCs|S)Nr)r&rrr__iter__szUnicodeReader.__iter__cCs|jjS)N)rr()r&rrrr(szUnicodeReader.dialectcCs|jjS)N)rline_num)r&rrrrAszUnicodeReader.line_num)Nr5r) r/r0r1r+r?r@r3r(rArrrrr4bs   r4c@s"eZdZdZd ddZd d Zd S) r u& >>> from cStringIO import StringIO >>> f = StringIO() >>> w = DictWriter(f, ['a', u'ñ', 'b'], restval=u'î') >>> w.writerow({'a':'1', u'ñ':'2'}) >>> w.writerow({'a':'1', u'ñ':'2', 'b':u'ø'}) >>> w.writerow({'a':u'é', u'ñ':'2'}) >>> f.seek(0) >>> r = DictReader(f, fieldnames=['a', u'ñ'], restkey='r') >>> r.next() == {'a': u'1', u'ñ':'2', 'r': [u'î']} True >>> r.next() == {'a': u'1', u'ñ':'2', 'r': [u'ø']} True >>> r.next() == {'a': u'é', u'ñ':'2', 'r': [u'î']} True r raiserutf-8rc OsL||_tjj||||||f|| t||f|||d| |_||_dS)N)rr)rrr r+r$rr%) r&csvfile fieldnamesrestvalZ extrasactionr(rrr)r*rrrr+s zDictWriter.__init__cCs tt|j|j}||dS)N)dictziprEr,)r&headerrrr writeheaderszDictWriter.writeheaderN)r rBrrCr)r/r0r1r2r+rJrrrrr s  c@s"eZdZdZd ddZdd ZdS) ru >>> from cStringIO import StringIO >>> f = StringIO() >>> w = DictWriter(f, fieldnames=['name', 'place']) >>> w.writerow({'name': 'Cary Grant', 'place': 'hollywood'}) >>> w.writerow({'name': 'Nathan Brillstone', 'place': u'øLand'}) >>> w.writerow({'name': u'Will ø. Unicoder', 'place': u'éSpandland'}) >>> f.seek(0) >>> r = DictReader(f, fieldnames=['name', 'place']) >>> print r.next() == {'name': 'Cary Grant', 'place': 'hollywood'} True >>> print r.next() == {'name': 'Nathan Brillstone', 'place': u'øLand'} True >>> print r.next() == {'name': u'Will ø. Unicoder', 'place': u'éSpandland'} True Nrutf-8rc s|dk rt|}tjj||||||f|| t||f||d| |_|dkrttjdst||f|di| } t| | j|_ |j dk rfdd|j D|_ ng|_ t ||_ dS)N)rrrErcsg|]}t|qSr)r#)rr')rrrrsz'DictReader.__init__..) rrrr+r4rhasattrr?rrEunicode_fieldnamesr#unicode_restkey) r&rDrErestkeyrFr(rrr)r*rr)rrr+s   zDictReader.__init__csJtj|tfddt|j|jD}|j}|rF|||j <|S)Nc3s|]\}}||fVqdS)Nr)rZstr_keyZuni_key)r-rr sz"DictReader.next..) rrr?rGrrErMgetrOrN)r&resultrestr)r-rr?s    zDictReader.next)NNNrrKr)r/r0r1r2r+r?rrrrrs )r)rr itertoolsrZ pass_throughs__all__Zpropgetattrglobalsrrr#objectr$rr4rr rrrrrs@     *