Module:Learners

Documentation for this module may be created at Module:Learners/doc

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

function p.Main( frame )
    local args = getArgs(frame)
    local play = args[1] or mw.title.getCurrentTitle().text
    local levels = args["level"]
    levels = levels and levels ~= ""
    
    local tables = (levels and "BeastieLearnset") or "BeastieLearnFriend"
    local fields = (levels and "Beastie,Plays") or "Beastie"
    local where = tables .. ".Plays HOLDS LIKE '" .. ((levels and play .. "%") or play) .. "'"
    local results = cargo.query(tables, fields, {where=where})

    local tbl = mw.html.create('table'):addClass('wikitable')
    local header = tbl:tag("tr")
        :tag("th"):wikitext("Beastie"):done()
    if levels then
        header:tag("th"):wikitext("Levels")
    end

    for _, result in ipairs(results) do
        local tr = tbl:tag("tr")
        tr:tag("td"):wikitext(result.Beastie)
        if levels then
            tr:tag("td"):wikitext(result.Plays)
        end
    end
    return tbl
end

return p