Difference between revisions of "Module:Yesno"

From FIThydrowiki
Jump to navigation Jump to search
what>Atdt
(Add inline comments)
what>Mr. Stradivarius
(get arguments from the parent frame if the current frame doesn't have any)
Line 1: Line 1:
 
local p = {}
 
local p = {}
 
   
 
   
function p.yesno( frame )
+
function p.yesno(frame)
 
   
 
   
 
     -- defaults
 
     -- defaults
Line 12: Line 12:
 
     -- 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.
 +
    local args = frame.args
 
     local any = false
 
     local any = false
     for k,v in pairs( frame.args ) do
+
     for k,v in pairs(args) do
 
         any = true
 
         any = true
 
         retvals[k] = v
 
         retvals[k] = v
 
     end
 
     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 = frame.args[1]
+
     val = args[1]
 
   
 
   
 
     -- According to the template docs, the input should be considered nil
 
     -- According to the template docs, the input should be considered nil

Revision as of 08:35, 23 March 2013

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]

   -- According to the template docs, the input should be considered nil
   -- only when no params were provided. If any params at all were present,
   -- the value must be considered blank. A bit weird, if you ask me.
   if (val == nil and not any) or val == '¬' then
       return retvals['¬']
   end

   val = (val or ):lower()          -- Coerce to blank if nil; make lowercase.
   val = val:match'^%s*(.*%S)' or   -- Trim whitespace.

   if val ==  then
       return retvals['blank'] ~= nil and 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