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

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
    
    -- Number or percent
    local entity = frame.args.entity
    local suffix = ''
    if entity == 'percent' then
        suffix = '%'
    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')
    for _, dataset in ipairs(datasets) do
        local formattedDate = 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

    local columnSums = {}
    local columnCounts = {}
    -- Füge die Datenzeilen hinzu
    for _, classData in ipairs(datasets[1].schemas.Dataset.classes) do
        local class = classData['class name']
        for i, subClassData in ipairs(classData['sub-classes']) do
            local row = tableNode:tag('tr')
            row:tag('td'):wikitext(class)
            for j, subSubClassData in ipairs(subClassData['sub-sub-classes']) do
                local value = {}
                for k, dataset in ipairs(datasets) do
                    local val = tonumber(subSubClassData.value) or 0
                    table.insert(value, val)
                    
                    -- Berechne Summen und Anzahlen für jede Spalte
                    columnSums[k] = columnSums[k] or 0
                    columnCounts[k] = columnCounts[k] or 0
                    columnSums[k] = columnSums[k] + val
                    columnCounts[k] = columnCounts[k] + 1
                end
                row:tag('td'):wikitext(table.concat(value, ", ") .. suffix)
            end
        end
    end

    -- Füge die Totalzeile hinzu
    local totalRow = tableNode:tag('tr'):addClass('sortbottom')
    totalRow:tag('th'):wikitext('Total')
    for i, dataset in ipairs(datasets) do
        local totalValue
        if columnCounts[i] and columnCounts[i] > 0 then
            if entity == 'percent' then
                totalValue = columnSums[i] / columnCounts[i] -- Berechne Durchschnitt, falls Werte vorhanden sind
            else
                totalValue = columnSums[i]
            end
        else
            totalValue = 'N/A'
        end
        totalRow:tag('th'):wikitext(round(totalValue, 2) .. suffix) 
    end

    return tostring(html)
end