ansible/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2024-09-14 20:13:21 +02:00
Code Releases Activity

Support different encoding types in csvfile lookup plugin

Browse source 
This fixes #15266.
This commit is contained in:
Dag Wieers 2016-04-04 20:11:44 +02:00 • committed by Brian Coca
parent 17f4db19ea
commit ac75069ee0
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
lib/ansible/plugins/lookup/csvfile.py
View file

@ -25,10 +25,10 @@ from ansible.plugins.lookup import LookupBase
class LookupModule(LookupBase): class LookupModule(LookupBase):
def read_csv(self, filename, key, delimiter, 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='utf-8') f = codecs.open(filename, 'r', encoding=encoding)
creader = csv.reader(f, delimiter=str(delimiter)) creader = csv.reader(f, delimiter=str(delimiter))
for row in creader: for row in creader:
@ -50,10 +50,11 @@ class LookupModule(LookupBase):
key = params[0] key = params[0]
paramvals = { paramvals = {
'file' : 'ansible.csv', 'col' : "1", # column to return
'default' : None, 'default' : None,
'delimiter' : "TAB", 'delimiter' : "TAB",
'col' : "1", # column to return 'file' : 'ansible.csv',
'encoding' : 'utf-8',
} }
# parameters specified? # parameters specified?
@ -69,7 +70,7 @@ class LookupModule(LookupBase):
paramvals['delimiter'] = "\t" paramvals['delimiter'] = "\t"
lookupfile = self._loader.path_dwim_relative(basedir, 'files', paramvals['file']) lookupfile = self._loader.path_dwim_relative(basedir, 'files', paramvals['file'])
var = self.read_csv(lookupfile, key, str(paramvals['delimiter']), paramvals['default'], paramvals['col']) var = self.read_csv(lookupfile, key, str(paramvals['delimiter']), paramvals['encoding'], paramvals['default'], paramvals['col'])
if var is not None: if var is not None:
if type(var) is list: if type(var) is list:
for v in var: for v in var: