查看“Module:TodayBirthday”的源代码

local p = {}

local array = require("Module:Var-array") --调用array

local config = {
	name = "角色",
    recent = 0,
}

local data = {}

--接近生日排序函数
function birth_comp(a,b)
    return a[2]<b[2]
end

--frame.args[1]：数据模块名称
--frame.args[2]：角色称呼
--frame.args[3]：最近生日天数
--frame.args[4]：指定当前日期

function p.main( frame )

    --变量定义
    local lang = mw.language.new("zh")  --语言库
    local today = {}      --今天过生日
    local recent = {}     --最近过生日
    local recent_aout = {}
    local today_out = ""
    local recent_out = ""
    local year = os.date("%Y")  --年份
    local now ="" -- 当前时间
    local count_today = 0  --今天过生日人数
    local count_recent = 0 --最近过生日人数

    -- 处理参数和默认值
    if frame.args[1] then
        local data_module = require( "Module:TodayBirthday/data/" .. frame.args[1] )
        data = data_module.data
    else
        return "<span class=\"scribunto-error\">请指定数据模块名！</span>"
    end
    if frame.args[2] then
        config.name = frame.args[2]
    end
    if tonumber(frame.args[3]) then
        config.recent = tonumber(frame.args[3])
    end
    if frame.args[4] then
        now = lang:formatDate("U",frame.args[4],true)
        year = lang:formatDate("Y",frame.args[4],true)
    else
        now = lang:formatDate("U","",true)
    end

    --计算是否达到/接近生日
    for k,v in ipairs(data) do
        local birth = lang:formatDate("U", year .. "-" .. v[3] .. "-" .. v[4] .. "-8hour", true)
        local birth_next = lang:formatDate("U", year .. "-" .. v[3] .. "-" .. v[4] .. "-8hour+1 year", true)
        local link = ""
        if v[2] == "" then
            link="[[" .. v[1] .."]]"
        else
            link="[[" ..v[2] .. "|" .. v[1] .. "]]"
        end
        if (now-birth>=0 and now-birth<86400) or (now-birth_next>=0 and now-birth_next<86400) then
        	table.insert(today,link)
            count_today = count_today + 1
        end
        if config.recent > 0 then
            if birth-now>0 and birth-now<=86400*config.recent then
                table.insert(recent,{link,birth-now,v[3],v[4]})
                table.insert(recent_aout,link)
            elseif birth_next-now>0 and birth_next-now<=86400*config.recent then
                table.insert(recent,{link,birth_next-now,v[3],v[4]})
                table.insert(recent_aout,link)
            end
        end
    end
    
    --生成今天生日字符串
    for k,v in ipairs(today) do
    	if today_out == "" then
                today_out = v
            else
                today_out = today_out .. "、" ..v
    	end
    end

    --处理接近生日顺序
    if config.recent > 0 then
        table.sort(recent,birth_comp)
        for k,v in ipairs(recent) do
            if recent_out == "" then
                recent_out =  v[1] .. "<small>（" .. v[3] .. "月" .. v[4] .. "日）</small>"
            else
                recent_out = recent_out .. "、" ..v[1] .. "<small>（" .. v[3] .. "月" .. v[4] .. "日）</small>"
            end
            count_recent = count_recent + 1
        end
    end

    --通过变量扩展输出信息
    frame:callParserFunction("#vardefine:tbirth_today_raw",today_out)
    frame:callParserFunction("#vardefine:tbirth_recent_raw",recent_out)
    frame:callParserFunction("#vardefine:tbirth_count",count_recent+count_today)
    frame:callParserFunction("#vardefine:tbirth_count_today",count_today)
    frame:callParserFunction("#vardefine:tbirth_count_recent",count_recent)
    
    --通过array输出信息
    array.new("tbirth_today",today)
    array.new("tbirth_recent",recent_aout)

    --处理输出
    local out =""
    if count_today == 0 then
        out = ""
    else
        out = "祝" .. today_out .. "生日快乐！"
    end
    if config.recent > 0 then
        if recent_out == "" then
            out = out .. config.recent .. "天内没有" .. config.name .. "过生日。"
        else
            out = out .. config.recent .. "天内过生日的" .. config.name .. "有：" .. recent_out .. "。"
        end
    else
        out = out .. "。"
    end
    return out
end


return p
说明	输入内容	输出结果
斜体	''斜体文字''	斜体文字
粗体	'''粗体文字'''	粗体文字
粗斜体	'''''粗斜体文字'''''	*粗斜体文字*