Quelltext der Seite Modul:SmwUtil

local p = {}
local classDebug = require('Module:Debug/class')
local getArgs = require('Module:Arguments').getArgs
local smw = mw.smw
local _TT = require('Module:TableTools')

local cfg = {
	nameMainLabel = 'PageName',
}

local buildQueryFrom = function(stash)
	local query = {}
	if type(stash) == 'table' then
		for k, v in pairs(stash) do
			if type(v) == 'table' then
				for _, vv in pairs(v) do
					local val = vv
					if type(vv) == 'boolean' then
						val = vv and 'wahr' or 'falsch'
					end
					if type(k) == 'number' then
						table.insert(query, val)
					elseif type(val) == 'table' then
						error( 'While building set-query, detected invalid table value for key ' .. k .. ': ' .. _TT.printTable(val) )
					else
						table.insert(query, k .. '=' .. val)
					end
				end
			else
				local val = v
				if type(v) == 'boolean' then
					val = v and 'wahr' or 'falsch'
				end
				if type(k) == 'number' then
					table.insert(query, val)
				else
					table.insert(query, k .. '=' .. val)
				end
			end
		end
	else
		query[1] = stash
	end
	return query
end

local extractFieldFrom = function(s)
	local ret = '?' .. s:match('^%??([^#=]+).*$')
	local headline = s:find('=', 1, true) and
		s:match('^.*=([^=]*)$') or nil
	local formatter = s:find('#', 1, true) and
		s:match('^.*#([^=]*).*$') or nil
	if formatter and not formatter:find('-', 1, true) then
		formatter = formatter .. '-'
	end
	return ret .. ( formatter and ('#' .. formatter) or '' ) .. ( headline and ( '=' .. headline ) or '' )
end

function p.ask(query, attributes, retainBlanks)
	classDebug:log(21, 'entering SmwUtil.ask', 'SmwUtil.ask')
	classDebug:log(23, ' with query: ' .. (query and _TT.printTable(query) or 'NONE'), 'SmwUtil.ask')
	classDebug:log(23, ' with attributes: ' .. (attributes and _TT.printTable(attributes) or 'NONE'), 'SmwUtil.ask')
	local select = query.select and query.select or (query.where and query.where or nil)
	local fields = query.fields
	local attributes = attributes or {}
	if not select or (type(select) == 'table' and _TT.size(select) == 0) or (#select == 0) then
		error('SmwUtil.ask: Argument query.select is missing or empty!', 2)
	end
	if type(attributes) == 'string' then
		attributes = { attributes }
	elseif type(attributes) ~= 'table' then
		error('SmwUtil.ask: Argument attributes has wrong parameter type (' .. type(attributes) .. '); must be table or string', 2)
	end
	if not attributes.limit then
		attributes.limit = 1000
	end
	local nameMainlabel = (attributes.mainlabel and attributes.mainlabel ~= '-') and attributes.mainlabel or cfg.nameMainLabel
	attributes.mainlabel = '-'
	-- now build the query
	local query = {}
	if type(select) == 'string' then
		table.insert(query, select)
	else
		for k, v in pairs(select) do
			if type(k) == 'number' then
				table.insert(query, v)
			else
				table.insert(query, '[[' .. k .. '::' .. v .. ']]')
			end
		end
	end
	table.insert(query, '?#-=' .. nameMainlabel)
	if type(fields) == 'string' then
		table.insert(query, extractFieldFrom(fields))
	elseif type(fields) == 'table' then
		for _, v in pairs(fields) do
			table.insert(query, extractFieldFrom(v))
		end
	end
	for k, v in pairs(attributes) do
		if type(k) == 'number' then
			table.insert(query, v)
		else
			table.insert(query, k .. '=' .. v)
		end
	end
	local result = smw.ask(query)
	if type(result) == 'table' and result[nameMainlabel] then
		result[1] = nameMainlabel
	end
	return result, query
end

function p.rawask(query, attributes)
	classDebug:log(21, 'entering SmwUtil.rawask', 'SmwUtil.rawask')
	classDebug:log(23, ' with query: ' .. (query and _TT.printTable(query) or 'NONE'), 'SmwUtil.rawask')
	classDebug:log(23, ' with attributes: ' .. (attributes and _TT.printTable(attributes) or 'NONE'), 'SmwUtil.rawask')
	local frame = mw.getCurrentFrame()
	local select = query.select and query.select or (query.where and query.where or nil)
	local fields = query.fields
	local selection = ''
	if type(select) == 'string' then
		selection = select
	elseif type(select) == 'table' then
		for k, v in pairs(select) do
			if type(k) == 'number' then
				selection = selection .. ' ' .. v
			else
				selection = selection .. ' [[' .. k .. '::' .. v .. ']]'
			end
		end
	else
		error('SmwUtil.rawask: Argument query.select has wrong parameter type (' .. type(select) .. '); must be table or string', 2)
	end
	local args = {}
	if type(fields) == 'string' then
		args[1] = mw.ustring.sub( fields, 1, 1 ) ~= '?' and '?' .. fields or fields
	elseif type(fields) == 'table' then
		for k, v in pairs(fields) do
			args[k] = mw.ustring.sub( v, 1, 1 ) ~= '?' and '?' .. v or v
		end
	end
	if attributes then
		if type(attributes) == 'string' then
			table.insert(args, attributes)
		elseif type(attributes) == 'table' then
			for k, v in pairs(attributes) do
				args[k] = v
			end
		else
			error('SmwUtil.rawask: Argument attributes has wrong parameter type (' .. type(attributes) .. '); must be table or string', 2)
		end
	end
	return frame:callParserFunction{ name = '#ask:' .. selection, args = args}
end

function p.set(stash)
	classDebug:log(21, 'entering SmwUtil.set', 'SmwUtil.set')
	classDebug:log(23, ' with stash: ' .. (stash and _TT.printTable(stash) or 'NONE'), 'SmwUtil.set')
	if type(stash) ~= 'string' and type(stash) ~= 'table' then
		error('SmwUtil.set: Argument stash has wrong parameter type (' .. type(stash) .. '); must be table or string', 2)
	end
	-- build and execute query
	local query = buildQueryFrom(stash)
	local result = smw.set(query)
	if result == true then
		return ''
	else
		return result.error
	end
end

function p.show(page, field)
	classDebug:log(21, 'entering SmwUtil.show', 'SmwUtil.show')
	classDebug:log(23, ' with page: ' .. (page and page or 'NONE'), 'SmwUtil.show')
	classDebug:log(23, ' with field: ' .. (field and field or 'NONE'), 'SmwUtil.show')
	local frame = mw.getCurrentFrame()
	local page = page:match('^[%[]*([^%[%]]+)[%]]*$')
	local field = mw.ustring.sub( field, 1, 1 ) ~= '?' and '?' .. field or field
	return frame:callParserFunction('#show:' .. page, field)
end

function p.subobject(stash, uid)
	classDebug:log(21, 'entering SmwUtil.subobject', 'SmwUtil.subobject')
	classDebug:log(23, ' with stash: ' .. (stash and _TT.printTable(stash) or 'NONE'), 'SmwUtil.subobject')
	classDebug:log(23, ' with uid: ' .. (uid and uid or 'NONE'), 'SmwUtil.subobject')
	if type(stash) ~= 'string' and type(stash) ~= 'table' then
		error('SmwUtil.subobject: Argument stash has wrong parameter type (' .. type(stash) .. '); must be table or string', 2)
	end
	-- build and execute query
	local uid = uid or nil
	local query = buildQueryFrom(stash)
	local result = smw.subobject(query, uid)
	if result == true then
		return ''
	else
		return result.error
	end
end 
-- ************************************
-- alias definition
-- ************************************

p.get = p.show
p.lua = p.ask

return p