rename file
This commit is contained in:
Sarah
parent
1067a193cf
commit
586ccfb997
|
|
@ -1,81 +1,69 @@
|
|||
from bookstack import Bookstack
|
||||
from vcinema_utils import VCinemaUtils
|
||||
|
||||
import argparse
|
||||
import base64
|
||||
from collections import Counter, OrderedDict
|
||||
import csv
|
||||
import os
|
||||
from progress.bar import Bar
|
||||
from progress.bar import IncrementalBar
|
||||
from wand.image import Image
|
||||
import worldmap
|
||||
import warnings
|
||||
warnings.filterwarnings("ignore")
|
||||
|
||||
|
||||
from bookstack import Bookstack
|
||||
from vcinema_utils import VCinemaUtils
|
||||
def get_flags_dict():
|
||||
flags = {}
|
||||
|
||||
with open('country_flags.csv', newline='') as f:
|
||||
reader = csv.reader(f, quotechar="\"")
|
||||
next(reader, None) # skip the headers
|
||||
|
||||
for row in reader:
|
||||
flags[row[0]] = row[1]
|
||||
|
||||
return flags
|
||||
|
||||
|
||||
FLAGS = {
|
||||
"Australia": "🇦🇺",
|
||||
"Belgium": "🇧🇪",
|
||||
"Canada": "🇨🇦",
|
||||
"China": "🇨🇳",
|
||||
"Czech Republic": "🇨🇿",
|
||||
"Czechia": "🇨🇿",
|
||||
"France": "🇫🇷",
|
||||
"Germany": "🇩🇪",
|
||||
"Hong Kong": "🇭🇰",
|
||||
"Hong Kong SAR China": "🇭🇰",
|
||||
"Italy": "🇮🇹",
|
||||
"Israel": "🇮🇱",
|
||||
"Japan": "🇯🇵",
|
||||
"Luxembourg": "🇱🇺",
|
||||
"Mexico": "🇲🇽",
|
||||
"New Zealand": "🇳🇿",
|
||||
"Nigeria": "🇳🇬",
|
||||
"North Korea": "🇰🇵",
|
||||
"Philippines": "🇵🇭",
|
||||
"Russia": "🇷🇺",
|
||||
"Spain": "🇪🇸",
|
||||
"South Africa": "🇿🇦",
|
||||
"South Korea": "🇰🇷",
|
||||
"Taiwan": "🇹🇼",
|
||||
"Thailand": "🇹🇭",
|
||||
"Turkey": "🇹🇷",
|
||||
"Uganda": "🇺🇬",
|
||||
"United Kingdom": "🇬🇧",
|
||||
"United States": "🇺🇸",
|
||||
"West Germany": "⬅️🇩🇪",
|
||||
}
|
||||
def build_table(films_by_country, bar=None):
|
||||
films_by_country_sorted = OrderedDict(sorted(films_by_country.items(), key=lambda t: t[0]))
|
||||
|
||||
|
||||
def build_table(fby, bar):
|
||||
fby_sorted = OrderedDict(sorted(fby.items(), key=lambda t: t[0]))
|
||||
flags = get_flags_dict()
|
||||
|
||||
table = "<table>"
|
||||
|
||||
table += "<thead><tr><th>Country</th><th>Films</th></tr>"
|
||||
table += "<tbody>"
|
||||
|
||||
for country in fby_sorted.keys():
|
||||
for country in films_by_country_sorted.keys():
|
||||
row = "<tr>"
|
||||
|
||||
row += "<td>"
|
||||
row += "{} {}".format(str(country), FLAGS[country])
|
||||
row += "{} {}".format(str(country), flags[country] if country in flags.keys() else "")
|
||||
row += "</td>"
|
||||
|
||||
row += "<td>"
|
||||
|
||||
row += "<br>".join(["<a href=\"{}\">{}</a>".format(film['imdb_id'], film['title']) for film in fby_sorted[country]])
|
||||
film_links = ["<a href=\"https://www.imdb.com/title/tt{}/\">{}</a>".format(film['imdb_id'], film['title'])
|
||||
for film in films_by_country_sorted[country]]
|
||||
|
||||
row += "<br>".join(film_links)
|
||||
|
||||
row += "</td>"
|
||||
row += "</tr>"
|
||||
|
||||
table += row
|
||||
|
||||
bar.next()
|
||||
if bar is not None:
|
||||
bar.next()
|
||||
|
||||
table += "</tbody>"
|
||||
table += "</table>"
|
||||
|
||||
if bar is not None:
|
||||
bar.finish()
|
||||
|
||||
return table
|
||||
|
||||
|
||||
|
|
@ -87,7 +75,7 @@ def draw_map(counter, file_name="vcinema_map.svg"):
|
|||
|
||||
opacity = [0.5 + (x / (float(max_count))/2.0) for x in counts]
|
||||
|
||||
worldmap.plot(countries, cmap=["#ff0000"], opacity=opacity, filename=file_name, verbose=False)
|
||||
worldmap.plot(countries, cmap=["#49E20E"], opacity=opacity, filename=file_name, verbose=False)
|
||||
|
||||
with Image(filename=file_name, width=1000, height=655) as i:
|
||||
png_data = i.make_blob("png")
|
||||
|
|
@ -109,28 +97,32 @@ def update_page(map_png, table, wiki_url, token_id, token_secret):
|
|||
Bookstack.update_page(wiki_url, token_id, token_secret, page_id, html=page_contents)
|
||||
|
||||
|
||||
def update_films_by_country_page(token_id, token_secret):
|
||||
def update_films_by_country(token_id, token_secret):
|
||||
print("Retrieving VCinema viewings")
|
||||
viewings = VCinemaUtils.get_vcinema_viewings(token_id, token_secret)
|
||||
|
||||
viewing_count = len(viewings)
|
||||
|
||||
with Bar('Retrieving movie data', max=viewing_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
|
||||
with IncrementalBar('Retrieving movie data', max=viewing_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
|
||||
VCinemaUtils.add_imdb_data_to_viewings(viewings, ['countries'], bar)
|
||||
|
||||
with Bar('Processing viewing data', max=viewing_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
|
||||
with IncrementalBar('Processing viewing data', max=viewing_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
|
||||
viewings_by_country = VCinemaUtils.filter_viewings(viewings, 'countries', bar)
|
||||
|
||||
country_count = len(viewings_by_country.keys())
|
||||
with Bar('Generating table', max=country_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
|
||||
with IncrementalBar('Generating table', max=country_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
|
||||
film_by_country_table = build_table(viewings_by_country, bar)
|
||||
|
||||
print("Drawing map")
|
||||
country_counter = Counter(viewings_by_country)
|
||||
png_data = draw_map(country_counter)
|
||||
|
||||
print("Updating page")
|
||||
|
||||
update_page(png_data, film_by_country_table, VCinemaUtils.JACKNET_WIKI_URL, token_id, token_secret)
|
||||
|
||||
print("Done!")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
parser = argparse.ArgumentParser(description='Update page displaying VCinema films sorted by country.')
|
||||
|
|
@ -139,4 +131,4 @@ if __name__ == '__main__':
|
|||
|
||||
args = parser.parse_args()
|
||||
|
||||
update_films_by_country_page(args.token_id, args.token_secret)
|
||||
update_films_by_country(args.token_id, args.token_secret)
|
||||
Loading…
Reference in New Issue