def apply_endpoint(entity): # find types
# print(entity)
ep = DBpediaEndpoint() # using ID/int
ent2 = entity.getIdstr()
types = ep.getTypesForEntity(ent2) # limit to 5
# print('types using endpoint id', types)
if len(types) == 0: # using entity: back up
types = entity.getTypes() # ont
# print('types using entity', types, '\n')
return types
def apply_endpoint_list(entity_list): # find types
types_list = []
ep = DBpediaEndpoint() # using ID/int
for entity in entity_list:
print(entity)
# ['N/A']
# < id: http://dbpedia.org/resource/Ana_Popović, label: Ana Popović, description: None, types: set(), source: DBpedia >
if entity != ['N/A']:
ent2 = entity.getIdstr()
types = ep.getTypesForEntity(ent2) # limit to 5
# print('types using endpoint id', types)
types_list.append(types)
if len(types) == 0: # using entity: back up
types = entity.getTypes() # ont
# print('types using entity', types, '\n')
types_list.append(types)
else:
types = []
types_list.append(types)
return types_list