Difference between revisions of "Yesno"
From Unofficial Stationeers Wiki
(removing code that's no longer used) |
(fix bugs highlighted in Module talk:Yesno/testcases) |
||
| Line 29: | Line 29: | ||
val = args[1] | val = args[1] | ||
| − | -- First deal with the case if val is nil | + | -- First deal with the case if val is nil, then deal with other cases. |
| − | if val == nil | + | if val == nil then |
return retvals['¬'] | return retvals['¬'] | ||
end | end | ||
| Line 39: | Line 39: | ||
if val == '' then | if val == '' then | ||
return retvals['blank'] or retvals['no'] | return retvals['blank'] or retvals['no'] | ||
| − | elseif val == 'n' or val == 'no' or val == | + | elseif val == 'n' or val == 'no' or tonumber(val) == 0 then |
return retvals['no'] | return retvals['no'] | ||
| − | elseif val == 'y' or val == 'yes' or val == | + | elseif val == 'y' or val == 'yes' or tonumber(val) == 1 then |
return retvals['yes'] | return retvals['yes'] | ||
| + | elseif val == '¬' then | ||
| + | return retvals['¬'] | ||
else | else | ||
| − | return retvals['def'] | + | return retvals['def'] or retvals['yes'] |
end | end | ||
end | end | ||
return p | return p | ||
Revision as of 08:12, 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
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() -- 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 tonumber(val) == 0 then
return retvals['no']
elseif val == 'y' or val == 'yes' or tonumber(val) == 1 then
return retvals['yes']
elseif val == '¬' then
return retvals['¬']
else
return retvals['def'] or retvals['yes']
end
end
return p