Caleb Cooper: 1 revision imported

2020-01-28T15:03:40Z

1 revision imported

← Older revision	Revision as of 15:03, 28 January 2020
(No difference)

powerstorewiki>Caleb Cooper: 1 revision imported

2020-01-21T18:20:50Z

1 revision imported

New page

local p = {}

function p.titleToJson(title)
-- Load mediawiki page and decodes it as json
local t = mw.title.new(title)
if not t.exists then
return nil
else
return mw.text.jsonDecode(t:getContent(), mw.text.JSON_TRY_FIXING)
end
end

function p.recodeListToMap(frame)
-- a function to re-code a json table in format [{},{},{}] to a map {},
-- with each source object containing the same set of values, and the response
-- mapping the "from" value to "to" value
-- parameters: tbl, from, to
local args = frame.args
local tbl = p.titleToJson(args[1])
local from = args[2]
local to = args[3]
local result = p.recodeListToMapFunc(tbl, from, to, args)
return mw.text.jsonEncode(result)
end

function p.recodeListToMapFunc(tbl, from, to, values)
-- a function to re-code a json table in format [{},{},{}] to a map {},
-- with each source object containing the same set of values, and the response
-- mapping the "from" values to "to" values
-- parameters: tbl, from, to
local dict = {}
for k, v in pairs(tbl) do
if v[from] ~= nil and v[to] ~= nil then
dict[v[from]] = v[to]
end
end
local result = {}
for key, val in pairs(values) do
if type(key) == "string" then
local k = dict[key]
if k ~= nil then
result[k] = val
end
end
end
return result
end

function p.parseCsv(frame)
local args = frame.args
local content = mw.title.new(args[1]):getContent()
local columnName = args[2]
local result = p.parseCsvFunc(content, columnName)
return mw.text.jsonEncode(result)
end

function p.parseCsvFunc(content, columnName)
-- Simple, no escaping, csv parser. Assumes first row contains column names
-- given csv text, extracts first column and names it 'id', and any other
-- column that has the same name as the given columnName, and names it 'v'
-- the result is in form: [{id=...,v=...}, {id=...,v=...}], ...]
-- if coulmn is not found, v= will not be given
local result = {}
local numberIdx = false

for v in string.gmatch(content, '([^\n]*)') do
local val = mw.text.trim(v)
if val ~= '' then
local parts = mw.text.split( val, ',', true )
if numberIdx == false then
numberIdx = true
for i, v in ipairs(parts) do
local val = mw.text.trim(v)
if val == columnName then
numberIdx = i
break
end
end
else
local val = nil
if numberIdx ~= true then
val = tonumber(mw.text.trim(parts[numberIdx]))
end
table.insert(result, {
id=mw.text.trim(parts[1]),
v=val
})
end
end
end
return result
end

function p.parseHCsv(frame)
local args = frame.args
local content = mw.title.new(args[1]):getContent()
local rowId = args[2]
local result = p.parseHCsvFunc(content, rowId)
return mw.text.jsonEncode(result)
end

function p.parseHCsvFunc(content, id)
-- Simple, no escaping, csv parser. Assumes first row contains column names
-- given csv text, extracts just the row whos first column matches given id
-- first column is not returned, all other column's headers become IDs
-- the result is in form: [{id=...,v=...}, {id=...,v=...}], ...]
-- DEBUG: =mw.text.jsonEncode(p.parseHCsvFunc('i,x,y,z\nb,1,2,3\na,2,3,4','a'))
local result = {}
local headers = false

for v in string.gmatch(content, '([^\n]*)') do
local val = mw.text.trim(v)
if val ~= '' then
local parts = mw.text.split( val, ',', true )
if headers == false then
headers = {}
for i, v in ipairs(parts) do
table.insert(headers, mw.text.trim(v))
end
elseif mw.text.trim(parts[1]) == id then
local isFirst = true
for i, v in ipairs(parts) do
if isFirst then
isFirst = false
else
table.insert(result, {
id=headers[i],
v=tonumber(mw.text.trim(v))
})
end
end
break
end
end
end
return result
end

function p.expandDict(frame)
return mw.text.jsonEncode(p.expandDictFunc(mw.text.jsonDecode(frame.args[1])))
end

function p.expandDictFunc(inp)
-- Converts compressed format json table to expanded format, for example:
-- { "a": 10, "b": 20 }
-- will be converted to
-- [{ "id": "a", "v": 10 }, { "id": "b", "v": 20 }]
-- DEBUG: =mw.text.jsonEncode(p.expandDictFunc(mw.text.jsonDecode('{ "a": 10, "b": 20 }')))
local result = {}
for k, v in pairs(inp) do
table.insert(result, {
id=k,
v=v
})
end
return result
end

return p

Module:Graph:Utils - Revision history

Caleb Cooper: 1 revision imported

powerstorewiki>Caleb Cooper: 1 revision imported