Difference between revisions of "Module:Election results"

From blackwiki
Jump to navigation Jump to search
blackwiki>Frietjes
(per talk page)
blackwiki>Frietjes
(per talk page)
Line 13: Line 13:
 
local row, secondrow
 
local row, secondrow
 
local tracking = ''
 
local tracking = ''
local max_rows = 20
+
local max_rows = 0
local over_max_rows = max_rows
 
 
 
 
-- helper functions
 
-- helper functions
Line 45: Line 44:
  
 
-- preprocess the input
 
-- preprocess the input
for i = 1, max_rows do
+
local stop_flag = false
for k, key in ipairs({'cand', 'vp', 'party', 'sc', 'seats'}) do
+
local i = 0
if args[key .. i] then
+
while stop_flag == false do
headings[key] = true
+
stop_flag = true
 +
for kk = 1, 20 do
 +
i = i + 1
 +
for k, key in ipairs({'cand', 'vp', 'party', 'sc', 'seats'}) do
 +
if args[key .. i] then
 +
headings[key] = true
 +
stop_flag = false
 +
max_rows = i > max_rows and i or max_rows
 +
end
 +
end
 +
if args['row' .. i] then
 +
stop_flag = false
 +
max_rows = i > max_rows and i or max_rows
 
end
 
end
end
+
 
-- temporary tracking
+
if args['cand' .. i] or args['party' .. i] then
for k, key in ipairs({'party', 'cand', 'vp'}) do
+
table.insert(index, i)
if args[key .. i] then
+
if args['votes' .. i] then
if unlink(args[key .. i]) == args[key .. i] then
+
local votesi = tonumber(args['votes' .. i]) or 0
tracking = tracking .. '[[Category:Pages using election results with unlinked ' .. key .. '|' .. i .. ']]'
+
args['votes' .. i] = votesi
 +
if votesi > winner_votes[1] then
 +
winner[1] = i
 +
winner_votes[1] = votesi
 +
end
 +
valid[1] = valid[1] + votesi
 +
end
 +
if args['votes' .. i .. '_2'] then
 +
rounds = 2
 +
local votesi = tonumber(args['votes' .. i .. '_2']) or 0
 +
args['votes' .. i .. '_2'] = votesi
 +
if votesi > winner_votes[2] then
 +
winner[2] = i
 +
winner_votes[2] = votesi
 +
end
 +
valid[2] = valid[2] + votesi
 
end
 
end
end
+
if args['seats' .. i] then
end
+
seats = seats + (tonumber(args['seats' .. i]) or 0)
if args['cand' .. i] or args['party' .. i] then
 
table.insert(index, i)
 
if args['votes' .. i] then
 
local votesi = tonumber(args['votes' .. i]) or 0
 
args['votes' .. i] = votesi
 
if votesi > winner_votes[1] then
 
winner[1] = i
 
winner_votes[1] = votesi
 
 
end
 
end
valid[1] = valid[1] + votesi
+
if args['sc' .. i] then
end
+
seatdiff = seatdiff + tonumdash(args['sc' .. i])
if args['votes' .. i .. '_2'] then
 
rounds = 2
 
local votesi = tonumber(args['votes' .. i .. '_2']) or 0
 
args['votes' .. i .. '_2'] = votesi
 
if votesi > winner_votes[2] then
 
winner[2] = i
 
winner_votes[2] = votesi
 
 
end
 
end
valid[2] = valid[2] + votesi
 
end
 
if args['seats' .. i] then
 
seats = seats + (tonumber(args['seats' .. i]) or 0)
 
end
 
if args['sc' .. i] then
 
seatdiff = seatdiff + tonumdash(args['sc' .. i])
 
 
end
 
end
 
end
 
end
Line 91: Line 95:
 
-- optional against all
 
-- optional against all
 
if args['against_all'] or args['against_all2'] then
 
if args['against_all'] or args['against_all2'] then
over_max_rows = over_max_rows + 1
+
max_rows = max_rows + 1
local i = over_max_rows
+
local i = max_rows
 
table.insert(index, i)
 
table.insert(index, i)
 
args['votes' .. i] = tonumber(args['against_all']) or 0
 
args['votes' .. i] = tonumber(args['against_all']) or 0
Line 98: Line 102:
 
args['colour' .. i] = 'inherit'
 
args['colour' .. i] = 'inherit'
 
args['color' .. i] = 'inherit'
 
args['color' .. i] = 'inherit'
args['cand' .. i] = 'Against all'
+
args['row' .. i] = 'Against all'
 
valid[1] = valid[1] + args['votes' .. i]
 
valid[1] = valid[1] + args['votes' .. i]
 
valid[2] = valid[2] + args['votes' .. i .. '_2']
 
valid[2] = valid[2] + args['votes' .. i .. '_2']
 +
tracking = tracking .. '[[Category:Pages using election results with against all]]'
 
end
 
end
+
 
 
local ovalid = {valid[1], valid[2]}
 
local ovalid = {valid[1], valid[2]}
 +
seats = (args['total_seats'] or '') == 'TOTAL' and seats or args['total_seats']
 +
seatdiff = (args['total_sc'] or '') == 'TOTAL' and seatdiff or args['total_sc']
 
if args['valid'] or ((rounds > 1) and args['valid2']) then
 
if args['valid'] or ((rounds > 1) and args['valid2']) then
over_max_rows = over_max_rows + 1
+
max_rows = max_rows + 1
local i = over_max_rows
+
local i = max_rows
 
table.insert(index,i)
 
table.insert(index,i)
 
args['votes' .. i] = valid[1]
 
args['votes' .. i] = valid[1]
Line 112: Line 119:
 
args['colour' .. i] = 'inherit'
 
args['colour' .. i] = 'inherit'
 
args['color' .. i] = 'inherit'
 
args['color' .. i] = 'inherit'
args['cand' .. i] = 'Total'
+
args['row' .. i] = 'Total'
 
args['seats' .. i] = seats
 
args['seats' .. i] = seats
 
args['sc' .. i] = seatdiff
 
args['sc' .. i] = seatdiff
Line 132: Line 139:
 
row
 
row
 
:tag('th')
 
:tag('th')
:wikitext(headings['vp'] and 'President' or 'Candidate')
+
:wikitext('Candidate')
 
:attr('scope', 'col')
 
:attr('scope', 'col')
 
:attr('colspan', 2)
 
:attr('colspan', 2)
Line 140: Line 147:
 
row
 
row
 
:tag('th')
 
:tag('th')
:wikitext('Vice president')
+
:wikitext('Running mate')
 
:attr('scope', 'col')
 
:attr('scope', 'col')
 
:attr('rowspan', rowspan)
 
:attr('rowspan', rowspan)
Line 218: Line 225:
 
end
 
end
  
if v > max_rows then
+
if args['row' .. v] then
 
row
 
row
 
:tag('td')
 
:tag('td')
:wikitext(args['cand' .. v])
+
:wikitext(args['row' .. v])
 
:attr('colspan', cs)
 
:attr('colspan', cs)
 
else
 
else
Line 341: Line 348:
 
:wikitext(pct(1,1))
 
:wikitext(pct(1,1))
 
   end
 
   end
if headings['seats'] and not args['valid'] then
+
  if headings['seats'] or headings['sc'] then
row
+
  if not args['valid'] and (seats or seatdiff) then
:tag('td')
+
if headings['seats'] then
:wikitext(seats)
+
row
end
+
:tag('td')
if headings['sc'] and not args['valid'] then
+
:wikitext(seats)
row
+
end
:tag('td')
+
if headings['sc'] then
:wikitext(seatdiff)
+
row
end
+
:tag('td')
if (headings['seats'] or headings['sc']) and args['valid'] then
+
:wikitext(seatdiff)
local cspan = (headings['seats'] and 1 or 0) + (headings['sc'] and 1 or 0)
+
end
row
+
else
:tag('td')
+
local cspan = (headings['seats'] and 1 or 0) + (headings['sc'] and 1 or 0)
:attr('colspan', cspan > 1 and cspan or nil)
+
row
 +
:tag('td')
 +
:attr('colspan', cspan > 1 and cspan or nil)
 +
end
 
end
 
end
 
row = root:tag('tr')
 
row = root:tag('tr')

Revision as of 15:43, 24 August 2020

Implements {{Election results}}


require('Module:No globals')
local p = {}

function p.main(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local index, headings = {}, {}
	local cols, rounds = 0, 1
	local winner, winner_votes = {0, 0}, {0, 0}
	local valid = {0, 0}
	local invalid = {tonumber(args.invalid) or 0, tonumber(args.invalid2) or 0}
	local electorate = {tonumber(args.electorate) or 0, tonumber(args.electorate2) or tonumber(args.electorate) or 0}
	local seats, seatdiff = 0, 0
	local row, secondrow
	local tracking = ''
	local max_rows = 0
	
	-- helper functions
	local lang = mw.getContentLanguage()
	local function fmt(n)
		return n and tonumber(n) and lang:formatNum(tonumber(n)) or nil
	end
	local function pct(n,d)
		n, d = tonumber(n), tonumber(d)
		if n and d and d > 0 then
			return string.format('%.2f', n / d * 100)
		end
		return '–'
	end
	local function tonumdash(s)
		if s then
			s = mw.ustring.gsub(s, '&[MmNn][Dd][Aa][Ss][Hh];', '-')
			s = mw.ustring.gsub(s, '&[Mm][Ii][Nn][Uu][Ss];', '-')
			s = mw.ustring.gsub(s, '[—–−]', '-')
			return tonumber(s) or 0
		end
	end
	local function unlink(s)
		if s then
			s = s:match("^.-%[%[(.-)|.-%]%].*$") or s
			s = s:match("^.-%[%[(.-)%]%].*$") or s
		end
		return s
	end

	-- preprocess the input
	local stop_flag = false
	local i = 0
	while stop_flag == false do
		stop_flag = true
		for kk = 1, 20 do
			i = i + 1
			for k, key in ipairs({'cand', 'vp', 'party', 'sc', 'seats'}) do
				if args[key .. i] then
					headings[key] = true
					stop_flag = false
					max_rows = i > max_rows and i or max_rows
				end
			end
			if args['row' .. i] then
				stop_flag = false
				max_rows = i > max_rows and i or max_rows
			end

			if args['cand' .. i] or args['party' .. i] then
				table.insert(index, i)
				if args['votes' .. i] then
					local votesi = tonumber(args['votes' .. i]) or 0
					args['votes' .. i] = votesi
					if votesi > winner_votes[1] then
						winner[1] = i
						winner_votes[1] = votesi
					end
					valid[1] = valid[1] + votesi
				end
				if args['votes' .. i .. '_2'] then
					rounds = 2
					local votesi = tonumber(args['votes' .. i .. '_2']) or 0
					args['votes' .. i .. '_2'] = votesi
					if votesi > winner_votes[2] then
						winner[2] = i
						winner_votes[2] = votesi
					end
					valid[2] = valid[2] + votesi
				end
				if args['seats' .. i] then
					seats = seats + (tonumber(args['seats' .. i]) or 0)
				end
				if args['sc' .. i] then
					seatdiff = seatdiff + tonumdash(args['sc' .. i])
				end
			end
		end
	end
	
	-- optional against all
	if args['against_all'] or args['against_all2'] then
		max_rows = max_rows + 1
		local i = max_rows
		table.insert(index, i)
		args['votes' .. i] = tonumber(args['against_all']) or 0
		args['votes' .. i .. '_2'] = tonumber(args['against_all_2']) or 0
		args['colour' .. i] = 'inherit'
		args['color' .. i] = 'inherit'
		args['row' .. i] = 'Against all'
		valid[1] = valid[1] + args['votes' .. i]
		valid[2] = valid[2] + args['votes' .. i .. '_2']
		tracking = tracking .. '[[Category:Pages using election results with against all]]'
	end

	local ovalid = {valid[1], valid[2]}
	seats = (args['total_seats'] or '') == 'TOTAL' and seats or args['total_seats']
	seatdiff = (args['total_sc'] or '') == 'TOTAL' and seatdiff or args['total_sc']
	if args['valid'] or ((rounds > 1) and args['valid2']) then
		max_rows = max_rows + 1
		local i = max_rows
		table.insert(index,i)
		args['votes' .. i] = valid[1]
		args['votes' .. i .. '_2'] = valid[2]
		args['colour' .. i] = 'inherit'
		args['color' .. i] = 'inherit'
		args['row' .. i] = 'Total'
		args['seats' .. i] = seats
		args['sc' .. i] = seatdiff
		args['font-weight' .. i] = 'bold'
		ovalid[1] = tonumber(args['valid']) or 0
		ovalid[2] = tonumber(args['valid2']) or 0
	end

	-- build the table
	local root = mw.html.create('table')
	root
		:addClass('wikitable sortable')
		:tag('caption')
			:wikitext(args.caption)

	local rowspan = (rounds > 1) and 2 or nil
	row = root:tag('tr')
	if headings['cand'] then
		row
			:tag('th')
				:wikitext('Candidate')
				:attr('scope', 'col')
				:attr('colspan', 2)
				:attr('rowspan', rowspan)
		cols = cols + 2
		if headings['vp'] then
			row
				:tag('th')
						:wikitext('Running mate')
						:attr('scope', 'col')
						:attr('rowspan', rowspan)
			cols = cols + 1
		end
		if headings['party'] then
			row
				:tag('th')
				:wikitext('Party')
				:attr('scope', 'col')
				:attr('rowspan', rowspan)
			cols = cols + 1
		end
	else
		row
			:tag('th')
				:wikitext(headings['party'] and 'Party' or '')
				:attr('scope', 'col')
				:attr('colspan', 2)
				:attr('rowspan', rowspan)
		cols = cols + 2
	end
	if rounds > 1 then
		row
			:tag('th')
				:wikitext('First round')
				:attr('scope', 'col')
				:attr('colspan', 2)
			:tag('th')
				:wikitext('Second round')
				:attr('scope', 'col')
				:attr('colspan', 2)
		secondrow = root:tag('tr')
	else
		secondrow = row
	end
	for k=1,rounds do
		secondrow
			:tag('th')
				:wikitext('Votes')
				:attr('scope', 'col')
			:tag('th')
				:wikitext('%')
				:attr('scope', 'col')
		cols = cols + 2
	end
	if headings['seats'] then
		row
			:tag('th')
				:wikitext('Seats')
				:attr('scope', 'col')
				:attr('rowspan', rowspan)
		cols = cols + 1
	end
	if headings['sc'] then
		row
			:tag('th')
				:wikitext(headings['seats'] and '+/–' or 'Seats±')
				:attr('scope', 'col')
				:attr('rowspan', rowspan)
		cols = cols + 1
	end
	local cs = cols - 2*rounds 
			- (headings['seats'] and 1 or 0)
			- (headings['sc'] and 1 or 0)
	local rsuff = (rounds > 1) and {'', '_2'} or {''}
	for i, v in ipairs(index) do
		row = root:tag('tr'):css('font-weight', args['font-weight' .. v])

		-- determine the color
		local color = args['colour' .. v] or args['color' .. v] or nil
		if color == nil then
			local party = unlink(args['party' .. v]) or ''
			if party ~= '' and mw.title.new('Template:' .. party .. '/meta/color').exists then
				color = frame:expandTemplate{title = party .. '/meta/color'}
			end
		end

		if args['row' .. v] then
			row
				:tag('td')
					:wikitext(args['row' .. v])
					:attr('colspan', cs)
		else
			-- create the empty color cell
			row
				:tag('td')
					:css('background-color', color)

			-- add the rest of the row
			if headings['cand'] then
				row
					:tag('td')
						:wikitext(args['cand' .. v])
				if headings['vp'] then
					row
						:tag('td')
							:wikitext(args['vp' .. v])
				end
			end
			if headings['party'] then
				row
					:tag('td')
						:wikitext(args['party' .. v])
			end
		end
		for kk, suf in ipairs(rsuff) do
			if(args['votes' .. v .. suf]) then
				row:tag('td')
					:css('text-align', 'right')
					:wikitext(fmt(args['votes' .. v .. suf]))
				row:tag('td')
					:css('text-align', 'right')
					:wikitext(pct(args['votes' .. v .. suf], valid[kk]))
			else
				row:tag('td'):attr('colspan', 2)
			end
		end
		if headings['seats'] then
			row
				:tag('td')
					:css('text-align', 'right')
					:wikitext(fmt(args['seats' .. v]))
		end
		if headings['sc'] then
			row
				:tag('td')
					:css('text-align', 'right')
					:wikitext(args['sc' .. v])
		end
	end

	row = root
		:tag('tr')
			:addClass('sortbottom')
			:css('background', '#eaecf0')
			:css('text-align', 'right')
	row
		:tag('th')
			:wikitext('Valid votes')
			:attr('scope', 'row')
			:attr('colspan', cs)
	        :css('text-align', 'left')
	        :css('font-weight', 'normal')
	for k=1,rounds do
		row
			:tag('td')
				:wikitext(fmt(ovalid[k]))
			:tag('td')
				:wikitext(pct(ovalid[k], ovalid[k] + invalid[k]))
	end
	if headings['seats'] or headings['sc'] then
		local cspan = (headings['seats'] and 1 or 0) + (headings['sc'] and 1 or 0)
		row
			:tag('td')
				:attr('colspan', cspan > 1 and cspan or nil)
	end
	
	row = root:tag('tr')
			:addClass('sortbottom')
			:css('background', '#eaecf0')
			:css('text-align', 'right')
	row
		:tag('th')
			:wikitext('Invalid votes')
			:attr('scope', 'row')
			:attr('colspan', cs)
 	        :css('text-align', 'left')
	        :css('font-weight', 'normal')
   	for k=1,rounds do
		row
			:tag('td')
				:wikitext(fmt(invalid[k]))
			:tag('td')
				:wikitext(pct(invalid[k], ovalid[k] + invalid[k]))
   	end
	if headings['seats'] or headings['sc'] then
		local cspan = (headings['seats'] and 1 or 0) + (headings['sc'] and 1 or 0)
		row
			:tag('td')
				:attr('colspan', cspan > 1 and cspan or nil)
	end
	
	row = root:tag('tr')
			:addClass('sortbottom')
			:css('font-weight', 'bold')
			:css('background', '#eaecf0')
			:css('text-align', 'right')
	row
		:tag('th')
			:wikitext('Total votes')
			:attr('scope', 'row')
			:attr('colspan', cs)
	        :css('text-align', 'left')
   	for k=1,rounds do
		row
			:tag('td')
				:wikitext(fmt(ovalid[k] + invalid[k]))
			:tag('td')
				:wikitext(pct(1,1))
   	end
   	if headings['seats'] or headings['sc'] then
	   	if not args['valid'] and (seats or seatdiff) then
			if headings['seats'] then
				row
					:tag('td')
						:wikitext(seats)
			end
			if headings['sc'] then
				row
					:tag('td')
						:wikitext(seatdiff)
			end
		else
			local cspan = (headings['seats'] and 1 or 0) + (headings['sc'] and 1 or 0)
			row
				:tag('td')
					:attr('colspan', cspan > 1 and cspan or nil)
		end
	end
	row = root:tag('tr')
			:addClass('sortbottom')
			:css('background', '#eaecf0')
			:css('text-align', 'right')
	row
		:tag('th')
			:wikitext('Registered voters/turnout')
			:attr('scope', 'row')
			:attr('colspan', cs)
 			:css('text-align', 'left')
	        :css('font-weight', 'normal')
   	for k=1,rounds do
		row:tag('td')
			:wikitext(fmt(electorate[k]))
		row:tag('td')
			:wikitext(pct(ovalid[k] + invalid[k], electorate[k]))
   	end
	if headings['seats'] or headings['sc'] then
		local cspan = (headings['seats'] and 1 or 0) + (headings['sc'] and 1 or 0)
   		row
			:tag('td')
				:attr('colspan', cspan > 1 and cspan or nil)
	end

	row = root:tag('tr')
			:addClass('sortbottom')
			:css('background', '#eaecf0')
			:css('text-align', 'right')
	row:tag('td')
		:wikitext('Source: ', args.source)
		:attr('colspan', cols)
 		:css('text-align', 'left')
 		
	return tostring(root) .. tracking
end

return p