Jump to content

Module:Articles by quality

Permanently protected module
From Wikipedia, the free encyclopedia

require('strict')
local p = {}
local classes = {'FA', 'A', 'GA', 'B', 'C', 'Start', 'Stub', 'FL', 'AL', 'BL', 'CL', 'List', 'SIA', 'Future', 'Category', 'Disambig', 'Draft', 'FM', 'File', 'Needed', 'Portal', 'Project', 'Redirect', 'Template', 'User', 'NA'}

p.main = function(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local title = args.page and mw.title.new(page) or mw.title.getCurrentTitle()
	local class, topic, typ = title.text:match('^(%a+)-Class (.+) (%a+)$')
	if not class then -- try to match unassessed category
		class, topic, typ = title.text:match('^(Unassessed) (.+) (%a+)$')
	end
	local out, exist, cats = '', {}, {}
	local add_category = function(cat, sort)
		local link = '[[Category:' .. cat .. (sort and ('|' .. sort) or '') .. ']]'
		table.insert(cats, link)
	end
	if args.class and class and args.topic and topic and (args.class:lower()~=class:lower() or args.topic~=topic) then
		add_category('WikiProject assessment categories needing attention')
	end
	topic = args.topic or topic or ''
	class = args.class or class or ''
	typ = typ or 'articles'
	if title.namespace==14 then
		out = frame:expandTemplate{title='Possibly empty category'}
	end
	local add_category = function(cat, sort)
		local link = '[[Category:' .. cat .. (sort and ('|' .. sort)) .. ']]'
		table.insert(cats, link)
	end
	local cat_in_use = function(cat)
		local title = mw.title.new('Category:' .. cat)
		return title.exists or mw.site.stats.pagesInCategory(cat, 'pages')>0
	end
	for _, class in ipairs(classes) do
		if cat_in_use(class .. '-Class' .. ' ' .. topic .. ' pages') then
			exist[class] = 'pages'
		elseif cat_in_use(class .. '-Class' .. ' ' .. topic .. ' articles') then
			exist[class] = 'articles'
		else
			exist[class] = false
		end
	end
	local class_template = function(class, page)
		return frame:expandTemplate{
			title = 'class',
			args = {
				[1] = class,
				topic = topic,
				category = topic .. ' ' .. page,
				bold = 'no'
			}
		}
	end
	local header_row = mw.html.create('tr')
	for _, class in ipairs(classes) do
		if exist[class] then
			header_row:node(class_template(class, exist[class]))
		end
	end
	if args.custom1 then
		header_row:node(class_template(args.custom1, 'articles'))
	end
	if args.custom2 then
		header_row:node(class_template(args.custom2, 'articles'))
	end
	header_row:node(class_template('Unassessed', 'articles'))
	local pages_in_cat = function(cat)
		local pages = mw.site.stats.pagesInCategory(cat, 'pages')
		local col = mw.html.create('td')
			:attr('align', 'right')
			:wikitext(mw.language.getContentLanguage():formatNum(pages))
		return col
	end
	local second_row = mw.html.create('tr')
	for _, class in ipairs(classes) do
		if exist[class] then
			second_row:node(pages_in_cat(class .. '-Class' .. ' ' .. topic .. ' ' .. exist[class]))
		end
	end
	if args.custom1 then
		second_row:node(pages_in_cat(args.custom1 .. '-Class' .. ' ' .. topic .. ' articles'))
	end
	if args.custom2 then
		second_row:node(pages_in_cat(args.custom2 .. '-Class' .. ' ' .. topic .. ' articles'))
	end
	second_row:node(pages_in_cat('Unassessed ' .. topic .. ' articles'))
	local tab = mw.html.create('table')
		:addClass('toccolours'):addClass('nomobile')
		:css('table-layout', 'fixed')
		:css('margin', '1em auto')
		:node(header_row)
		:node(second_row)
	if class and class~='' and class~='nocat' then
		add_category(args.parent or (topic .. ' articles by quality'), class)
		add_category(class .. (class=='Unassessed' and '' or '-Class') .. ' ' .. typ, args.sort or topic)
	end
	return out .. tostring(tab) .. table.concat(cats)
end

return p