imported>Johnuniq: update from sandbox to allow aliases

2022-02-26T04:29:51Z

update from sandbox to allow aliases

New page

local aliases = {
wikidata = 'www.wikidata',
meta = 'meta.wikimedia',
commons = 'commons.wikimedia',
foundation = 'foundation.wikimedia',
wikimania = 'wikimania.wikimedia',
wikitech = 'wikitech.wikimedia',
}

local function trimArg(arg, i)
arg = mw.text.trim(arg or '')
if arg == '' then
if i then
error('Parameter ' .. i .. ' is missing. See template documentation')
end
return nil
end
return mw.ustring.lower(arg)
end

local function getValue(stats, action, map)
if action == 'depth' then
-- https://meta.wikimedia.org/wiki/Wikipedia_article_depth
-- This gives silly results if, for example, the number of articles is small.
local n = { 'articles', 'edits', 'pages' }
if map then
for i, v in ipairs(n) do
n[i] = map[v]
end
end
for i, v in ipairs(n) do
n[i] = stats[v] or 0
end
local articles, edits, pages = n[1], n[2], n[3]
if pages == 0 or articles == 0 then
return 0
end
return math.floor((edits/pages) * ((pages - articles)/articles)^2)
end
if map then
action = map[action]
end
return stats[action]
end

local function getIfLocal(site, action)
-- If wanted site is the local site where module is running,
-- return numberof result for given action, or nil.
-- This is faster than reading the cached table, and gives the current value.
local localSite = string.match(mw.site.server, '.*//(.*)%.org$') -- examples: 'af.wikipedia', 'commons.wikimedia'
if site == localSite then
if action == 'activeusers' then
action = 'activeUsers'
end
return getValue(mw.site.stats, action)
end
end

local function main(frame)
local metaWords = { active = true, closed = true, languages = true, }
local args = frame:getParent().args
local action = trimArg(args[1], 1) -- activeusers, admins, articles, edits, files, pages, users, depth, active, closed, languages
if action:sub(1, 8) == 'numberof' then -- numberofX is an alias for X
action = trimArg(action:sub(9), 1)
end
local wantMeta = metaWords[action]
local site = trimArg(args[2], 2)
site = aliases[site] or site
if not wantMeta and not site:find('.', 1, true) then
-- site is like "af" or "af.wikipedia" or "af.wikiquote" etc., including "total"
site = site .. '.wikipedia'
end
local wantComma = trimArg(args[3]) -- nil for no commas in output; "N" or anything nonblank inserts commas
local result
if wantMeta then
local data = mw.loadData('Module:NUMBEROF/meta')
local nrActive = data.nrActive[site]
local nrClosed = data.nrClosed[site]
if nrActive or nrClosed then
-- If either is set, site is valid but there may not be an entry for both active and closed.
nrActive = nrActive or 0
nrClosed = nrClosed or 0
if action == 'active' then
result = nrActive
elseif action == 'closed' then
result = nrClosed
elseif action == 'languages' then
result = nrActive + nrClosed
end
end
else
result = getIfLocal(site, action)
if not result then
local data = mw.loadData('Module:NUMBEROF/data')
local map = data.map
data = data.data
result = data[site]
if result then
result = getValue(result, action, map)
end
end
end
if result then
if wantComma then
result = mw.language.getContentLanguage():formatNum(result)
end
return result -- number or formatted string
end
return -1
end

local function rank(frame)
-- Rank sites in a specified sister project by their number of articles.
local args = frame:getParent().args
local parm = trimArg(args[1], 1) -- a number like 12 or a site name like "af" (not "af.wikipedia")
local base = trimArg(args[2]) or 'wikipedia' -- base of full site name like "wikipedia" or "wikiquote"
local wantComma = trimArg(args[3])
local data = mw.loadData('Module:NUMBEROF/' .. (base == 'wikipedia' and 'rank' or 'other'))
data = data[base]
if data then
local result
parm = tonumber(parm) or parm
if type(parm) == 'number' then
result = data.rankByIndex[parm]
else
result = data.rankBySite[parm]
if result and wantComma then
result = mw.getContentLanguage():formatNum(result)
end
end
if result then
return result -- number or string
end
end
return -1
end

return {
main = main,
rank = rank,
}

Module:NUMBEROF - Revision history

imported>Johnuniq: update from sandbox to allow aliases