Module:Medical cases chart/data: Difference between revisions

Content deleted Content added
mNo edit summary
Fixed off-by-one error
 
(9 intermediate revisions by 4 users not shown)
Line 1:
--- Example usage:
--- {{#invoke:Module:Medical cases chart/data|externalData|page name=COVID-19 Casescases in Santa Clara County, California.tab|recoveries=hospitalized|cases=totalConfirmedCases}}
--- =p.externalData_externalData(mw.getCurrentFrame():newChild{args={["page name"]datapage="COVID-19 Casescases in Santa Clara County, California.tab",recoveriesdatarecoveries="hospitalized",casesdatacases="totalConfirmedCases"}})
 
local p = {}
Line 7:
local english = mw.getLanguage("en")
 
local function round(x)
return (math.modf(x + (x < 0 and -0.5 or 0.5)))
end
 
local function formatChange(previous, current)
if not previous or previous == 0 then
return
Line 21:
local change = current / previous * 100 - 100
local sign = change < 0 and "−" or "+"
return mw.ustring.format("%s%s%%", sign, lang:formatNum(round(math.abs(change))))
if (math.abs(change)) >= 10 then
return mw.ustring.format("%s%.0f%%", sign, (math.abs(change)))
end
if (math.abs(change)) < 1 then
return mw.ustring.format("%s%.3f%%", sign, (math.abs(change)))
end
return mw.ustring.format("%s%s.2f%%", sign, lang:formatNum(round(math.abs(change))))
end
 
function p.externalData_externalData(frameargs)
local data = mw.ext.data.get(frame.args["page name"].datapage)
local dateIndex
Line 34 ⟶ 41:
local class5Index
for i, field in ipairs(data.schema.fields) do
if field.name == "date" or field.name == frame.args.datedatadate then
dateIndex = i
elseif field.name == "deaths" or field.name == frame.args.deathsdatadeaths then
deathsIndex = i
elseif field.name == "recoveries" or field.name == frame.args.recoveriesdatarecoveries then
recoveriesIndex = i
elseif field.name == "cases" or field.name == frame.args.casesdatacases then
casesIndex = i
elseif field.name == "class4" or field.name == frame.args.class4dataclass4 then
class4Index = i
elseif field.name == "class5" or field.name == frame.args.class5dataclass5 then
class5Index = i
end
end
assert(dateIndex, "Date field not found.")
assert(deathsIndex or not frame.args.deathsdatadeaths, "Deaths field not found.")
assert(recoveriesIndex or not frame.args.recoveriesdatarecoveries, "Recoveries field not found.")
assert(casesIndex or not frame.args.casesdatacases, "Cases field not found.")
assert(class4Index or not frame.args.class4dataclass4, "Class 4 field not found.")
assert(class5Index or not frame.args.class5dataclass5, "Class 5 field not found.")
-- Restructure the data as tables with keys.
Line 69 ⟶ 76:
}
local prevRecord = records[#records] or {}
if casesIndex and not prevRecord.cases and record.cases and record.cases > 0 then
record.options.firstright1 = "y"
end
if deathsIndex and prevRecord.deaths == 0 and record.deaths and record.deaths > 0 then
record.options.firstright2 = "y"
end
if deathsIndex and (prevRecord.deaths or prevRecord.assumedDeaths) and not record.deaths then
record.assumedDeaths = prevRecord.deaths or prevRecord.assumedDeaths
end
if casesIndex and (prevRecord.cases or prevRecord.assumedCases) and not record.cases then
record.assumedCases = prevRecord.cases or prevRecord.assumedCases
end
if record.deaths == prevRecord.deaths
Line 91 ⟶ 101:
for i = #records, 1, -1 do
local record = records[i]
if i < #records and record.streak > 3 then
for j = i, i - record.streak + 3, -1 do
table.remove(records, j)
Line 111 ⟶ 121:
tostring(record.deaths or record.assumedDeaths or ""),
tostring(record.recoveries or ""),
tostring(record.cases or record.assumedCases or ""),
tostring(record.class4 or ""),
tostring(record.class5 or ""),
record.cases and lang:formatNum(record.cases) or "",
record.cases and formatChange(prevRecord.cases or prevRecord.assumedCases, record.cases) or "",
record.deaths and lang:formatNum(record.deaths) or "",
record.deaths and formatChange(prevRecord.deaths or prevRecord.assumedDeaths, record.deaths) or "",
Line 127 ⟶ 137:
end
 
function p.externalData(frame)
local args = {}
for k,v in pairs(frame.args) do
if (v or '') ~= '' then
args['data'..k] = v
end
end
return p._externalData(args)
end
return p