You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
wikiteam/listsofwikis/mediawiki/shoutwiki-spider.py

75 lines
2.4 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (C) 2022 Simon Liu
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import time
import requests
from tqdm import tqdm
def main():
ids, wikis = [], []
gcont = 'tmp'
url = 'http://www.shoutwiki.com/w/api.php'
headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux i686; rv:24.0) Gecko/20100101 Firefox/24.0'}
# grab wiki pages
params = {
'action': 'query',
'format': 'json',
'prop': 'info',
'generator': 'categorymembers',
'inprop': 'url',
'gcmtitle': 'Category:Flat_list_of_all_wikis',
'gcmlimit': 'max'
}
while gcont:
if gcont != 'tmp':
params['gcmcontinue'] = gcont
json = requests.get(url, params=params, headers=headers).json()
gcont = json['continue']['gcmcontinue'] if 'continue' in json else ''
query = json['query']['pages']
for wiki in query:
ids.append(wiki)
# grab wiki API
params = {
'action': 'query',
'format': 'json',
'prop': 'revisions',
'formatversion': '2',
'rvprop': 'content',
'rvslots': '*'
}
for n in tqdm(range(0, len(ids), 50)):
params['pageids'] = '|'.join(ids[n:n+50])
json = requests.get(url, params=params, headers=headers).json()
for wiki in json['query']['pages']:
for val in wiki['revisions'][0]['slots']['main']['content'].split('\n|'):
if 'subdomain' in val:
wikis.append('http://%s.shoutwiki.com/w/api.php' % val.split('subdomain =')[-1].strip())
break
time.sleep(0.3)
wikis = list(set(wikis))
wikis.sort()
with open('shoutwiki.com', 'w') as f:
f.write('\n'.join(wikis))
if __name__ == '__main__':
main()