Spaces:
Sleeping
Sleeping
File size: 2,215 Bytes
402daee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
--[[
CSVWriter.lua - Write CSV files
]]--
CSVWriter = Polo {
TIME_FORMAT = '%02d:%02d:%02d,%03d',
DELIMITERS = {
{ char = ',', name = 'Comma' },
{ char = ';', name = 'Semicolon' },
{ char = '\t', name = 'Tab' },
},
new = function(options)
options = options or {}
return {
file = options.file,
delimiter = options.delimiter or ',',
include_header_row = options.include_header_row or false,
}
end,
}
function CSVWriter:init()
assert(self.file, 'missing file')
end
CSVWriter.format_time = function (time)
local milliseconds = math.floor(time * 1000) % 1000
local seconds = math.floor(time) % 60
local minutes = math.floor(time / 60) % 60
local hours = math.floor(time / 3600)
return string.format(CSVWriter.TIME_FORMAT, hours, minutes, seconds, milliseconds)
end
function CSVWriter:write(transcript)
if self.include_header_row then
self:write_header_row()
end
local sequence_number = 1
for _, segment in pairs(transcript:get_segments()) do
self:write_segment(segment, sequence_number)
sequence_number = sequence_number + 1
end
end
function CSVWriter:write_header_row()
local fields = {
CSVWriter._quoted('Sequence Number'),
CSVWriter._quoted('Start Time'),
CSVWriter._quoted('End Time'),
CSVWriter._quoted('Text'),
CSVWriter._quoted('File'),
}
self.file:write(table.concat(fields, self.delimiter))
self.file:write('\n')
end
function CSVWriter:write_segment(segment, sequence_number)
local start = segment:get('start')
local end_ = segment:get('end')
local text = segment:get('text')
local file = segment:get_file_with_extension()
self:write_line(text, sequence_number, start, end_, file)
end
function CSVWriter:write_line(line, sequence_number, start, end_, file)
local fields = {
sequence_number,
CSVWriter._quoted(CSVWriter.format_time(start)),
CSVWriter._quoted(CSVWriter.format_time(end_)),
CSVWriter._quoted(line),
CSVWriter._quoted(file),
}
self.file:write(table.concat(fields, self.delimiter))
self.file:write('\n')
end
function CSVWriter._quoted(input_string)
return table.concat({'"', input_string:gsub('"', '""'), '"'})
end
|