mirror of
https://github.com/geoffleyland/lua-csv.git
synced 2024-11-23 01:34:19 +00:00
handle blank lines better: check whether they're blank before the transform runs. Also allow blank lines before the header
This commit is contained in:
parent
8eb59774ee
commit
8db3740230
22
lua/csv.lua
22
lua/csv.lua
@ -82,15 +82,19 @@ function column_map:read_header(header)
|
||||
|
||||
-- Match the columns in the file to the columns in the name map
|
||||
local found = {}
|
||||
local found_any
|
||||
for i, word in ipairs(header) do
|
||||
word = word:lower():gsub("[^%w%d]+", " "):gsub("^ *(.-) *$", "%1")
|
||||
local r = self.name_map[word]
|
||||
if r then
|
||||
index_map[i] = r
|
||||
found[r.name] = true
|
||||
found_any = true
|
||||
end
|
||||
end
|
||||
|
||||
if not found_any then return end
|
||||
|
||||
-- check we found all the columns we need
|
||||
local not_found = {}
|
||||
for name, r in pairs(self.name_map) do
|
||||
@ -123,6 +127,7 @@ function column_map:read_header(header)
|
||||
end
|
||||
|
||||
self.index_map = index_map
|
||||
return true
|
||||
end
|
||||
|
||||
|
||||
@ -284,7 +289,7 @@ local function separated_values_iterator(buffer, parameters)
|
||||
local sep = guess_separator(buffer, parameters)
|
||||
local line_start = 1
|
||||
local line = 1
|
||||
local field_count, fields, starts = 0, {}, {}
|
||||
local field_count, fields, starts, nonblanks = 0, {}, {}
|
||||
local header, header_read
|
||||
local field_start_line, field_start_column
|
||||
|
||||
@ -332,6 +337,7 @@ local function separated_values_iterator(buffer, parameters)
|
||||
end
|
||||
|
||||
value = tidy(value)
|
||||
if #value > 0 then nonblanks = true end
|
||||
field_count = field_count + 1
|
||||
|
||||
-- Insert the value into the table for this "line"
|
||||
@ -354,18 +360,18 @@ local function separated_values_iterator(buffer, parameters)
|
||||
-- if we ended on a newline then yield the fields on this line.
|
||||
if not this_sep or this_sep == "\r" or this_sep == "\n" then
|
||||
if parameters.column_map and not header_read then
|
||||
parameters.column_map:read_header(fields)
|
||||
header_read = true
|
||||
header_read = parameters.column_map:read_header(fields)
|
||||
elseif parameters.header and not header then
|
||||
header = fields
|
||||
header_read = true
|
||||
if nonblanks or field_count > 1 then -- ignore blank lines
|
||||
header = fields
|
||||
header_read = true
|
||||
end
|
||||
else
|
||||
local k, v = next(fields)
|
||||
if v ~= "" or field_count > 1 then -- ignore blank lines
|
||||
if nonblanks or field_count > 1 then -- ignore blank lines
|
||||
coroutine.yield(fields, starts)
|
||||
end
|
||||
end
|
||||
field_count, fields, starts = 0, {}, {}
|
||||
field_count, fields, starts, nonblanks = 0, {}, {}
|
||||
end
|
||||
|
||||
-- If we *really* didn't find a separator then we're done.
|
||||
|
Loading…
Reference in New Issue
Block a user