|
|
|
#!/bin/env python3
|
|
|
|
# a tool for collecting mastodon /about/ & ToS and CoC pages.
|
|
|
|
# (c) roel roscam abbing 2020
|
|
|
|
# gplv3
|
|
|
|
import os, time, json
|
|
|
|
import fedicrawler
|
|
|
|
|
|
|
|
filters = [fedicrawler.only_netloc]
|
|
|
|
|
|
|
|
scrape_data = json.loads(open('instance_scrape.json').read())
|
|
|
|
|
|
|
|
output_dir = 'about_pages'
|
|
|
|
tos_dir = os.path.join(output_dir,'with_tos')
|
|
|
|
|
|
|
|
if not os.path.exists(output_dir):
|
|
|
|
os.mkdir(output_dir)
|
|
|
|
if not os.path.exists(tos_dir):
|
|
|
|
os.mkdir(tos_dir)
|
|
|
|
|
|
|
|
def find_mastodon_instances(scrape_data):
|
|
|
|
mastodon_instances = []
|
|
|
|
|
|
|
|
for i in scrape_data:
|
|
|
|
if 'nodeinfo' in scrape_data[i].keys():
|
|
|
|
if 'software' in scrape_data[i]['nodeinfo'].keys():
|
|
|
|
if scrape_data[i]['nodeinfo']['software']['name'].lower() == 'mastodon':
|
|
|
|
mastodon_instances.append(i)
|
|
|
|
|
|
|
|
#mastodon_instances = fedicrawler.multi_filter(filters,mastodon_instances)
|
|
|
|
return mastodon_instances
|
|
|
|
|
|
|
|
|
|
|
|
from selenium import webdriver # you need to set up a driver
|
|
|
|
from selenium.webdriver.firefox.options import Options
|
|
|
|
|
|
|
|
options = Options()
|
|
|
|
options.add_argument('-headless')
|
|
|
|
|
|
|
|
browser = webdriver.Firefox(options=options)
|
|
|
|
#browser.set_window_size(1024, 768) # set the window size that you need
|
|
|
|
|
|
|
|
for mi in find_mastodon_instances(scrape_data):
|
|
|
|
|
|
|
|
if not os.path.exists(os.path.join(output_dir,mi+'.png')) and not os.path.exists(os.path.join(tos_dir,mi+'.png')):
|
|
|
|
|
|
|
|
browser.get('https://{}/about/more/'.format(mi))
|
|
|
|
page = browser.find_element_by_tag_name('html')
|
|
|
|
height=page.size['height']
|
|
|
|
browser.set_window_size(1400,height+100)
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
bw = browser.find_element_by_css_selector('div .box-widget')
|
|
|
|
about_text = bw.find_element_by_css_selector('div .rich-formatting')
|
|
|
|
html = browser.execute_script('return arguments[0].innerHTML;',about_text)
|
|
|
|
if html:
|
|
|
|
output_dir = tos_dir
|
|
|
|
with open(os.path.join(output_dir,mi+'_tos.txt'),'w') as f:
|
|
|
|
f.write(html)
|
|
|
|
except Exception as e:
|
|
|
|
output_dir = 'about_pages'
|
|
|
|
print(e)
|
|
|
|
|
|
|
|
time.sleep(0.1)
|
|
|
|
print('taking screenshot of', mi)
|
|
|
|
browser.save_screenshot(os.path.join(output_dir,mi+'.png'))
|
|
|
|
with open(os.path.join(output_dir,mi+'_about.html'),'w') as f:
|
|
|
|
f.write(browser.page_source)
|
|
|
|
else:
|
|
|
|
print(mi, '✔')
|
|
|
|
#browser.save_screenshot()
|
|
|
|
browser.quit()
|