从Wikidata获取Wikipedia infoxbox数据

import requests

def get_wikidata_info(entity_id):
    url = f"https://www.wikidata.org/wiki/Special:EntityData/{entity_id}.json"
    response = requests.get(url)
    
    if response.status_code == 200:
        data = response.json()
        return data['entities'][entity_id]
    else:
        return None

def extract_infobox_data(entity_data):
    if 'P31' in entity_data:  # P31 is the property for instance of
        labels = entity_data.get('labels', {})
        descriptions = entity_data.get('descriptions', {})
        # 这里可以根据需要提取更多的属性
        return {
            'labels': labels,
            'descriptions': descriptions,
            # 添加其他需要的字段
        }
    return None

# 使用示例
entity_id = 'Q9527'  # 例如，Albert Einstein的Wikidata ID
entity_data = get_wikidata_info(entity_id)
if entity_data:
    infobox_data = extract_infobox_data(entity_data)
    print(infobox_data)
else:
    print("Failed to retrieve data.")