Jump to content

Module:SongContestData

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by TheThomanski (talk | contribs) at 18:16, 17 March 2025. The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.

local getArgs = require('Module:Arguments').getArgs
local p = {}

local function makeInvokeFunc(funcName)
	return function (frame)
		local args = getArgs(frame)
		return p[funcName](args)
	end
end

local function processOutput(o)
	if type(o) == "table" then
	    local result = {}
	    for _, v in pairs(o) do
	        table.insert(result, tostring(v))
	    end
	    return table.concat(result, ";")
	end
	return o
end

local function getData(contest, year)
	return mw.loadData('Module:SongContestData/'..contest..'/'..year)
end

local function sortEntries(data, att, desc)
    -- deep copy since data is non writable
    local entries = {}
	for _,v in pairs(data) do table.insert(entries, v) end
	
    -- filter entries that do not have att
    local filtered_data = {}
    for _,v in pairs(entries) do
        if v[att] then  -- keep entries with valid att
            table.insert(filtered_data, v)
        end
    end
    
    -- sort entries based on att
    table.sort(filtered_data, function(a, b)
    	return desc and a[att] > b[att] or a[att] < b[att]
    end)
    
    return filtered_data
end

p.main = makeInvokeFunc('_main')
function p._main(args)
	local data = getData(args[1], args[2])
	local entryData = data[args[3]]
	
	if entryData and entryData[args[4]] then
        return processOutput(entryData[args[4]])
    end

    return ""
end

p.entryAmount = makeInvokeFunc('_entryAmount')
function p._entryAmount(args)
	local data = getData(args[1], args[2])
	local amount = 0
	for _ in pairs(data) do amount = amount + 1 end
	return amount
end

p.order = makeInvokeFunc('_order')
function p._order(args)
	local data = getData(args[1], args[2])
	return sortEntries(data, args[3], args[4])
end

return p