mirror of
https://github.com/MISP/misp-galaxy.git
synced 2024-11-23 07:17:17 +00:00
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
|
import requests
|
||
|
|
|
||
|
|
class WikipediaAPI():
|
||
|
|
def __init__(self):
|
||
|
|
self.base_url = 'https://en.wikipedia.org/w/api.php'
|
||
|
|
|
||
|
|
def get_page_summary(self, page_title):
|
||
|
|
params = {
|
||
|
|
'action': 'query',
|
||
|
|
'format': 'json',
|
||
|
|
'titles': page_title,
|
||
|
|
'prop': 'extracts',
|
||
|
|
'explaintext': True,
|
||
|
|
}
|
||
|
|
|
||
|
|
try:
|
||
|
|
response = requests.get(self.base_url, params=params)
|
||
|
|
data = response.json()
|
||
|
|
page_id = next(iter(data['query']['pages']))
|
||
|
|
return data['query']['pages'][page_id]['extract']
|
||
|
|
except Exception as e:
|
||
|
|
print(f'Error: {e}')
|
||
|
|
return None
|
||
|
|
|
||
|
|
def get_page_content(self, page_title):
|
||
|
|
params = {
|
||
|
|
'action': 'query',
|
||
|
|
'format': 'json',
|
||
|
|
'titles': page_title,
|
||
|
|
'prop': 'revisions',
|
||
|
|
'rvprop': 'content',
|
||
|
|
}
|
||
|
|
try:
|
||
|
|
response = requests.get(self.base_url, params=params)
|
||
|
|
data = response.json()
|
||
|
|
page_id = next(iter(data['query']['pages']))
|
||
|
|
return data['query']['pages'][page_id]['revisions'][0]['*']
|
||
|
|
except Exception as e:
|
||
|
|
print(f'Error: {e}')
|
||
|
|
return None
|
||
|
|
|
||
|
|
def get_page_html(self, page_title):
|
||
|
|
params = {
|
||
|
|
'action': 'parse',
|
||
|
|
'format': 'json',
|
||
|
|
'page': page_title,
|
||
|
|
'prop': 'text',
|
||
|
|
'disableeditsection': True,
|
||
|
|
}
|
||
|
|
try:
|
||
|
|
response = requests.get(self.base_url, params=params)
|
||
|
|
data = response.json()
|
||
|
|
return data['parse']['text']['*']
|
||
|
|
except Exception as e:
|
||
|
|
print(f'Error: {e}')
|
||
|
|
return None
|