Module:Medical cases chart/data: Difference between revisions

Content deleted Content added
Also fill in gaps in cases
Fixed off-by-one error
 
(8 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
Line 94 ⟶ 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 130 ⟶ 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