mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
Implement own CSVRecoder and CSVReader from documentation
As detailed in the python manual: https://docs.python.org/2/library/csv.html
This commit is contained in:
parent
ac75069ee0
commit
055aae55f7
1 changed files with 32 additions and 2 deletions
|
@ -23,13 +23,43 @@ import csv
|
||||||
from ansible.errors import AnsibleError
|
from ansible.errors import AnsibleError
|
||||||
from ansible.plugins.lookup import LookupBase
|
from ansible.plugins.lookup import LookupBase
|
||||||
|
|
||||||
|
class CSVRecoder:
|
||||||
|
"""
|
||||||
|
Iterator that reads an encoded stream and reencodes the input to UTF-8
|
||||||
|
"""
|
||||||
|
def __init__(self, f, encoding='utf-8'):
|
||||||
|
self.reader = codecs.getreader(encoding)(f)
|
||||||
|
|
||||||
|
def __iter__(self):
|
||||||
|
return self
|
||||||
|
|
||||||
|
def next(self):
|
||||||
|
return self.reader.next().encode("utf-8")
|
||||||
|
|
||||||
|
class CSVReader:
|
||||||
|
"""
|
||||||
|
A CSV reader which will iterate over lines in the CSV file "f",
|
||||||
|
which is encoded in the given encoding.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, f, dialect=csv.excel, encoding='utf-8', **kwds):
|
||||||
|
f = CSVRecoder(f, encoding)
|
||||||
|
self.reader = csv.reader(f, dialect=dialect, **kwds)
|
||||||
|
|
||||||
|
def next(self):
|
||||||
|
row = self.reader.next()
|
||||||
|
return [unicode(s, 'utf-8') for s in row]
|
||||||
|
|
||||||
|
def __iter__(self):
|
||||||
|
return self
|
||||||
|
|
||||||
class LookupModule(LookupBase):
|
class LookupModule(LookupBase):
|
||||||
|
|
||||||
def read_csv(self, filename, key, delimiter, encoding='utf-8', dflt=None, col=1):
|
def read_csv(self, filename, key, delimiter, encoding='utf-8', dflt=None, col=1):
|
||||||
|
|
||||||
try:
|
try:
|
||||||
f = codecs.open(filename, 'r', encoding=encoding)
|
f = open(filename, 'r')
|
||||||
creader = csv.reader(f, delimiter=str(delimiter))
|
creader = CSVReader(f, delimiter=str(delimiter), encoding=encoding)
|
||||||
|
|
||||||
for row in creader:
|
for row in creader:
|
||||||
if row[0] == key:
|
if row[0] == key:
|
||||||
|
|
Loading…
Reference in a new issue