Module:Yesno: Difference between revisions
From Vigyanwiki
Template>Mr. Stradivarius (tweak logic to make it work like the actual template - the documentation was slightly wrong) |
Template>Mr. Stradivarius (fix logic for the case where the first positional parameter is "¬") |
||
Line 30: | Line 30: | ||
val = args[1] | val = args[1] | ||
-- First deal with the case if val is nil, then deal with other cases. | -- First deal with the case if val is nil or "¬", then deal with other cases. | ||
if val == nil then | if val == nil or val == '¬' then | ||
return retvals['¬'] | return retvals['¬'] | ||
end | end |
Revision as of 15:51, 23 March 2013
Documentation for this module may be created at Module:Yesno/doc
local p = {}
function p.yesno(frame)
-- defaults
local retvals = {
yes = "yes",
no = "",
["¬"] = ""
}
-- Allow arguments to override defaults.
-- 'any' tracks the presence of any arguments at all.
local args = frame.args
local any = false
for k,v in pairs(args) do
any = true
retvals[k] = v
end
-- If there are no arguments, try and get them from the parent frame.
if any == false then
local pframe = frame:getParent()
args = pframe.args
for k,v in pairs(args) do
any = true
retvals[k] = v
end
end
val = args[1]
-- First deal with the case if val is nil or "¬", then deal with other cases.
if val == nil or val == '¬' then
return retvals['¬']
end
val = val:lower() -- Coerce to blank if nil; make lowercase.
val = val:match'^%s*(.*%S)' or '' -- Trim whitespace.
if val == '' then
return retvals['blank'] or retvals['no']
elseif val == 'n' or val == 'no' or val == '0' then
return retvals['no']
elseif val == 'y' or val == 'yes' or val == '1' or retvals['def'] == nil then
return retvals['yes']
else
return retvals['def']
end
end
return p