From 339b9f8e9ce363507f38eaff1344ba7bbaa30f78 Mon Sep 17 00:00:00 2001 From: erickertz Date: Fri, 25 Nov 2016 20:25:45 -0500 Subject: [PATCH] Fix _io.BufferedReader AttributeError * Update csv2es.py to open doc_file in text mode * Update csv2es.py to parse doc_file using readline --- csv2es.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/csv2es.py b/csv2es.py index 642e3ba..c4ce167 100644 --- a/csv2es.py +++ b/csv2es.py @@ -52,9 +52,9 @@ def documents_from_file(es, filename, delimiter, quiet): :return: generator returning document-indexing operations """ def all_docs(): - with open(filename, 'rb') if filename != '-' else sys.stdin as doc_file: + with open(filename, 'r') if filename != '-' else sys.stdin as doc_file: # delimited file should include the field names as the first row - fieldnames = doc_file.next().strip().split(delimiter) + fieldnames = doc_file.readline().strip().split(delimiter) echo('Using the following ' + str(len(fieldnames)) + ' fields:', quiet) for fieldname in fieldnames: echo(fieldname, quiet)