Difference between revisions of "Module:Yesno"

From FIThydrowiki
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. Arguments are taken from
+
     -- Allow arguments to override defaults.
    -- the parent frame; other arguments are ignored.
 
 
     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