Difference between revisions of "Module:Yesno"

From The HILLSIDE
Jump to navigation Jump to search
en>Mr. Stradivarius
(removing code that's no longer used)
en>Mr. Stradivarius
(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 or "¬", then deal with other cases.
+
     -- First deal with the case if val is nil, then deal with other cases.
     if val == nil or val == '¬' then
+
     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 == '0' then
+
     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 == '1' or retvals['def'] == nil then
+
     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 16: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