Module:Medical cases chart/data: Difference between revisions

Revision as of 09:11, 21 April 2020

Latest revision as of 05:04, 1 September 2021

(18 intermediate revisions by 4 users not shown)

Line 1:

--- Example usage:

--- =p.externalData(mw.getCurrentFrame():newChild{args={["#invoke:Medical cases chart/data|externalData|page name"]="COVID-19 Casescases in Santa Clara County, California.tab",|recoveries="hospitalized",|cases="totalConfirmedCases"}})

--- =p._externalData({datapage="COVID-19 cases in Santa Clara County, California.tab",datarecoveries="hospitalized",datacases="totalConfirmedCases"})

local p = {}

local lang = mw.getContentLanguage()

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 19 ⟶ 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%.2f%%", sign, (math.abs(change)))

end

function p.externalData_externalData(frameargs)

local data = mw.ext.data.get(frame.args["page name"].datapage)

local dateIndex

Line 32 ⟶ 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.

local rows = {}

local prevCasesrecords = {}

for i, row in ipairs(data.data) do

local prevDeaths

for i, local record in ipairs(data.data)= do{

date = row[dateIndex],

local row = {

deaths = deathsIndex and row[deathsIndex],

tostring(record[dateIndex]),

recoveries = recoveriesIndex and row[recoveriesIndex],

deathsIndex and tostring(record[deathsIndex]) or "",

cases = casesIndex and row[casesIndex],

recoveriesIndex and tostring(record[recoveriesIndex]) or "",

class4 = class4Index and row[class4Index],

casesIndex and tostring(record[casesIndex]) or "",

class5 = class5Index and row[class5Index],

class4Index and tostring(record[class4Index]) or "",

options = {},

class5Index and tostring(record[class5Index]) or "",

streak = 1,

casesIndex and lang:formatNum(record[casesIndex]) or "",

casesIndex and formatChange(prevCases, record[casesIndex]) or "",

deathsIndex and lang:formatNum(record[deathsIndex]) or "",

deathsIndex and formatChange(prevDeaths, record[deathsIndex]) or "",

}

local prevRecord = records[#records] or {}

if casesIndex and not prevCases and record[casesIndex] > 0 then

if casesIndex and not prevRecord.cases and record.cases and record.cases > 0 then

table.insert(row, "firstright1=y")

record.options.firstright1 = "y"

end

if deathsIndex and prevDeathsprevRecord.deaths == 0 and record[deathsIndex].deaths and record.deaths > 0 then

tablerecord.options.insert(row, "firstright2 = "y")

end

if deathsIndex and (prevRecord.deaths or prevRecord.assumedDeaths) and not record.deaths then

record.assumedDeaths = prevRecord.deaths or prevRecord.assumedDeaths

table.insert(rows, table.concat(row, ";"))

if casesIndex then

prevCases = record[casesIndex]

end

if casesIndex and (prevRecord.cases or prevRecord.assumedCases) and not record.cases then

if deathsIndex then

record.assumedCases = prevRecord.cases or prevRecord.assumedCases

prevDeaths = record[deathsIndex]

end

if record.deaths == prevRecord.deaths

and record.recoveries == prevRecord.recoveries

and record.cases == prevRecord.cases

and record.class4 == prevRecord.class4

and record.class5 == prevRecord.class5 then

record.streak = prevRecord.streak + 1

end

table.insert(records, record)

end

-- Collapse streaks of identical data.

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)

end

i = i - record.streak + 2

record = records[i]

record.options.collapse = "y"

record.options.id = english:formatDate("M", record.date):lower()

record.date = nil

end

end

-- Stringify the data.

local rows = {}

for i, record in ipairs(records) do

local prevRecord = records[i - 1] or {}

local row = {

record.date or "",

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 "",

}

for k, v in pairs(record.options) do

table.insert(row, string.format("%s=%s", k, v))

end

table.insert(rows, table.concat(row, ";"))

end

return table.concat(rows, "\n")

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