Module:ElectionStats

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

local json = mw.text.jsonDecode
local p = {}

local electionColors = {
    General = "#4682b4", -- blue
    Local = "#8a2be2", -- purple
    Statewide = "#008000", -- green
    Referendum = "#ffa500", -- orange
    Total = "#808080" -- gray
}

function p.generateCards(frame)
    local dataPage = 'Data:Election Data.json'
    local success, data = p.loadJson(dataPage)
    if not success then
        return "could not load JSON data from " .. dataPage .. "!"
    end

    local currentTime = os.time()

    local counts = { Local = 0, General = 0, Statewide = 0, Referendum = 0 }
    local total = 0

    for _, elections in pairs(data) do
        for _, election in ipairs(elections) do
            local electionType = election.election_type
            local electionDate = p.parseDate(election.date)
            if electionDate and electionDate <= currentTime then
                if counts[electionType] then
                    counts[electionType] = counts[electionType] + 1
                    total = total + 1
                end
            end
        end
    end

    local cards = {}
    for electionType, count in pairs(counts) do
        if count > 0 then
            table.insert(cards, p.generateCard(electionType, count))
        end
    end

    table.insert(cards, p.generateCard("Total", total))

    local electionGrid = '<div class="election-grid" style="display: flex; justify-content: center; align-items: center; flex-wrap: wrap; gap: 20px; padding: 20px;">\n' .. table.concat(cards, "\n") .. '\n</div>'
    return electionGrid
end

function p.loadJson(page)
    local title = mw.title.new(page)
    if not title or not title.exists then
        return false, nil
    end
    local content = title:getContent()
    return true, json(content)
end

function p.parseDate(dateString)
    local pattern = "(%d+)-(%d+)-(%d+)T(%d+):(%d+):(%d+)"
    local year, month, day, hour, min, sec = dateString:match(pattern)
    if year and month and day and hour and min and sec then
        return os.time({
            year = tonumber(year),
            month = tonumber(month),
            day = tonumber(day),
            hour = tonumber(hour),
            min = tonumber(min),
            sec = tonumber(sec)
        })
    else
        return nil
    end
end

function p.generateCard(electionType, count)
    local color = electionColors[electionType] or "#1C408C" -- default color

    return string.format(
        [[
<div class="election-card" style="width: 200px; height: 250px; background-color: %s; color: white; text-align: center; border: 1px solid %s; padding: 10px; font-family: Arial, sans-serif; display: flex; flex-direction: column; justify-content: center; align-items: center;">
  <div style="font-size: 2em; font-weight: bold; line-height: 1.5;">%d</div>
  <div style="border-top: 2px solid white; width: 80%%; margin: 10px 0;"></div>
  <div style="font-size: 1em; text-transform: uppercase; line-height: 1.5;">%s Elections</div>
</div>
        ]],
        color, -- background color
        color, -- border color
        count,
        electionType
    )
end

return p