Difference between revisions of "Module:Yesno"
Jump to navigation
Jump to search
what>Mr. Stradivarius (better code for getting args) |
what>Mr. Stradivarius (edit inaccurate comment) |
||
Line 10: | Line 10: | ||
} | } | ||
− | -- Allow arguments to override defaults | + | -- Allow arguments to override defaults. |
− | |||
local args; | local args; | ||
if frame == mw.getCurrentFrame() then | if frame == mw.getCurrentFrame() then |
Revision as of 10:26, 28 March 2013
local p = {}
function p.yesno(frame)
-- defaults local retvals = { yes = "yes", no = "", ["¬"] = "" }
-- Allow arguments to override defaults. local args; if frame == mw.getCurrentFrame() then -- We're being called via #invoke. If the invoking template passed any args, use -- them. Otherwise, use the args that were passed into the template. args = frame:getParent().args; for k, v in pairs(frame.args) do args = frame.args; break end else -- We're being called from another module or from the debug console, so assume -- the args are passed in directly. args = frame; end for k,v in pairs(args) do retvals[k] = v 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