Module:Indicator datasets table

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()

    -- Tabelle erstellen
    local tableNode = html:tag('table'):addClass('wikitable')

    -- Header-Zeile erstellen
    local headerRow = tableNode:tag('tr')
    headerRow:tag('th'):wikitext('Class')
    headerRow:tag('th'):wikitext('Sub-class')
    
    -- Iteriere über jedes Dataset und füge es als Spaltenüberschrift hinzu
    local datasetTitles = {}
    for _, dataset in ipairs(datasets) do
        table.insert(datasetTitles, dataset.schemas.Dataset.unique_title)
    end

    -- Entferne Duplikate aus den Spaltenüberschriften
    local uniqueDatasetTitles = {}
    for _, title in ipairs(datasetTitles) do
        if not uniqueDatasetTitles[title] then
            uniqueDatasetTitles[title] = true
            headerRow:tag('th'):wikitext(title)
        end
    end

    -- Datenzeilen erstellen
    for _, dataset in ipairs(datasets) do
        -- Iteriere über jede Klasse im Dataset
        for _, classData in ipairs(dataset.schemas.Dataset.classes) do
            local class = classData['class name']
            -- Iteriere über jede Sub-Klasse im Dataset
            for _, subClassData in ipairs(classData['sub-classes']) do
                local subClass = subClassData['sub-class name']
                local dataRow = tableNode:tag('tr')
                dataRow:tag('td'):wikitext(class)
                dataRow:tag('td'):wikitext(subClass)
                -- Iteriere über jedes Dataset und füge den Wert für diese Klasse und Sub-Klasse hinzu
                for _, datasetTitle in ipairs(datasetTitles) do
                    local value = p.findValue(dataset, class, subClass)
                    dataRow:tag('td'):wikitext(value or "N/A")
                end
            end
        end
    end

    return tostring(html)
end

function p.findValue(dataset, class, subClass)
    for _, clsData in ipairs(dataset.schemas.Dataset.classes) do
        if clsData['class name'] == class then
            for _, subClsData in ipairs(clsData['sub-classes']) do
                if subClsData['sub-class name'] == subClass then
                    return subClsData.value
                end
            end
        end
    end
    return nil
end

return p