vcinema/update_films_by_year_page.py

56 lines
1.7 KiB
Python

import argparse
from collections import OrderedDict
from vcinema_utils import VCinemaUtils
from wiki_utils import WikiUtils
JACKNET_WIKI_URL = "https://wiki.jacknet.io"
def build_table(fby):
fby_sorted = OrderedDict(sorted(fby.items(), key=lambda t: t[0], reverse=True))
page_table = "| Year | Films |\n| - | - |\n"
for year in fby_sorted.keys():
page_table += str(year) + " | "
page_table += "<br>".join("[{}](https://www.imdb.com/title/tt{}/)".format(film['title'], film['imdb_id']) for film in fby_sorted[year])
page_table += "\n"
return page_table
def update_films_by_year_page(token_id, token_secret):
print("Retrieving VCinema viewings")
viewings = VCinemaUtils.get_vcinema_viewings(token_id, token_secret)
print("Retrieving movie data")
VCinemaUtils.add_imdb_data_to_viewings(viewings, 'year')
print("Processing viewing data")
viewings_by_year = VCinemaUtils.filter_viewings(viewings, 'year')
print("Generating table")
film_by_year_table = build_table(viewings_by_year)
wiki_url = JACKNET_WIKI_URL
# Page ID of https://wiki.jacknet.io/books/vcinema/page/films-by-release-year
page_id = "24"
print("Updating page")
WikiUtils.update_page(page_id, film_by_year_table, wiki_url, token_id, token_secret)
print("Done!")
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Update page displaying VCinema films sorted by release year.')
parser.add_argument('token_id', help='API token ID.')
parser.add_argument('token_secret', help='API token secret.')
args = parser.parse_args()
update_films_by_year_page(args.token_id, args.token_secret)