Editing Module:Sandbox/Jts1882/Covid

require('strict')
local p = {}
local Date = require('Module:Date')._Date
local mm = require('Module:Math')

local list = require('Module:Sandbox/Jts1882/Covid/data')
local list2 = {
				{ "Austria", "Q86847911", "Q40"},
				{ "Belgium", "Q84446340", "Q31",}, 
                { "Bahrain", "Q87070999", "Q398" },
                { "Bangladesh", "Q87540454", "Q902" },
                { "Czech Republic", "Q86919406", "Q213" },
			--	{ "Poland", "Q87250695", "Q36" },
			    { "Spain", "Q84166704", "Q29" },

}

p.main = function(frame)
	local output = "{|\n"
	local getData = false
	local output = '{| class="wikitable sortable" \n'
                .. "! Country !! Cases !! Deaths !! Population !! Cases/million pop !! Deaths/million pop "
	for k,v in pairs(list) do
		-- code for getting population data in blocks
		--if v[1] == "Afghanistan" then getData = true end -- start from country X
--		if v[1] == "Germany" then break end           -- end with country Y
		getData = true
		if getData then
			local qid = v[2] -- "Q84446340" -- Belgium
	    	
	    	--output = output .. v[1] .. ": " .. p.getWikidata(v) .. "<br/>"
	    	output = output   .. p.getWikidata(v) 
	    	
	    	-- line to get population from wikidata in blocks
	    	--output = output .. '{ "' .. v[1] .. '", "' .. v[2] .. '", "' .. v[3] .. '", ' .. p.getWikidataPopulation(v[3]).. ' },<br/>'
    	end
	end
		output = output .. "\n|}"
	return output	
end
p.getWikidata =function(v)
	
	local countryName = v[1]
	local covidQid    = v[2]
	local countryQid  = v[3]
	local population  = v[4]
	
    local value = "EMPTY"
    local item

	local wd = {}
	local covidCases
	local covidDeaths
	local dateCases
	local dateDeaths

if 1==1 then
	if covidQid and mw.wikibase.isValidEntityId( covidQid ) then -- valid id
    	
    	item = mw.wikibase.getEntity(covidQid)
        if not item then return "Covid Qid not found" end 

 
    	local i= 1
    	-- get cases 
    	while item:getBestStatements('P1603')[i]  do
    		local timeStamp 
    	    local statements = item:getBestStatements('P1603')[i]
    	    covidCases = statements.mainsnak.datavalue.value.amount 
    	    if statements.qualifiers and statements.qualifiers['P585'] and statements.qualifiers['P585'][1]['datavalue'] then
	   	    	timeStamp = statements.qualifiers['P585'][1]['datavalue']['value']['time'] --point in time
    	    end
    	    --output = covidCases
    	    
    	    dateCases = "missing"
    	    if timeStamp  then 
    	    	dateCases = Date(timeStamp):text("dmy") or "missing"
    	    end	
    	    i=i+1
    	end
        --output =  (covidCases or "n/a") ..' (' .. (dateCases or "missing date") .. ')' 
    	
    	-- get deaths
    	i=1
    	while item:getBestStatements('P1120')[i]  do
    		local timeStamp 
    	    local statements = item:getBestStatements('P1120')[i]
    	    covidDeaths = statements.mainsnak.datavalue.value.amount 
    	    if statements.qualifiers and statements.qualifiers['P585'] and statements.qualifiers['P585'][1]['datavalue'] then
	   	    	timeStamp = statements.qualifiers['P585'][1]['datavalue']['value']['time'] --point in time
    	    end
    	    
    	    
    	    if timeStamp  then 
    	    	dateDeaths = Date(timeStamp):text("dmy") or "missing"
    	    end
    	    i=i+1
    	end
     	--output = output .. " | " .. (covidDeaths or "n/a") .. ' (' .. (dateCases or "missing date") .. ')'
	end
end	
if 1==2 then 
    local statements -- don't want a lot of large local objects
    if countryQid and mw.wikibase.isValidEntityId( countryQid ) then -- valid id

    	--item = mw.wikibase.getEntity(countryQid)
        --if not item then return "Country Qid not found" end 

    	-- get population 
    	local i=1
    	statements = nil
     	statements  = mw.wikibase.getBestStatements( countryQid, 'P1082' )
    	i = 1 --#statements 
    	
    	while statements[i]  do
    	--while item:getBestStatements('P1082')[i]  do
    		local timeStamp 
    	    local statement = statements[i] --item:getBestStatements('P1082')[i]
    	    population = statement.mainsnak.datavalue.value.amount
    	    if statement.qualifiers and statement.qualifiers['P585'] and statement.qualifiers['P585'][1]['datavalue'] then
	   	    	timeStamp = statement.qualifiers['P585'][1]['datavalue']['value']['time'] --point in time
    	    end

    	    datePopulation = "missing"
    	    if timeStamp  then 
    	    	datePopulation = Date(timeStamp):text("dmy") or "missing"
    	    end	
    	    i=i+1
    	end
    end    
end        
        
    --[[  
    local output = "Cases: "      .. (covidCases or "n/a")  .. ' (' .. (dateCases or "missing date") .. ')' 
              .. ". Deaths: "     .. (covidDeaths or "n/a") .. ' (' .. (dateDeaths or "missing date") .. ')'
              .. ". Population: " .. (population or "n/a")  -- .. ' (' .. (datePopulation or "missing date") .. ')' 
    ]]
    
    
    local output = '\n|-\n|' .. countryName 
                .. '||' .. (covidCases or "n/a")  .. ' (' .. (dateCases or "missing date") .. ')'
                .. '||' .. (covidDeaths or "n/a") .. ' (' .. (dateDeaths or "missing date") .. ')'
                .. '||' .. (population or "n/a")

    if covidCases == nil then covidCases = 0.001 end
    if covidDeaths == nil then covidDeaths = 0.001 end
    
    local output = output
                .. '||' .. mm._precision_format(covidCases/population*1000000 ,0)  
                .. '||' .. mm._precision_format(covidDeaths/population*1000000 ,0) 
    




    return output or "no value found"
        
end
p.getWikidataPopulation =function(countryQid)

	local population
	local datePopulation
    local statements -- don't want a lot of large local objects
    if countryQid and mw.wikibase.isValidEntityId( countryQid ) then -- valid id

    	--item = mw.wikibase.getEntity(countryQid)
        --if not item then return "Country Qid not found" end 

    	-- get population 
    	local i=1
    	statements = nil
     	statements  = mw.wikibase.getBestStatements( countryQid, 'P1082' )
    	i = 1 --#statements 
    	
    	while statements[i]  do
    	--while item:getBestStatements('P1082')[i]  do
    		local timeStamp 
    	    local statement = statements[i] --item:getBestStatements('P1082')[i]
    	    population = statement.mainsnak.datavalue.value.amount
    	    if statement.qualifiers and statement.qualifiers['P585'] and statement.qualifiers['P585'][1]['datavalue'] then
	   	    	timeStamp = statement.qualifiers['P585'][1]['datavalue']['value']['time'] --point in time
    	    end

    	    datePopulation = "missing"
    	    if timeStamp  then 
    	    	datePopulation = Date(timeStamp):text("dmy") or "missing"
    	    end	
    	    i=i+1
    	end
    end    
    return (population or "n/a")  

end

return p