prerna.wikidata.WikiLogicalNameCallable Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.wikidata;

import java.util.List;
import java.util.Map;
import java.util.Vector;
import java.util.concurrent.Callable;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.wikidata.wdtk.datamodel.implementation.ItemIdValueImpl;
import org.wikidata.wdtk.datamodel.implementation.PropertyIdValueImpl;
import org.wikidata.wdtk.datamodel.implementation.StringValueImpl;
import org.wikidata.wdtk.datamodel.interfaces.Claim;
import org.wikidata.wdtk.datamodel.interfaces.EntityDocument;
import org.wikidata.wdtk.datamodel.interfaces.ItemDocument;
import org.wikidata.wdtk.datamodel.interfaces.MonolingualTextValue;
import org.wikidata.wdtk.datamodel.interfaces.PropertyIdValue;
import org.wikidata.wdtk.datamodel.interfaces.Snak;
import org.wikidata.wdtk.datamodel.interfaces.Statement;
import org.wikidata.wdtk.datamodel.interfaces.StatementGroup;
import org.wikidata.wdtk.datamodel.interfaces.Value;
import org.wikidata.wdtk.datamodel.interfaces.ValueSnak;
import org.wikidata.wdtk.wikibaseapi.WbSearchEntitiesResult;
import org.wikidata.wdtk.wikibaseapi.WikibaseDataFetcher;

public class WikiLogicalNameCallable implements Callable>  {

	public static final Logger LOGGER = LogManager.getLogger(WikiLogicalNameCallable.class.getName());
	private Logger logger;
	
	WbSearchEntitiesResult res;
	WikibaseDataFetcher wbdf;
	
	public WikiLogicalNameCallable(WikibaseDataFetcher wbdf, WbSearchEntitiesResult res) {
		this.wbdf = wbdf;
		this.res = res;
	}

	@Override
	public List call() throws Exception {
		return getLogicalNames();
	}
	
	private List getLogicalNames() throws Exception {
		Logger logger = getLogger();
		List logicalNames = new Vector();
		
		String entityId = res.getEntityId();
		EntityDocument entity = wbdf.getEntityDocument(entityId);
		if(entity instanceof ItemDocument) {
			ItemDocument document = (ItemDocument) entity;
			String documentLabel = null;
			
			// for logging
			{
				Map labels = document.getLabels();
				if(labels.get("en") != null) {
					documentLabel= labels.get("en").getText();
					logger.info("Processing document = " + documentLabel);
				}
			}
			
			List pList = new Vector(5);
			// instance of
			// https://www.wikidata.org/wiki/Property:P31
			pList.add(new PropertyIdValueImpl("P31", "http://www.wikidata.org/entity/"));
			// occupation
			// https://www.wikidata.org/wiki/Property:P106
			pList.add(new PropertyIdValueImpl("P106", "http://www.wikidata.org/entity/"));
			// subclass of
			// https://www.wikidata.org/wiki/Property:P279
			pList.add(new PropertyIdValueImpl("P279", "http://www.wikidata.org/entity/"));
			// part of
			// https://www.wikidata.org/wiki/Property:P361
			pList.add(new PropertyIdValueImpl("P361", "http://www.wikidata.org/entity/"));
			// commons category
			// https://www.wikidata.org/wiki/Property:P373
			pList.add(new PropertyIdValueImpl("P373", "http://www.wikidata.org/entity/"));

			List statementGroups = document.getStatementGroups();
			for(StatementGroup group : statementGroups) {
				PropertyIdValue gProp = group.getProperty();
				if(pList.contains(gProp)) {
					List stmts = group.getStatements();
					for(Statement stmt : stmts) {
						Claim claim = stmt.getClaim();
						if(claim == null) {
							continue;
						}
						Snak snak = claim.getMainSnak();
						if(snak instanceof ValueSnak) {
							Value value = ((ValueSnak) snak).getValue();
							if(value instanceof StringValueImpl) {
								logicalNames.add(((StringValueImpl) value).getValue());
							} else if(value instanceof ItemIdValueImpl) {
								String innerId = ((ItemIdValueImpl) value).getId();
								EntityDocument innerEntity = wbdf.getEntityDocument(innerId);
								if(innerEntity instanceof ItemDocument) {
									ItemDocument innerDocument = (ItemDocument) innerEntity;
									MonolingualTextValue labels = innerDocument.getLabels().get("en");
									if(labels != null) {
										logicalNames.add(labels.getText());
									}
									// if it is the sublcass
									// also add the document label
									if(gProp.equals(pList.get(2))) {
										logicalNames.add(documentLabel);
									}
								}
							} else {
								System.out.println("Need to account for " + value.getClass().getName());
							}
						}
					}
				}
			}
		}
		
		return logicalNames;
	}

	public void setLogger(Logger logger) {
		this.logger = logger;
	}
	
	/**
	 * Get the correct logger
	 * @return
	 */
	private Logger getLogger() {
		if(this.logger == null) {
			return LOGGER;
		}
		return this.logger;
	}
}