Module:Yesno: Difference between revisions

Jump to navigation Jump to search
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]
 
     -- According to the template docs, the input should be considered nil
     -- First deal with the case if val is nil, then deal with other cases.
    -- only when no params were provided. If any params at all were present,
     if val == nil then
    -- the value must be considered blank. A bit weird, if you ask me.
     if (val == nil and not any) or val == '¬' then
         return retvals['¬']
         return retvals['¬']
     end
     end
 
     val = (val or ''):lower()          -- Coerce to blank if nil; make lowercase.
     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 retvals['blank'] ~= nil and retvals['blank'] or retvals['no']
         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 10: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