wrapped.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import urllib.request
from datetime import datetime
import os
import isodate


def checkYear(time, year):
    try:
        date_to_check = datetime.strptime(time, "%Y-%m-%dT%H:%M:%S.%fZ")
    except ValueError:
        date_to_check = datetime.strptime(time, "%Y-%m-%dT%H:%M:%SZ")

    return date_to_check.year == year


def getVideoLength(url):
    temp = url.split('?v=')
    video_id = temp[-1]
    api_key = os.environ["API_KEY"] # free from https://console.developers.google.com/ 
    searchUrl = (
        "https://www.googleapis.com/youtube/v3/videos?id="
        + video_id
        + "&key="
        + api_key
        + "&part=contentDetails"
    )
    response = urllib.request.urlopen(searchUrl).read()
    data = json.loads(response)
    all_data = data["items"]
    try:
        contentDetails = all_data[0]["contentDetails"]
        duration = contentDetails["duration"]
        duration = isodate.parse_duration(duration)
        video_dur = duration.total_seconds()
        return video_dur
    except:
        return 0


def run(year, file):
    results = dict()
    with open(file, encoding='utf-8') as f:
        data = json.load(f)
    for video in data:
        try:
            url = video["titleUrl"]
        except KeyError:
            pass  # lazy handling, but usually video was deleted

        if checkYear(video["time"], year):
            if "subtitles" not in video:
                pass
            else:
                youtuber = video["subtitles"][0]["name"]
                if youtuber in results:
                    results[youtuber] += getVideoLength(url)
                else:
                    results[youtuber] = getVideoLength(url)
    top = sorted(results, key=results.get, reverse=True)
    for v in top:
      print(v, results[v])


if __name__ == "__main__":
    run(2020, "watch-history.json") # from google takeout