Modul:Wikidata utilities

Die Dokumentation für dieses Modul kann unter Modul:Wikidata utilities/doc erstellt werden

-- Wikidata convenience utilities

-- documentation
local WikidataUtilities = {
	suite  = 'WikidataUtilities',
	serial = '2020-09-30',
	item   = 65439025
}

-- module variable
local wu = {}

-- table storing property ids used
local catTable = {
	P0 = ''
}

local function isSet( arg )
	return arg and arg ~= ''
end

function wu.getEntity( id )
	local wrongQualifier = false
	local entity = nil
	
	if not isSet( id ) then
		return '', entity, wrongQualifier
	end
	if mw.wikibase.isValidEntityId( id ) then
		-- expensive function call
		-- redirect ids marked false, too
		entity = mw.wikibase.getEntity( id )
	end
	if not entity then
		id = ''
		wrongQualifier = true
	end

	return id, entity, wrongQualifier
end

function wu.getEntityId( id )
	local wrongQualifier = false
	local entity = nil
	
	if not isSet( id ) then
		id = ''
	elseif mw.wikibase.isValidEntityId( id ) and mw.wikibase.entityExists( id ) then
		-- expensive function call
		-- redirect ids marked false, too
		entity = id
	else
		id = ''
		wrongQualifier = true
	end

	return id, entity, wrongQualifier
end

function wu.getLabel( entity, lang )
	if not isSet( entity ) then
		return nil
	end
	if type( entity ) == 'string' then -- entity is id
		return isSet( lang ) and mw.wikibase.getLabelByLang( entity, lang )
			or mw.wikibase.getLabel( entity )
	else -- entity is table
		return isSet( lang ) and entity:getLabel( lang ) or entity:getLabel()
	end
end

function wu.getSitelink( entity, globalSiteId )
	if not isSet( entity ) then
		return nil
	end
	if type( entity ) == 'string' then -- entity is id
		return isSet( globalSiteId ) and mw.wikibase.getSitelink( entity, globalSiteId )
			or mw.wikibase.getSitelink( entity )
	else -- entity is table
		return isSet( globalSiteId ) and entity:getSitelink( globalSiteId )
			or entity:getSitelink( globalSiteId )
	end
end

function wu.getBestStatements( entity, p )
	if type( entity ) == 'string' then
		return mw.wikibase.getBestStatements( entity, p )
	else
		return entity:getBestStatements( p )
	end
end

function wu.getStatements( entity, p, count )
	local ar = {}
	if not ( isSet( entity ) and isSet( p ) ) then
		return ar
	end

	local statements = wu.getBestStatements( entity, p )
	count = math.min( count or #statements, #statements )
	if count <= 0 then
		return ar
	end

	local i = 0
	repeat
		i = i + 1
		if statements[ i ].mainsnak.snaktype == 'value' then
			table.insert( ar, statements[ i ] )
		end
	until i >= #statements or #ar >= count

	return ar
end

function wu.getValue( entity, p )
	local statements = wu.getStatements( entity, p, 1 )
	if #statements > 0 then
		catTable[ p ] = ''
		return statements[ 1 ].mainsnak.datavalue.value
	else
		return ''
	end
end

function wu.getId( entity, p )
	local value = ''
	local statements = wu.getStatements( entity, p, 1 )
	if #statements > 0 then
		value = statements[ 1 ].mainsnak.datavalue.value
		value = value.id or ''
		if value ~= '' then
			catTable[ p ] = ''
		end
	end
	return value
end

function wu.getValues( entity, p, count )
	local statements = wu.getStatements( entity, p, count )
	if #statements > 0 then
		catTable[ p ] = ''
		for i = 1, #statements, 1 do
			statements[ i ] = statements[ i ].mainsnak.datavalue.value
		end
	end
	return statements
end

function wu.getValuesByLang( entity, p, count, lang )
	local ar = {}
	local statements = wu.getStatements( entity, p, count )
	if #statements > 0 then
		local value
		for i = 1, #statements, 1 do
			value = statements[ i ].mainsnak.datavalue.value
			if value.language and lang == value.language then
				table.insert( ar, value.text )
			end
		end
	end
	if #ar > 0 then
		catTable[ p ] = ''
	end
	return ar
end	

-- get values array for monolingual text
function wu.getMonolingualValues( entity, p )
	local result = {}
	local statements = wu.getStatements( entity, p, nil )
	if #statements > 0 and statements[ 1 ].mainsnak.datatype == 'monolingualtext' then
		local hyphen, lng, value
		catTable[ p ] = ''
		for i = 1, #statements, 1 do
			value = statements[ i ].mainsnak.datavalue.value
			lng = value.language
			hyphen = lng:find( '-' )
			if hyphen then
				lng = lng:sub( 1, hyphen - 1 )
			end
			if not result[ lng ] then
				result[ lng ] = value.text
			end
		end
	end
	return result
end

function wu.getValuesByQualifier( entity, p, qualifierP, defaultId )
	local result = {}
	if not isSet( qualifierP ) then
		return result
	elseif type( defaultId ) ~= 'string' or defaultId == '' then
		defaultId = 'unknown'
	end

	local statements = wu.getStatements( entity, p, nil )
	if #statements > 0 then
		catTable[ p ] = ''
		local id, value
		for _, statement in ipairs( statements ) do
			-- defaultId is used if a qualifier is missing
			id = defaultId
			value = statement.mainsnak.datavalue.value
			if statement.qualifiers and statement.qualifiers[ qualifierP ] then
				for _, qualifier in ipairs( statement.qualifiers[ qualifierP ] ) do
					if qualifier.snaktype == 'value' then
						id = qualifier.datavalue.value.id
						if id then
							break
						end
					end
				end
			end
			result[ id ] = value
		end
	end
	return result
end

local function getValueFromDatavalue( datavalue )
	local v = datavalue.value
	local t = datavalue.type
	if type( v ) == 'table' then
		-- items which can be reduced to a string
		if t == 'wikibase-entityid' then
			v = v.id
		elseif t == 'quantity' then
			if v.unit == '1' then
				v = tonumber( v.amount ) or 1
			else
				v = v.amount:gsub( '^+', '' ) .. ' ' .. v.unit
			end
		elseif t == 'time' then
			v = v.time
		end
	end
	return v, t
end

-- The following function is an experimental one, not for extensive use
function wu.getValuesWithQualifiers( entity, p, qualifiers, count )
	local result = {}
	local statements = wu.getStatements( entity, p, count )
	if #statements == 0 then
		return result
	end

	if type( qualifiers ) == 'string' then
		qualifiers = { qualifiers }
	end

	local array
	for _, statement in ipairs( statements ) do
		array = { value = getValueFromDatavalue( statement.mainsnak.datavalue ),
			[ 'value-type' ] = statement.mainsnak.datavalue.type }
		if statement.qualifiers then
			if not qualifiers then -- all qualifier properties
				for key, qualTab in pairs( statement.qualifiers ) do
					for _, qual in ipairs( qualTab ) do
						if qual.snaktype == 'value' then
							array[ key ], array[ key .. '-type' ] =
								getValueFromDatavalue( qual.datavalue )
							break
						end
					end
				end
			else -- table of selected qualifier properties
				for key, value in ipairs( qualifiers ) do
					if statement.qualifiers[ value ] then
						for _, qual in ipairs( statement.qualifiers[ value ] ) do
							if qual.snaktype == 'value' then
								array[ value ], array[ value .. '-type' ] =
									getValueFromDatavalue( qual.datavalue )
								break
							end
						end
					end
				end
			end
		end
		table.insert( result, array )
	end
	return result
end

function wu.getAliases( entity, lang )
	if type( entity ) == 'string' then -- is Q id
		entity = mw.wikibase.getEntity( entity )
	end
	if not lang then
		lang = mw.getContentLanguage():getCode()
	end
	local aliases = {}
	if entity and entity.aliases and entity.aliases[ lang ] then
		for _, alias in ipairs( entity.aliases[ lang ] ) do
			table.insert( aliases, alias.value )
		end
	end
	return aliases
end

-- maintenance utilities
function wu.addProperty( p )
	catTable[ p ] = ''
end

function wu.getCategories( formatStr )
	local result = ''

	if not isSet( formatStr ) then
		formatStr = '[[Category:%s]]'
	end

	catTable.P0 = nil
	for key, value in pairs( catTable ) do
		result = result .. string.format( formatStr, key )
	end
	return result
end

return wu