Книги
Lua
Розділ 10. Формати даних та серіалізація
Робота з CSV

Lua:
Робота з CSV

Як:

У Lua робота з файлами CSV може бути підходом за допомогою базових операцій введення-виведення файлів, які надає мова, без потреби у зовнішніх бібліотеках для простих завдань. Для більш складних операцій, наприклад, обробки спеціальних випадків (наприклад, коми всередині значень), може бути корисним використання сторонніх бібліотек, як-от lua-csv.

Читання файлу CSV

Ось простий приклад читання файлу CSV рядок за рядком, розділення кожного рядка на значення засноване на роздільнику-комі.

function parseCSVLine(line)
    local result = {}
    local from = 1
    local sep = ","
    local field
    while true do
        local start, finish = string.find(line, sep, from)
        if not start then
            table.insert(result, string.sub(line, from))
            break
        end
        field = string.sub(line, from, start - 1)
        table.insert(result, field)
        from = finish + 1
    end
    return result
end

local file = io.open("example.csv", "r")
for line in file:lines() do
    local values = parseCSVLine(line)
    for i, v in ipairs(values) do
        print(i, v)
    end
end
file:close()

Приклад виведення (для example.csv з вмістом “name,age\newlineJohn Doe,30\newlineJane Doe,32”):

1	name
2	age
1	John Doe
2	30
1	Jane Doe
2	32

Запис файлу CSV

Щоб згенерувати файл CSV, ви просто створюєте рядки з кома-розділеними значеннями і записуєте їх у файл рядок за рядком.

local data = {
    {"name", "age"},
    {"John Doe", "30"},
    {"Jane Doe", "32"}
}

local file = io.open("output.csv", "w")
for _, v in ipairs(data) do
    file:write(table.concat(v, ","), "\n")
end
file:close()

Цей метод створить (або перезапише) файл output.csv з вказаними даними.

Використання lua-csv

Для більш складного оброблення CSV, включаючи підтримку лапок та символів екранування, бібліотека lua-csv є надійним вибором.

Спочатку встановіть її за допомогою LuaRocks:

luarocks install lua-csv

Тоді читання файлу CSV стає таким простим:

local csv = require("csv")

-- Читання з файлу
for fields in csv.open("example.csv") do
    for i, v in ipairs(fields) do
        print(i, v)
    end
end

І запис у CSV з належними лапками та екрануванням:

local file = csv.open("output.csv", {write=true})

local data = {
    {"name", "profession", "location"},
    {"John Doe", "Software Engineer", "New York, NY"},
    {"Jane Doe", "Data Scientist", "\"San Francisco, CA\""}
}

for _, v in ipairs(data) do
    file:write(v)
end

Цей підхід автоматично обробляє складності, такі як коми та лапки всередині значень.

Робота з JSON