Difference between revisions of "Module:Protection banner"
From Timelines
(add namespace parser) |
(expand category matching code - still buggy) |
||
Line 114: | Line 114: | ||
if not nskey and ns % 2 == 1 then | if not nskey and ns % 2 == 1 then | ||
nskey = 'talk' | nskey = 'talk' | ||
− | + | elseif not nskey then | |
nskey = 'all' | nskey = 'all' | ||
end | end | ||
Line 120: | Line 120: | ||
end | end | ||
− | function p.matchCategory(cats, protType, protLevel, namespace, expiry | + | function p.matchCategory(cats, protType, protLevel, namespace, reason, expiry) |
− | + | -- Preprocess parameters | |
+ | cats = cats or categories | ||
+ | protType = protType or 'all' | ||
+ | protLevel = protLevel or 'all' | ||
+ | namespace = p.parseNamespace(namespace) | ||
+ | reason = reason or 'all' | ||
+ | if not expiry then | ||
+ | expiry = 'all' | ||
+ | elseif expiry ~= 'indef' then | ||
+ | expiry = 'temp' | ||
+ | end | ||
+ | |||
+ | -- Define the key table and the order to test it in | ||
+ | local keyTable = {protType, protLevel, namespace, reason, expiry} | ||
+ | local tryOrder = {5, 4, 3, 2, 1} | ||
+ | |||
+ | -- Try successively generic matches until we run out of key combinations | ||
+ | local function attemptMatch() | ||
+ | local key = table.concat(keyTable, '-') | ||
+ | mw.log(key) -- for debugging | ||
+ | return cats[key] | ||
+ | end | ||
+ | local attempt = attemptMatch() | ||
+ | if attempt then | ||
+ | return attempt | ||
+ | end | ||
+ | for i, keyTableKey in ipairs(tryOrder) do | ||
+ | keyTable[keyTableKey] = 'all' | ||
+ | attempt = attemptMatch() | ||
+ | if attempt then | ||
+ | return attempt | ||
+ | end | ||
+ | end | ||
+ | error('No category match found; please define the category for key "all-all-all-all-all"') | ||
end | end | ||
return p | return p |
Revision as of 17:51, 3 March 2014
Documentation for this module may be created at Module:Protection banner/doc
-- This module implements {{pp-meta}} and its daughter templates such as -- {{pp-dispute}}, {{pp-vandalism}} and {{pp-sock}}. local categories = { ['edit-autoconfirmed-user-all-all'] = 'Wikipedia semi-protected user and user talk pages', ['edit-autoconfirmed-project-all-all'] = 'Semi-protected project pages', ['edit-autoconfirmed-file-all-all'] = 'Semi-protected images', ['edit-autoconfirmed-template-all-all'] = 'Wikipedia semi-protected templates', ['edit-autoconfirmed-portal-all-all'] = 'Semi-protected portals', ['edit-autoconfirmed-talk-all-all'] = 'Semi-protected talk pages', ['edit-autoconfirmed-all-vandalism-all'] = 'Wikipedia pages semi-protected against vandalism', ['edit-sysop-user-all-all'] = 'Wikipedia protected user and user talk pages', ['edit-sysop-file-all-all'] = 'Protected images', ['edit-sysop-project-all-all'] = 'Protected project pages', ['edit-sysop-template-all-all'] = 'Wikipedia protected templates', ['edit-sysop-talk-all-all'] = 'Protected talk pages', ['edit-sysop-all-vandalism-all'] = 'Wikipedia pages protected against vandalism', ['edit-autoconfirmed-all-dispute-all'] = 'Wikipedia pages semi-protected due to dispute', ['edit-sysop-all-dispute-all'] = 'Wikipedia pages protected due to dispute', ['move-sysop-all-dispute-all'] = 'Wikipedia pages move-protected due to dispute', ['move-sysop-user-all-all'] = 'Wikipedia move-protected user and user talk pages', ['move-sysop-user-all-all'] = 'Wikipedia move-protected project pages', ['move-sysop-portal-all-all'] = 'Wikipedia move-protected portals', ['edit-autoconfirmed-template-all-all'] = 'Wikipedia semi-protected templates', ['move-sysop-template-all-all'] = 'Wikipedia move-protected templates', ['edit-all-template-all-all'] = 'Wikipedia protected templates', ['move-sysop-template-all-all'] = 'Wikipedia move-protected portals', ['edit-autoconfirmed-all-sock-all'] = 'Wikipedia pages semi-protected from banned users', ['edit-sysop-all-sock-all'] = 'Wikipedia pages protected from banned users', ['edit-autoconfirmed-all-blp-temp'] = 'Wikipedia temporarily semi-protected biographies of living people', ['edit-sysop-all-blp-temp'] = 'Wikipedia temporarily protected biographies of living people', ['edit-autoconfirmed-all-blp-indef'] = 'Wikipedia indefinitely semi-protected biographies of living people', ['edit-sysop-all-blp-indef'] = 'Wikipedia indefinitely protected biographies of living people', ['edit-autoconfirmed-all-all-indef'] = 'Wikipedia indefinitely semi-protected pages', ['move-sysop-talk-all-all'] = 'Wikipedia move-protected talk pages', ['move-sysop-all-all-indef'] = 'Wikipedia indefinitely move-protected pages', ['edit-autoconfirmed-all-all-all'] = 'Wikipedia semi-protected pages', ['all-all-all-all-all'] = 'Wikipedia protected pages', ['edit-autoconfirmed-category-all-all'] = 'Wikipedia semi-protected categories', ['edit-sysop-all-all-all'] = 'Wikipedia protected categories', ['move-sysop-all-all-all'] = 'Wikipedia move-protected pages', ['all-all-all-office-all'] = 'Wikipedia Office-protected pages', ['pc-autoconfirmed-all-all-all'] = 'Wikipedia pending changes protected pages (level 1)', ['pc-reviewer-all-all-all'] = 'Wikipedia pending changes protected pages (level 2)', } local nskeys = { [2] = 'user', [3] = 'user', [4] = 'project', [6] = 'file', [10] = 'template', [12] = 'project', [14] = 'category', [100] = 'portal', } local error_categories = { incorrect = 'Wikipedia pages with incorrect protection templates', no_expiry = 'Wikipedia protected pages without expiry', create = 'Wikipedia pages tagged as create-protected' } local template_types = { vandalism = { namespaces = false, -- defaults to all namespaces editLevels = false, -- all edit protection levels moveLevels = false, -- all move protection levels createLevels = false, -- all create protection levels header = false, reason = 'due to [[Wikipedia:Vandalism|vandalism]]', icon_reason = 'due to vandalism', } } -------------------------------------------------------------------------------- -- Main functions -------------------------------------------------------------------------------- local getArgs = require('Module:Arguments').getArgs local yesno = require('Module:Yesno') local p = {} function p.main(frame) -- Get data local args = getArgs(frame) local title = p.getTitle(args.page) local protType, protLevel = p.getProtectionStatus(title) local namespace = p.getNamespace(title) local expiry = p.getExpiry(args.expiry) local reason = p.getReason(args.reason) -- Sort categories and build a category link with the best match categories = p.sortCategories(categories, protType, protLevel, namespace, expiry, reason) local category = categories[1].name category = string.format('[[Category:%s]]', category) -- Find whether we are small or not and output a padlock or a banner as appropriate local isSmall = yesno(args.small, true) if isSmall then return p.exportPadlock() else return p.exportBanner() end end function p.parseNamespace(ns) -- The ns variable is the namespace number. if not ns or type(ns) ~= 'number' then return 'all' end local nskey = nskeys[ns] if not nskey and ns % 2 == 1 then nskey = 'talk' elseif not nskey then nskey = 'all' end return nskey end function p.matchCategory(cats, protType, protLevel, namespace, reason, expiry) -- Preprocess parameters cats = cats or categories protType = protType or 'all' protLevel = protLevel or 'all' namespace = p.parseNamespace(namespace) reason = reason or 'all' if not expiry then expiry = 'all' elseif expiry ~= 'indef' then expiry = 'temp' end -- Define the key table and the order to test it in local keyTable = {protType, protLevel, namespace, reason, expiry} local tryOrder = {5, 4, 3, 2, 1} -- Try successively generic matches until we run out of key combinations local function attemptMatch() local key = table.concat(keyTable, '-') mw.log(key) -- for debugging return cats[key] end local attempt = attemptMatch() if attempt then return attempt end for i, keyTableKey in ipairs(tryOrder) do keyTable[keyTableKey] = 'all' attempt = attemptMatch() if attempt then return attempt end end error('No category match found; please define the category for key "all-all-all-all-all"') end return p