Module:Indicator datasets table

From Visual Data Wiki
Jump to navigation Jump to search

Documentation for this module may be created at Module:Indicator datasets table/doc

local p = {}

function p.getDatasets(frame)
    local articleids = frame.args.articleids
    if not articleids then
        return "Error: articleids not provided"
    end
    
    -- Entferne alle Kommas und Leerzeichen am Ende von articleids
    articleids = mw.ustring.gsub(articleids, '[,%s]*$', '')
    
    local ids = mw.text.split(articleids, ",")
    local datasets = {}
    local titleobject
    
    for _, id in ipairs(ids) do
        titleobject = mw.title.new(tonumber(id))
        local existspage = titleobject.exists
        
        if existspage then
            local content = titleobject:getContent()
            -- Speichere den Inhalt der Seite
            table.insert(datasets, mw.text.jsonDecode(content))
        else
            return "Error: Page not found for ID " .. id
        end
    end
    
    return datasets
end

function p.generateTable(frame)
    local datasets = p.getDatasets(frame)
    if type(datasets) == "string" then
        return datasets -- Fehlermeldung zurückgeben
    end

    local html = mw.html.create()
    local tableNode = html:tag('table'):addClass('wikitable')

    -- Erstelle die Tabelle mit Legenden aus dem ersten Datensatz
    if #datasets > 0 then
        local headerRow = tableNode:tag('tr')
        headerRow:tag('th'):wikitext('Class')
        headerRow:tag('th'):wikitext('Sub-class')
        local firstDataset = datasets[1]
        for _, dataset in ipairs(firstDataset) do
            headerRow:tag('th'):wikitext(dataset.schemas.Dataset['date'])
        end
    end

    -- Füge die Datenzeilen hinzu
    for _, dataset in ipairs(datasets) do
        local datasetTable = mw.html.create('table'):addClass('wikitable')
        for _, classData in ipairs(dataset.schemas.Dataset.classes) do
            local class = classData['class name']
            for _, subClassData in ipairs(classData['sub-classes']) do
                local subClass = subClassData['sub-class name']
                local row = datasetTable:tag('tr')
                row:tag('td'):wikitext(class)
                row:tag('td'):wikitext(subClass)
                for _, currentSubClassData in ipairs(subClassData['sub-sub-classes']) do
                    row:tag('td'):wikitext(currentSubClassData.value or "N/A")
                end
            end
        end
        -- Füge die erstellte Tabelle zum Haupt-Tabellenknoten hinzu
        tableNode:node(datasetTable)
    end

    return tostring(html)
end

return p