vcinema/update_films_by_year_page.py

import argparse
from collections import OrderedDict

from bookstack import Bookstack
from vcinema_utils import VCinemaUtils


def build_table(films_by_year):
    fby_sorted = OrderedDict(sorted(films_by_year.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)

    # Page ID of https://wiki.jacknet.io/books/vcinema/page/films-by-release-year
    page_id = "24"

    print("Updating page")
    Bookstack.update_page(VCinemaUtils.JACKNET_WIKI_URL, token_id, token_secret, page_id, markdown=film_by_year_table)

    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)