from collections import Counter import webbrowser import pygal import numpy as np import pygeoip from wikibania.ban.BanDB import BanDB from wikibania.ban.BanDBWrapper import BanDBWrapper import sysproxy # PARAMS GEOIP_DB = "/usr/share/GeoIP/GeoIP.dat" BAN_MAP_FILE = "output/ban-map.svg" BAN_DURATION_MAP_FILE = "output/ban-duration-map.svg" HIST_FILE = "output/histogram.svg" STATS_FILE = "output/stats.txt" BAN_DB_FILE = "resources/ban_list.json" FETCH_SAMPLES = 2000 FETCH_DB = False DUMP_DB = False LOAD_DB = True OPEN_FILES = False # SETUP sysproxy.configure_system_proxy() geoip_looker = pygeoip.GeoIP(GEOIP_DB) ban_db = BanDB(geoip_looker) if FETCH_DB: ban_db.fetch(FETCH_SAMPLES) if LOAD_DB: ban_db.load_file(BAN_DB_FILE) if DUMP_DB: ban_db.dump_file(BAN_DB_FILE) ban_db_wrapper = BanDBWrapper(ban_db) # HISTOGRAM ban_durations = ban_db_wrapper.get_all_durations() (ban_durations_bars, bins) = np.histogram(ban_durations, bins=range(1, 50 + 2)) bar_chart = pygal.Bar(legend_at_bottom=True) bar_chart.title = "Active Wikipedia bans by duration (%d samples)" % len(ban_db.list()) bar_chart.x_labels = map(str, range(1, len(ban_durations_bars) + 1)) bar_chart.add("Number of active bans", ban_durations_bars) bar_chart.render_to_file(HIST_FILE) if OPEN_FILES: webbrowser.open(HIST_FILE, 2) # BAN DURATION STATS with open(STATS_FILE, "w") as stats_file: stats_file.write("Ban duration:\n") stats_file.write("Mean: %.2f days\n" % np.mean(ban_durations)) stats_file.write("Median: %.2f days\n" % np.median(ban_durations)) stats_file.write("Deviation: %.2f\n" % np.std(ban_durations)) stats_file.write("Variance: %.2f\n" % np.var(ban_durations)) if OPEN_FILES: webbrowser.open(STATS_FILE, 2) # NB BAN MAP country_ban_list = ban_db_wrapper.get_all_countries() nb_bans_by_country = Counter(country_ban_list) worldmap_chart = pygal.Worldmap(legend_at_bottom=True) worldmap_chart.title = "World active Wikipedia bans by country (%d samples)" % len(ban_db.list()) worldmap_chart.add("Active bans", nb_bans_by_country) worldmap_chart.render_to_file(BAN_MAP_FILE) if OPEN_FILES: webbrowser.open(BAN_MAP_FILE, 2) # BAN DURATION MAP average_ban_duration_by_country = ban_db_wrapper.calc_average_duration_by_country() worldmap_chart = pygal.Worldmap(legend_at_bottom=True) worldmap_chart.title = "Average Wikipedia ban duration by country (%d samples)" % len(ban_db.list()) worldmap_chart.add("Average ban duration (months)", average_ban_duration_by_country) worldmap_chart.render_to_file(BAN_DURATION_MAP_FILE) if OPEN_FILES: webbrowser.open(BAN_DURATION_MAP_FILE, 2)