Module:Essai

--[[
test de fonctionnalités. J'effacerai moi-même lorsque ce sera
terminé
--]]

local p = {}

-- module wikidata
local wd = require 'Module:Wikidata'

-- les données associées
local data = require "Module:Essai/données"

p.dump = ""
function p.add(txt)
	p.dump = p.dump .. txt .. "<br/>"
end


--[[
  fonction parcourant les "parents" de l'élément indiqué et retournant
  le premier élément présent dans la table 'association' durant ce processus
  (ou référencé par un 'identique à').
  Fait au maximum 'prof' itérations.
  Retourne 'nil' si rien n'est trouvé.
]]--
function p.cherche_parents(el, prof)
	if (prof <= 0) then
		return nil  -- fin de récursion
	end
	
	-- si l'élément courant est dans la table on le retourne (récursion)
	if (data.association[el] ~= nil) then
		return el
	end
	
	-- on récupère les P460 (réputé identique à) au cas où
	local args = { ["property"] = "P460", ["entity"] = el, link = "-", displayformat = "raw" }
	local id = wd.stringTable(args)
	if (id ~= nil) then
		-- correspond ?
		for k, v in pairs(id) do
			if (data.association[v] ~= nil) then
				return v  -- trouvé, on quitte
			end
		end
	end
	-- si on arrive ici c'est qu'il n'a pas d'équivalence. On parcours
	-- les "sous-classe de" (P279) récursivement
	args = { ["property"] = "P279", ["entity"] = el, link = "-", displayformat = "raw" }
	id = wd.stringTable(args)
	if (id ~= nil) then
		-- pour chaque "sur-classe" on se ré-appelle
		local ret
		for k, v in pairs(id) do
			ret = p.cherche_parents(v, prof-1)
			if (ret ~= nil) then
				return ret
			end
		end
	end
	-- ici on n'a rien trouvé → nil
	return nil
end

--[[
  fonction retournant une liste des éléments de la table 'p.association'
  trouvés à partir de la propriété 'property' pour l'entité 'entity' (ou l'article
  courant si 'entity' vaut 'nil')
  si une propriété présente n'est pas référencée ???
]]--
function p.propriete_parente(property, entity, prof)
	local args = { ["property"] = property, link = "-", displayformat = "raw" }
	local out = {} -- table des résultats
	if (entity ~= nil) then
		args["entity"] = entity
	end
	local res = wd.stringTable(args)
	
	-- si aucun résultat on retourne nil
	if (type(res) ~= "table") then
		return nil
	end
	
	if (prof == nil) then
		prof = 4
	end
	
	-- on parcours tous les résultats à la recherche de ceux qui sont dans la
	-- liste 'association' ou ayant un élément fils de l'un d'entre eux
	for k, v in pairs(res) do
		-- si l'élément est présent on l'insert directement
		if (data.association[v] ~= nil) then
			table.insert(out, v)
		else
			-- on cherche "récursivement" chez les parents
			-- la valeur numérique est la profondeur max de recherche
			local d = p.cherche_parents(v, prof)
			if (d ~= nil) then
				table.insert(out, d)
			end
		end
	end
	
	return out	
end

--[[
  fonction retournant true si la valeur 'value' est présente dans la liste des
  propriétés 'property' pour l'entité 'entity' (ou l'article courant si 'entity'
  vaut 'nil')
  retourne false sinon
]]--
function p.presence_propriete(property, value, entity)
	local args = { ["property"] = property, link = "-", displayformat = "raw" }
	if (entity ~= nil) then
		args["entity"] = entity
	end
	local res = wd.stringTable(args)
	
	-- pas de résultat, cette propriété WD est absente je présume
	if (res == nil) then
		return false
	end
	-- on parcours les résultats pour voir si celle qu'on veut est là
	for k, v in pairs(res) do
		if (v == value) then
			return true  -- trouvé
		end
	end
	return false -- pas trouvé
end

-- dédoublonne le contenu
function p.dedoublonne(tbl)
	if (type(tbl) ~= "table") then
		return tbl
	end
	local out = {}
	local tmp = {}
	for k, v in ipairs(tbl) do
		if (tmp[v] == nil) then
			table.insert(out, v)
			tmp[v] = true
		end
	end
	return out
end

-- conversion affichage d'une liste de 'association'
function p.affiche_elements(tbl)
	local ret = ""
	for k, v in ipairs(tbl) do
		ret = ret .. data.association[v]["texte"] .. " (" .. v .. ") "
	end
	return ret
end


--[[
  fonction qui retourne true si un élément wikidata existe pour l'article
  courant (ou l'élément 'entity' si différent de 'nil') et false sinon
  Note : le paramètre 'entity' n'a pas de sens, c'est juste pour du test. En
         effet si on a une 'entity' c'est que ça existe…
]]--
function p.a_wikidata(entity)
	if (entity == nil) then
		if (mw.wikibase.getEntity() ~= nil) then
			return true
		else
			return false
		end
	else -- cette partie disparaitra
		if (mw.wikibase.getEntity(entity) ~= nil) then
			return true
		else
			return false
		end
	end
end


function p.traite(entity, nom, prof)
	local ret = "Pour " .. nom .. " :<br/>"
	local t1 = p.a_wikidata(entity)
	if (t1) then
		ret = ret .. "possède une entrée WD<br/>"
		local t2 = p.presence_propriete("P31", "Q5", entity)
		if (t2) then
			ret = ret .. "c'est un être humain<br/>"
			-- suite : les métiers
			local t3 = p.propriete_parente("P106", entity, prof)
			if (t3) then
				t3 = p.dedoublonne(t3)
				ret = ret .. "métier(s) (P106) trouvés : "
				ret = ret .. p.affiche_elements(t3) .. "<br/>"
			else
				ret = ret .. "pas de métiers (P106) trouvé dans la liste<br/>"
			end
		else
			ret = ret .. "ce n'est pas un être humain<br/>"
		end
	else
		ret = ret .. "n'a pas d'entrée WD<br/>"
	end
	ret = ret .. "fin test pour " .. nom .. ".<br/><br/>"
	return ret
end


function p.essai(frame)
	local ret = ""
	-- test sur l'article en cours (page de discussion)
	ret = ret .. p.traite(nil, "[[Discussion module:Essai]]") -- traite la page en cours
	ret = ret .. p.traite("Q392", "[[Bob Dylan]]") -- traite Bob Dylan
    ret = ret .. p.traite("Q2042", "[[Charles de Gaulle]]")

	return ret
end


function p.test(frame)
	local ret = ""
	-- test sur l'article en cours (page de discussion)
    ret = ret .. p.traite("Q2042", "[[Charles de Gaulle]]", 4)

	return ret
end

function p.test2(frame)
	local ret = ""
	-- test sur l'article en cours (page de discussion)
    ret = ret .. p.traite("Q2042", "[[Charles de Gaulle]]", 6)

	return ret
end


--[[
  Fonction externe (appelable) qui catégorise l'article courant en fonction de WD
  Ne reçoit actuellement aucun paramètre
]]--
function p.categorise(frame)
	-- si on n'est pas dans l'espace encyclopédique on ne fait rien
	local titre = mw.title.getCurrentTitle()
	if ((titre == nil) or (titre.namespace ~= 0)) then
		return ""  -- on retourne un texte vide, donc l'action est inexistante
	end
	
	-- on regarde si la page a une entrée wikidata
	local awd = p.a_wikidata(nil)
	if (not awd) then
		-- on catégorise et on quitte
		return "[[Catégorie:Wikipédia:Article sans Wikidata appelant le modèle Suivi des biographies]]"
	end
	
	-- on regarde si c'est un humain
	local hwd = p.presence_propriete("P31", "Q5", nil)
	if (not hwd) then
		-- on catégorise et on quitte
		return "[[Catégorie:Wikipédia:Article non biographique appelant le modèle Suivi des biographies]]"
	end
	
	-- là il faut aller plus loin et récupérer les métiers pour catégoriser
	local mwd = p.propriete_parente("P106", nil)
	-- si aucune réponse on le place dans "autres"
	if (mwd == nil) then
		-- on catégorise et on quitte
		return "[[Catégorie:Portail:Biographie/Articles liés/Général]]"
	end
	-- on a (au moins) une réponse : on regarde ce que contient le 1er élément (le "principal")
	--[[ note : on pourrait faire du multi-catégorie (ne pas utiliser de else/elseif alors
	              mais "cumuler" diverses catégories et retourner le tout à la fin)
	            on pourrait aussi faire des tests plus complexes du style si militaire et pas
	              homme politique alors faire ci sinon si cela faire ça…
	]]--  
	if (mwd[1] == "Q483501") then
		return "[[Catégorie:Portail:Biographie/Articles liés/Culture et arts]]"
	elseif (mwd[1] == "Q47064") then
		return "[[Catégorie:Portail:Biographie/Articles liés/Militaire]]"
	elseif (mwd[1] == "Q82955") then
		return "[[Catégorie:Portail:Biographie/Articles liés/Politique]]"
	elseif (mwd[1] == "Q2066131") then
		return "[[Catégorie:Portail:Biographie/Articles liés/Sport]]"
	else
		-- catégorie par défaut
		return "[[Catégorie:Portail:Biographie/Articles liés/Général]]"
	end
	return "" -- on ne devrait pas arriver ici
end



--[[
  Fonction externe (appelable) qui catégorise l'article courant en fonction de WD
  Ne reçoit actuellement aucun paramètre
  version "2", avec cumul de catégories
]]--
function p.categorise2(frame)
	-- si on n'est pas dans l'espace encyclopédique on ne fait rien
	local titre = mw.title.getCurrentTitle()
	if ((titre == nil) or (titre.namespace ~= 0)) then
		return ""  -- on retourne un texte vide, donc l'action est inexistante
	end
	
	-- on regarde si la page a une entrée wikidata
	local awd = p.a_wikidata(nil)
	if (not awd) then
		-- on catégorise et on quitte
		return data.cat_badwd
	end
	
	-- on regarde si c'est un humain
	local hwd = p.presence_propriete("P31", "Q5", nil)
	if (not hwd) then
		-- on catégorise et on quitte
		return data.cat_badcall
	end
	
	-- pour cumuler les éléments à retourner (maintenant on sait que c'est une biographie)
	local out = data.cat_biographie
	
	-- là il faut aller plus loin et récupérer les métiers pour catégoriser
	local mwd = p.propriete_parente("P106", nil)
	-- si aucune réponse on le place dans "autres"
	if (mwd == nil) then
		-- on catégorise
		return out .. data.cat_defaut
	else
		-- on a (au moins) une réponse : on parcours chaque entrée et on cumule les catégories associées
		nb = 0
		for k, v in ipairs(mwd) do
			out = out .. data.association[v]["catégorie"]
			nb = nb + 1
		end
		-- si 0 catégorie insérée on met celle par défaut
		if (nb == 0) then
			out = out .. data.cat_defaut
		end
	end

	-- on retourne le résultat final
	return out
end


return p