Module:License Wikidata/sandbox

require('strict')

local p = {} --p stands for package

local getArgs = require('Module:Arguments').getArgs local tableToolsModule = require("Module:TableTools") local dateModule = require("Module:Era")

local property_ids = { ['deathyear'] = 'P570', ['pubyear'] = 'P577', ['edition'] = 'P629' }

local function clean_table(t) return tableToolsModule.removeDuplicates(tableToolsModule.compressSparseArray(t)) end

--[=[	Death/publication year ]=]

local function getYearFromSingleStatement(statement) local snak = statement.mainsnak -- We're not using mw.wikibase.formatValue because we only want years. if snak.snaktype ~= 'value' then return nil end -- Precision is less than a year. if snak.datavalue.value.precision < 9 then return nil end -- Year is approximate. -- P1480 = sourcing circumstances, Q5727902 = circa. if statement.qualifiers ~= nil and statement.qualifiers.P1480 ~= nil then for _, qualifier in pairs(statement.qualifiers.P1480) do			if qualifier.datavalue ~= nil and qualifier.datavalue.value.id == 'Q5727902' then return nil end end end -- Extract year from the time value. local _, _, extractedYear = string.find(snak.datavalue.value.time, '([%+%-]%d%d%d+)%-') local year = tonumber(extractedYear) -- Handle BCE years. -1 BCE = 0. if year < 0 then year = year + 1 end return year end

local function getItemFromWikidataID(wikidata_id) if wikidata_id ~= nil and wikidata_id ~= '' then return mw.wikibase.getEntity(wikidata_id) or nil else return mw.wikibase.getEntity or nil end end

local function getYearsFromWikidata(args) local item = getItemFromWikidataID(args.wikidata_id) if not item then return nil end local year_prop = args.year_prop -- Compile a list of year statements. local statements = item:getBestStatements(year_prop) if #statements == 0 then return nil end -- Compile a list of years, one from each statement. local years = {} for _, statement in pairs(statements) do		table.insert(years, getYearFromSingleStatement(statement)) end years = clean_table(years) if #years == 0 then return nil end -- Return a sorted list of years. table.sort(years) return years end

local function sortForBestYear(args) local years = args.years if not years then return nil end if args.reverse_sort then table.sort(years, function(a, b) return a > b end) else table.sort(years) end for _, year in pairs(years) do		if tonumber(year) then return tonumber(year) end end return nil end

local function getBestYearFromWikidata(args) local wikidata_years = getYearsFromWikidata({		['wikidata_id'] = args.wikidata_id,		['year_prop'] = args.year_prop	}) return sortForBestYear({['years'] = wikidata_years, ['reverse_sort'] = args.reverse_sort}) end

local function getAuthorDeathYearFromWikidata(wikidata_id) return getBestYearFromWikidata({		['wikidata_id'] = wikidata_id,		['year_prop'] = property_ids['deathyear'],		['reverse_sort'] = true	}) end

local function getPublicationYearFromWikidata(wikidata_id) return getBestYearFromWikidata({		['wikidata_id'] = wikidata_id,		['year_prop'] = property_ids['pubyear'],		['reverse_sort'] = false	}) end

--[=[	Creator death years ]=]

local function getCreatorsFromSingleProp(item, creator_prop) -- Compile a list of creator statements. local statements = item:getBestStatements(creator_prop) if #statements == 0 then return nil end -- Compile a list of creators, one from each statement. local creators = {} for _, statement in pairs(statements) do		local snak = statement.mainsnak if snak.snaktype == 'value' and snak.datavalue['type'] == 'wikibase-entityid' and snak.datavalue.value['entity-type'] == 'item' then table.insert(creators, snak.datavalue.value.id) end end creators = clean_table(creators) if #creators == 0 then return nil end return creators end

local function getCreators(item, creator_props) local all_creators = {} -- is there a better way to merge tables? for _, creator_prop in pairs(creator_props) do		local creators = getCreatorsFromSingleProp(item, creator_prop) if creators then for _, creator in pairs(creators) do				table.insert(all_creators, creator) end end end all_creators = clean_table(all_creators) if #all_creators == 0 then return nil end return all_creators end

local function getCreatorDeathYearsFromWikidata(args) local item = getItemFromWikidataID(args.wikidata_id) if not item then return nil end --[=[		What I really want is a list of all subproperties of P170 and P767 (unless there are some that shouldn't be included). ]=]	local creator_props = args.creator_props or { ['creator'] = 'P170', ['author'] = 'P50', ['librettist'] = 'P87', ['composer'] = 'P86', ['orchestrator'] = 'P10806', ['adapted by'] = 'P5202', ['screenwriter'] = 'P58', ['lyrics by'] = 'P676', ['performer'] = 'P175', ['illustrator'] = 'P110', ['letterer'] = 'P9191', ['inker'] = 'P10836', ['penciller'] = 'P10837', ['colorist'] = 'P6338', ['contributor'] = 'P767', ['translator'] = 'P655' }	if not creator_props then return nil end local creators = getCreators(item, creator_props) if not creators then return nil end local deathyears = {} for _, creator in pairs(creators) do		local deathyear = getAuthorDeathYearFromWikidata(creator) if deathyear then table.insert(deathyears, deathyear) else return nil -- if creator doesn't have deathyear end end deathyears = clean_table(deathyears) if #deathyears == 0 then return nil end table.sort(deathyears) return deathyears end

local function getBestCreatorDeathYearFromWikidata(args) local wikidata_years = getCreatorDeathYearsFromWikidata({		['wikidata_id'] = args.wikidata_id,		['creator_props'] = args.creator_props	}) return sortForBestYear({['years'] = wikidata_years, ['reverse_sort'] = true}) end

local function getBestYear(args) local manual_years = args.manual_years local wikidata_function = args.wikidata_function local wikidata_function_args = args.wikidata_function_args if manual_years then for _, year in pairs(manual_years) do			if tonumber(year) then return tonumber(year) end end end if wikidata_function then return wikidata_function(wikidata_function_args) end return nil end

local function getAuthorDeathYear(args) return getBestYear({		['manual_years'] = args.deathyears, 		['wikidata_function'] = getAuthorDeathYearFromWikidata,		['wikidata_function_args'] = args.wikidata_id	}) end

local function getWorkCreatorDeathYear(args) return getBestYear({		['manual_years'] = args.deathyears, 		['wikidata_function'] = getBestCreatorDeathYearFromWikidata,		['wikidata_function_args'] = {['wikidata_id'] = args.wikidata_id, ['creator_props'] = args.creator_props}	}) end

local function getPublicationYear(args) --[=[	return getBestYear({		['manual_years'] = args.pubyears, 		['wikidata_function'] = getPublicationYearFromWikidata,		['wikidata_function_args'] = args.wikidata_id	}) ]=]	return getBestYear({['manual_years'] = args.pubyears}) end

local function getWorkCreatorOrAuthorDeathYear(args) local deathyear local namespace = args.namespace or mw.title.getCurrentTitle.nsText if namespace == 'Author' or namespace == 'Author talk' then deathyear = getAuthorDeathYear(args) else deathyear = getWorkCreatorDeathYear(args) end return deathyear end

local function getWorkOrAuthorPublicationYear(args) local pubyear = nil local namespace = args.namespace or mw.title.getCurrentTitle.nsText if namespace ~= 'Author' and namespace ~= 'Author talk' then pubyear = getPublicationYear(args) end return pubyear end

local function getWorksFromEdition(wikidata_id) local item = getItemFromWikidataID(wikidata_id) if not item then return nil end local statements = item:getBestStatements(property_ids['edition']) if #statements == 0 then return nil end local work_ids = {} for i = 1, #statements do		local statement = statements[i] if statement.mainsnak and statement.mainsnak.datatype == 'wikibase-item' and statement.mainsnak.datavalue and statement.mainsnak.datavalue.value and statement.mainsnak.datavalue.value.id then table.insert(work_ids, statement.mainsnak.datavalue.value.id) end end if #work_ids == 0 then return nil end return work_ids end

local function getWorkOrEditionYear(yearFunction, args) args = args or {} local wikidata_id = args.wikidata_id local work_ids = getWorksFromEdition(wikidata_id) or {} for i = 1, #work_ids do		args.wikidata_id = work_ids[i] local work_year = yearFunction(args) if work_year then return work_year end end args.wikidata_id = wikidata_id return yearFunction(args) end

--[=[	Public functions ]=]

function p.getAuthorDeathYear(args) return getWorkOrEditionYear(getAuthorDeathYear, args) end

function p.getPublicationYear(args) return getWorkOrEditionYear(getPublicationYear, args) end

function p.getWorkCreatorDeathYear(args) return getWorkOrEditionYear(getWorkCreatorDeathYear, args) end

function p.getWorkCreatorOrAuthorDeathYear(args) return getWorkOrEditionYear(getWorkCreatorOrAuthorDeathYear, args) end

function p.getWorkOrAuthorPublicationYear(args) return getWorkOrEditionYear(getWorkOrAuthorPublicationYear, args) end

return p