>Dinoguy1000: simplify month_number (instead of a separate test, `sept` can simply be added as an entry to the months_abbr array)

2022-11-22T10:54:50Z

simplify month_number (instead of a separate test, `sept` can simply be added as an entry to the months_abbr array)

New page

local p = {}

local function month_number(month_name)
local months_full = {january=1, february=2, march=3, april=4, may=5, june=6, july=7, august=8, september=9, october=10, november=11, december=12}
local months_abbr = {jan=1, feb=2, mar=3, apr=4, may=5, jun=6, jul=7, aug=8, sep=9, sept=9, oct=10, nov=11, dec=12}
local month_lc, _ = string.gsub(string.lower(month_name),'%.','',1)
local month_num = months_full[month_lc] or months_abbr[month_lc] or 0
return month_num
end

local function days_in_month(month_num,year)
-- modified from code in Module:Citation/CS1/Date_validation
local days = {31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31}
local month_length
if month_num == 2 then -- February: 28 days, unless leap year
month_length = 28
if year <= 1582 then -- Julian calendar before Oct 1582
if (year%4) == 0 then
month_length = 29 -- if leap year, then 29 days
end
else -- Gregorian calendar since Oct 1582
if ( (year%4)==0 and ((year%100)~=0 or (year%400)==0) ) then
month_length = 29 -- if leap year, then 29 days
end
end
else -- not February, get number of days for month
month_length = days[month_num]
end
return month_length
end

local function zero_pad(string)
if string.len(string) == 1 then
return '0' .. string
else
return string
end
end

function p.main(frame)
if frame.args[1] == nil then
return '' -- first argument is missing
end
local arg1, _ = string.gsub(mw.text.trim(frame.args[1]),'_',' ')
return p._main(arg1)
end
function p._main(arg1)
if arg1 == '' then
return '' -- first argument is empty
end
if not arg1:match('^%d%d%d%d %a%a%a%a?%.?%a?%a?%a?%a?%a?%a? *%d%d?$') then
return arg1 -- invalid date pattern
end
local year, month_name, day = string.match(arg1, '^(%d%d%d%d) *(%a%a%a%a?%.?%a?%a?%a?%a?%a?%a?) *(%d%d?)$')
if month_number(month_name) == 0 then
return arg1 -- invalid month name or abbreviation
end
if tonumber(day) < 1 or tonumber(day) > days_in_month(month_number(month_name),tonumber(year)) then
return arg1 -- invalid day number for given month
end
return year .. '-' .. zero_pad(tostring(month_number(month_name))) .. '-' .. zero_pad(day)
end

return p

Module:YMD to ISO - Revision history

>Dinoguy1000: simplify month_number (instead of a separate test, `sept` can simply be added as an entry to the months_abbr array)