Modul:Wikidata/Filterers
Skočit na navigaci
Skočit na vyhledávání
Věci přenesené přes šablony z cs:wiki
require "Modul:No globals"
local p = {}
local lib = require 'Modul:Wikidata/lib'
local function checkLimit(sequence, limit)
local limit = limit and tonumber(limit)
if limit then
return #sequence >= limit
end
return true
end
local function applyLimit(sequence, limit)
local limit = limit and tonumber(limit)
if limit then
while #sequence > limit do
table.remove(sequence)
end
end
return sequence
end
local function IsInLanguage(snak, lang)
if snak.datatype ~= 'monolingualtext' then
return error(lib.raiseInvalidDatatype('IsInLanguage', snak.datatype, 'monolingualtext'))
end
if lib.IsSnakValue(snak) then
if snak.datavalue.value.language == lang then
return true
end
end
return false
end
function p.filterStatementsFromEntity(entity, options)
if not options.property or options.property == '' then
return error(lib.formatError('param-not-provided', 'property'))
end
if not entity or not entity.claims then
return {}
end
local property = mw.ustring.upper(options.property)
if not entity.claims[property] then
return {}
end
return p.filterStatements(entity.claims[property], options)
end
function p.filterStatements(statements, options)
local options = lib.common.cleanArgs(options)
local Statements, oldStatements = statements, {}
-- apply filter by rank
if not options.rank or options.rank ~= "all" then
oldStatements, Statements = Statements, {}
if not options.rank or options.rank == "best" or options.rank == "valid" then
if options.rank == "best" then
for _, statement in pairs(oldStatements) do
if statement.rank == "preferred" then
table.insert(Statements, statement)
end
end
if #Statements == 0 then
for _, statement in pairs(oldStatements) do
if statement.rank == "normal" then
table.insert(Statements, statement)
end
end
end
else
for _, statement in pairs(oldStatements) do
if statement.rank ~= "deprecated" then
table.insert(Statements, statement)
end
end
end
else
for _, statement in pairs(oldStatements) do
if statement.rank == options.rank then
table.insert(Statements, statement)
end
end
end
if #Statements == 0 then return {} end
end
-- apply filter by source
if options.ref then
oldStatements, Statements = Statements, {}
for _, statement in pairs(oldStatements) do
if statement.references then
if #p.filterReferences(statement.references, options) > 0 then
table.insert(Statements, statement)
end
end
end
if #Statements == 0 then return {} end
end
-- apply filter by snak type
if not lib.IsOptionTrue(options, 'showspecial') then
oldStatements, Statements = Statements, {}
for _, statement in pairs(oldStatements) do
if lib.IsSnakValue(statement.mainsnak) then
table.insert(Statements, statement)
end
end
if #Statements == 0 then return {} end
end
-- apply filter by qualifier property
if options.withqualifier then
oldStatements, Statements = Statements, {}
for _, statement in pairs(oldStatements) do
if statement.qualifiers and statement.qualifiers[options.withqualifier:upper()] then
table.insert(Statements, statement)
end
end
if #Statements == 0 then return {} end
end
-- apply filter by class
if options.instance then
local datatype = Statements[math.random(1, #Statements)].mainsnak.datatype
if datatype == 'wikibase-item' or datatype == 'wikibase-property' then
local Module = require 'Modul:Wikidata/Tree'
oldStatements, Statements = Statements, {}
for _, statement in pairs(oldStatements) do
if lib.IsSnakValue(statement.mainsnak) then
local item = lib.getEntityIdFromValue(statement.mainsnak.datavalue.value)
if Module.IsInstance(item, options) then
table.insert(Statement, statement)
end
end
end
if #Statements == 0 then return {} end
else
return error(lib.raiseInvalidDatatype('inClass', datatype, {'wikibase-item', 'wikibase-property'}))
end
end
-- apply filter by language
if options.withlang then
oldStatements, Statements = Statements, {}
for _, statement in pairs(oldStatements) do
if IsInLanguage(statement.mainsnak, options.withlang) then
table.insert(Statements, statement)
end
end
if #Statements == 0 then return {} end
end
-- apply filter by time
if options.date then
local date
local Time = require 'Modul:Time'
if type(options.date) == 'table' then
date = options.date
elseif options.date == '#now' then
date = Time.new(os.date('!*t'))
--elseif mw.ustring.find(options.date, '^[Pp][1-9]%d-$') then TODO
else
local Time = require 'Modul:Time'
date = Time.newFromIso8601(options.date)
end
if date then
oldStatements, Statements = Statements, {}
local temp_value
local Formatters = require 'Modul:Wikidata/Formatters'
local Date = require 'Modul:Wikidata/datum'
for _, statement in pairs(oldStatements) do
if statement.qualifiers then
local Values = {}
for key, array in pairs(lib.props) do
for _, prop in pairs(array) do
if statement.qualifiers[prop] then
for _, snak in pairs(statement.qualifiers[prop]) do
if lib.IsSnakValue(snak) then
Values[key] = Formatters.getRawValue(snak)
break
end
end
end
end
end
if Values.point then
if not Date.IsSecondLaterThanFirst(date, Values.point) then
if not temp_value then
Statements = { statement }
temp_value = Values.point
else
if Date.IsSecondLaterThanFirst(Values.point, temp_value) then
Statements = { statement }
temp_value = Values.point
elseif Date.AreDatesSame(temp_value, Values.point) then
table.insert(Statements, statement)
end
end
end
else
if Values.begin then
if Date.IsSecondLaterThanFirst(Values.begin, date) then
if not Values.ending then
table.insert(Statements, statement)
elseif Date.IsSecondLaterThanFirst(date, Values.ending) then
table.insert(Statements, statement)
end
end
elseif Values.ending then
if Date.IsSecondLaterThanFirst(date, Values.ending) then
if not Values.begin then
table.insert(Statements, statement)
elseif Date.IsSecondLaterThanFirst(Values.begin, date) then
table.insert(Statements, statement)
end
end
end
end
end
end
if #Statements == 0 then return {} end
else
return error(lib.formatError('invalid-date', options.date))
end
end
if lib.IsOptionTrue(options, 'withlabel') then
local datatype = Statements[math.random(1, #Statements)].mainsnak.datatype
if datatype == 'wikibase-item' or datatype == 'wikibase-property' then
oldStatements, Statements = Statements, {}
local Formatters = require 'Modul:Wikidata/Formatters'
for _, statement in pairs() do
if lib.IsSnakValue(statement.mainsnak) then
local value = Formatters.getRawValue(statement.mainsnak)
if mw.wikibase.label(value) then
table.insert(Statements, statement)
end
end
end
else
return error(lib.raiseInvalidDatatype('withlabel', datatype, {'wikibase-item', 'wikibase-property'}))
end
end
-- sort statements if needed
if options.sort then
local Sorters = require "Modul:Wikidata/Sorters"
Statements = Sorters.sortStatements(Statements, options)
end
-- apply filter by limit
Statements = applyLimit(Statements, options.limit)
return Statements
end
function p.filterQualifiers(qualifiers, options)
local options = lib.common.cleanArgs(options)
local Qualifiers, oldQualifiers = qualifiers, {}
if options['qualifiers withlang'] then
oldQualifiers, Qualifiers = Qualifiers, {}
for _, snak in pairs(oldQualifiers) do
if IsInLanguage(snak, options['qualifiers withlang']) then
table.insert(Qualifiers, snak)
end
end
end
if options['class qualifiers'] then
-- TODO: Modul:Wikidata/Tree
end
if options['sort qualifiers'] then
local Sorters = require "Modul:Wikidata/Sorters"
Qualifiers = Sorters.sortQualifiers(Qualifiers, options)
end
Qualifiers = applyLimit(Qualifiers, options['limit qualifiers'])
return Qualifiers
end
function p.filterReferences(references, options)
local options = lib.common.cleanArgs(options)
if options.ref == '#any' then
return references
end
local oldReferences, References = references, {}
if options.ref == 'valid' then
local props = {'P248', 'P854'} --(require 'Modul:Wikidata/cite').props
for _, ref in pairs(oldReferences) do
for _, prop in pairs(props) do
if ref.snaks[prop] then
table.insert(References, ref)
end
end
end
end
if options.min_ref and not checkLimit(References, options.min_ref) then
return {}
end
return References
end
function p.test(id)
local entity = mw.wikibase.getEntity(id)
for prop, claim in pairs(entity.claims) do
for i, statement in pairs(claim) do
if statement.references then
mw.log(prop, i)
mw.logObject(statement.mainsnak)
mw.logObject(statement.references)
return
end
end
end
end
return p