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 or articleids == '' then
	    return {}
	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()
            local dataset = mw.text.jsonDecode(content)
            -- Überprüfe, ob belongs to der aktuellen PageID entspricht
            if tonumber(dataset.schemas.Dataset['belongs to']) == mw.title.getCurrentTitle().id then
                table.insert(datasets, dataset)
            end
        else
            return "Error: Page not found for ID " .. id
        end
    end
    return datasets
end

function p.generateTable(frame)
    local datasets = p.getDatasets(frame)
    if not datasets or #datasets == 0 then
        return "No data" -- Keine Datensätze gefunden, gib leeren String zurück
    end
    
    -- Sortiere die Datensätze nach dem Datum
    table.sort(datasets, function(a, b) return a.schemas.Dataset['date'] < b.schemas.Dataset['date'] end)

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

    -- Erstelle die Tabelle mit Legenden aus dem ersten Datensatz
    local headerRow = tableNode:tag('tr')
    headerRow:tag('th'):wikitext('Class')
    headerRow:tag('th'):wikitext('Sub-class')
    for _, dataset in ipairs(datasets) do
        local formattedDate = p.formatDate(dataset.schemas.Dataset['date'])
        headerRow:tag('th'):wikitext(formattedDate):tag('div'):tag('small'):wikitext(mw.message.new('webmo-' .. dataset.schemas.Dataset['dataset type']):plain())
    end

    -- Initialisiere die Summen für jede Spalte
    local columnSums = {}
    local columnCounts = {}
    for _, dataset in ipairs(datasets) do
        for _, classData in ipairs(dataset.schemas.Dataset.classes) do
            for _, subClassData in ipairs(classData['sub-classes']) do
                local subClassName = subClassData['sub-class name']
                columnSums[subClassName] = columnSums[subClassName] or {}
                columnCounts[subClassName] = columnCounts[subClassName] or {}
                for _, currentSubSubClassData in ipairs(subClassData['sub-sub-classes']) do
                    local value = tonumber(currentSubSubClassData.value) or 0
                    columnSums[subClassName][dataset.schemas.Dataset['date']] = (columnSums[subClassName][dataset.schemas.Dataset['date']] or 0) + value
                    columnCounts[subClassName][dataset.schemas.Dataset['date']] = (columnCounts[subClassName][dataset.schemas.Dataset['date']] or 0) + 1
                end
            end
        end
    end

    -- Füge die Datenzeilen hinzu
    for _, classData in ipairs(datasets[1].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 = tableNode:tag('tr')
            row:tag('td'):wikitext(class)
            row:tag('td'):wikitext(subClass)
            for _, dataset in ipairs(datasets) do
                local sum = columnSums[subClass][dataset.schemas.Dataset['date']] or "N/A"
                local count = columnCounts[subClass][dataset.schemas.Dataset['date']] or 0
                local average = count > 0 and sum / count or "N/A"
                row:tag('td'):wikitext(tostring(average))
            end
        end
    end
    return tostring(html)
end