Difference between revisions of "Yesno"
From Unofficial Stationeers Wiki
(get arguments from the parent frame if the current frame doesn't have any) |
(tweak logic to make it work like the actual template - the documentation was slightly wrong) |
||
| Line 1: | Line 1: | ||
local p = {} | local p = {} | ||
| − | + | ||
function p.yesno(frame) | function p.yesno(frame) | ||
| − | + | ||
-- defaults | -- defaults | ||
local retvals = { | local retvals = { | ||
| Line 9: | Line 9: | ||
["¬"] = "" | ["¬"] = "" | ||
} | } | ||
| − | + | ||
-- Allow arguments to override defaults. | -- Allow arguments to override defaults. | ||
-- 'any' tracks the presence of any arguments at all. | -- 'any' tracks the presence of any arguments at all. | ||
| Line 27: | Line 27: | ||
end | end | ||
end | end | ||
| − | + | ||
val = args[1] | val = args[1] | ||
| − | + | ||
| − | -- | + | -- First deal with the case if val is nil, then deal with other cases. |
| − | + | if val == nil then | |
| − | |||
| − | if | ||
return retvals['¬'] | return retvals['¬'] | ||
end | end | ||
| − | + | ||
| − | val = | + | val = val:lower() -- Coerce to blank if nil; make lowercase. |
val = val:match'^%s*(.*%S)' or '' -- Trim whitespace. | val = val:match'^%s*(.*%S)' or '' -- Trim whitespace. | ||
| − | + | ||
if val == '' then | if val == '' then | ||
| − | return | + | return retvals['blank'] or retvals['no'] |
elseif val == 'n' or val == 'no' or val == '0' then | elseif val == 'n' or val == 'no' or val == '0' then | ||
return retvals['no'] | return retvals['no'] | ||
| Line 50: | Line 48: | ||
end | end | ||
end | end | ||
| − | + | ||
return p | return p | ||
Revision as of 05:18, 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, then deal with other cases.
if val == nil 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