Difference between revisions of "Module:For loop/sandbox"

From blackwiki
Jump to navigation Jump to search
blackwiki>Frietjes
(Created page with 'p = {} local args local function callTemplate(template, targs) return mw.getCurrentFrame():expandTemplate{title = template, args = targs} end local fun...')
 
m (23 revisions imported)
 
(22 intermediate revisions by 5 users not shown)
Line 1: Line 1:
p = {}
+
-- This module implements {{for loop}}.
+
 
local args
+
local getArgs = require('Module:Arguments').getArgs
+
local yesno = require('Module:Yesno')
local function callTemplate(template, targs)
+
local p = {}
    return mw.getCurrentFrame():expandTemplate{title = template, args = targs}
+
 
 +
function p.main(frame)
 +
local args = getArgs(frame, {
 +
trim = false,
 +
removeBlanks = false
 +
})
 +
return p._main(args)
 
end
 
end
+
 
local function getArgNums(prefix, suffix)
+
function p._main(args)
    -- Returns a table containing the numbers of the arguments that exist
+
local template = args['call'] or 'void'
    -- for the specified prefix and suffix.
+
local calltemplates = yesno(args.substall or "", true) or not mw.isSubsting()
    local nums = {}
+
local variableParam = args.pv
    for k, v in pairs(args) do
+
variableParam = tonumber(variableParam) or variableParam or 1 -- fix for positional parameters
        local num = tostring(k):match('^' .. prefix .. '([1-9]%d*)' .. suffix .. '$')
+
local variableValPrefix = args.prefix or ''
        if num then table.insert(nums, tonumber(num)) end
+
local variableValPostfix = args.postfix or ''
    end
+
local sep = args[1] or ''
    table.sort(nums)
+
local constantArgs = p.getConstants(args)
    return nums
+
local variableVals = p.getVariableVals(args)
 +
 
 +
local result = ''
 +
local addSeparator = false;
 +
for _, v in ipairs(variableVals) do
 +
v = mw.text.trim(v)
 +
if #v > 0 or not yesno(args.skipBlanks) then
 +
if addSeparator then
 +
result = result .. sep
 +
end
 +
addSeparator = true;
 +
local targs = constantArgs
 +
targs[variableParam] = variableValPrefix .. v .. variableValPostfix
 +
if calltemplates then
 +
local output = p.callTemplate(template, targs)
 +
if #mw.text.trim(output) == 0 then
 +
addSeparator = false
 +
end
 +
result = result .. output
 +
else
 +
local makeTemplate = require('Module:Template invocation').invocation
 +
result = result .. makeTemplate(template, targs)
 +
end
 +
end
 +
end
 +
return result
 
end
 
end
+
 
local function getConstants()
+
function p.getConstants(args)
    local constantArgNums = getArgNums('pc', 'n')
+
local constantArgNums = p.getArgNums(args, 'pc', 'n')
    local constantArgs = {}
+
local constantArgs = {}
    for _, v in ipairs(constantArgNums) do
+
for _, num in ipairs(constantArgNums) do
        local keyArgName = 'pc' .. tostring(v) .. 'n'
+
local keyArg = 'pc' .. tostring(num) .. 'n'
        local valArgName = 'pc' .. tostring(v) .. 'v'
+
local valArg = 'pc' .. tostring(num) .. 'v'
        constantArgs[args[keyArgName]] = args[valArgName]
+
local key = args[keyArg]
    end
+
key = tonumber(key) or key
    return constantArgs
+
local value = args[valArg]
 +
constantArgs[key] = value
 +
end
 +
return constantArgs
 
end
 
end
+
 
local function getVariableVals()
+
function p.getVariableVals(args)
    local variableVals = {}
+
local variableVals = {}
    for i, v in ipairs(args) do
+
if args.start or args.stop or args.by then
        if i ~= 1 then
+
if args[2] then
            variableVals[i - 1] = v
+
error("Both start/stop/by and numbered parameters specified")
        end
+
end
    end
+
local start = tonumber(args.start or 1)
    return variableVals
+
local stop = tonumber(args.stop or 1)
 +
local by = tonumber(args.by or 1)
 +
for i = start, stop, by do
 +
variableVals [#variableVals + 1] = i
 +
end
 +
else
 +
for i, v in ipairs(args) do
 +
if i ~= 1 then
 +
variableVals[i - 1] = v
 +
end
 +
end
 +
end
 +
return variableVals
 
end
 
end
+
 
local function _main()
+
function p.getArgNums(args, prefix, suffix)
    local template = args['call'] or 'void'
+
-- Returns a table containing the numbers of the arguments that exist
    local variableParam = args.pv or '1'
+
-- for the specified prefix and suffix.
    local variableValPrefix = args.prefix or ''
+
local nums = {}
    local variableValPostfix = args.postfix or ''
+
local pattern = '^' .. prefix .. '([1-9]%d*)' .. suffix .. '$'
    local sep = args[1] or ''
+
for k, _ in pairs(args) do
    local constantArgs = getConstants()
+
local num = tostring(k):match(pattern)
    local variableVals = getVariableVals()
+
if num then
+
nums[#nums + 1] = tonumber(num)
    local result = ''
+
end
    for i, v in ipairs(variableVals) do
+
end
        v = mw.text.trim(v) -- trim whitespace
+
table.sort(nums)
        local targs = constantArgs
+
return nums
        targs[variableParam] = variableValPrefix .. v .. variableValPostfix
 
        result = result .. tostring(callTemplate(template, targs))
 
        if variableVals[i + 1] then
 
            result = result .. sep
 
        end
 
    end
 
    return result
 
 
end
 
end
+
 
function p.main(frame)
+
function p.callTemplate(template, targs)
    -- If called via #invoke, use the args passed into the invoking template.
+
return mw.getCurrentFrame():expandTemplate{title = template, args = targs}
    -- Otherwise, for testing purposes, assume args are being passed directly in.
 
    if frame == mw.getCurrentFrame() then
 
        args = frame:getParent().args
 
        for k, v in pairs(frame.args) do
 
            args = frame.args
 
            break
 
        end
 
    else
 
        args = frame
 
    end
 
    return _main()
 
 
end
 
end
+
 
 
return p
 
return p

Latest revision as of 16:40, 26 September 2020

Documentation for this module may be created at Module:For loop/sandbox/doc

-- This module implements {{for loop}}.

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

function p.main(frame)
	local args = getArgs(frame, {
		trim = false,
		removeBlanks = false
	})
	return p._main(args)
end

function p._main(args)
	local template = args['call'] or 'void'
	local calltemplates = yesno(args.substall or "", true) or not mw.isSubsting()
	local variableParam = args.pv
	variableParam = tonumber(variableParam) or variableParam or 1 -- fix for positional parameters
	local variableValPrefix = args.prefix or ''
	local variableValPostfix = args.postfix or ''
	local sep = args[1] or ''
	local constantArgs = p.getConstants(args)
	local variableVals = p.getVariableVals(args)

	local result = ''
	local addSeparator = false;
	for _, v in ipairs(variableVals) do
		v = mw.text.trim(v)
		if #v > 0 or not yesno(args.skipBlanks) then
			if addSeparator then
				result = result .. sep
			end
			addSeparator = true;
			local targs = constantArgs
			targs[variableParam] = variableValPrefix .. v .. variableValPostfix
			if calltemplates then
				local output = p.callTemplate(template, targs)
				if #mw.text.trim(output) == 0 then
					addSeparator = false
				end
				result = result .. output
			else
				local makeTemplate = require('Module:Template invocation').invocation
				result = result .. makeTemplate(template, targs)
			end
		end
	end
	return result
end

function p.getConstants(args)
	local constantArgNums = p.getArgNums(args, 'pc', 'n')
	local constantArgs = {}
	for _, num in ipairs(constantArgNums) do
		local keyArg = 'pc' .. tostring(num) .. 'n'
		local valArg = 'pc' .. tostring(num) .. 'v'
		local key = args[keyArg]
		key = tonumber(key) or key
		local value = args[valArg]
		constantArgs[key] = value
	end
	return constantArgs
end

function p.getVariableVals(args)
	local variableVals = {}
	if args.start or args.stop or args.by then
		if args[2] then
			error("Both start/stop/by and numbered parameters specified")
		end
		local start = tonumber(args.start or 1)
		local stop = tonumber(args.stop or 1)
		local by = tonumber(args.by or 1)
		for i = start, stop, by do
			variableVals [#variableVals + 1] = i
		end
	else
		for i, v in ipairs(args) do
			if i ~= 1 then
				variableVals[i - 1] = v
			end
		end
	end
	return variableVals
end

function p.getArgNums(args, prefix, suffix)
	-- Returns a table containing the numbers of the arguments that exist
	-- for the specified prefix and suffix.
	local nums = {}
	local pattern = '^' .. prefix .. '([1-9]%d*)' .. suffix .. '$'
	for k, _ in pairs(args) do
		local num = tostring(k):match(pattern)
		if num then
			nums[#nums + 1] = tonumber(num)
		end
	end
	table.sort(nums)
	return nums
end

function p.callTemplate(template, targs)
	return mw.getCurrentFrame():expandTemplate{title = template, args = targs}
end

return p