Module:WikiProject assessment progression/sandbox

This is an old revision of this page, as edited by Sasuke Sarutobi (talk | contribs) at 01:28, 4 December 2020 (Stripping classes down to just class names, as running totals are generated in _main). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
require('Module:No globals')

local classes = {
	'List','Stub','Start','C','B',
	'GA','A','FA','FL','FM',
	'Book','Category','Disambig','File',
	'Portal','Project','Redirect','Template',
}

local getArgs = require('Module:Arguments').getArgs
local p = {}

local function categoryCount(category, project)
	return mw.site.stats.pagesInCategory(
		string.format('%s %s articles', category, project),
		'pages'
	)
end

-- rounding to first decimal, from http://lua-users.org/wiki/SimpleRound
local function round(num)
	return math.floor(num * 10 + 0.5) / 10
end

local function percentComplete(sum, total)
	return tostring(round(100 * sum / total))
end

local function arg_or_default(args, from_arg, default)
	if args[from_arg] and args[from_arg] ~= '' then
		return mw.ustring.lower(args[from_arg])
	else
		return default
	end
end

function p.main(frame)
	local args = getArgs(frame)
	return p._main(args)
end

function p._main(args)
	local project = arg_or_default(args, "project", nil)
	local classCount = 0
	
	for _, class in ipairs(classes) do
		local category = class..'-Class'
		classCount = classCount + categoryCount(category, project)
	end
	
	local total = classCount + categoryCount('Unassessed', project)
	return percentComplete(classCount, total)
end

return p