check.py

#!/usr/bin/env python3
from __future__ import unicode_literals
import yt_dlp
import json
import sys

def outputFormatter(versions):
    toReturn = []
    for x in versions:
        toReturn.append({'ref': x})
    return json.dumps(toReturn)

class PlaylistProcessor(object):
    def __init__(self):
        self.vidlist = []

    def process_playlist_json(self, msg, format):
        playlist = json.loads(msg)
        ok = False
        for x in playlist['entries']:
            if not ok:
                # Want to make sure first entry is accessible
                if self.is_ok(x['id'], format):
                    ok = True
                else:
                    # This entry will be skipped because it is not accessible (yet)
                    continue
            self.vidlist.append(x['id'])
        return

    def count(self):
        return len(self.vidlist)

    def most_recent(self):
        return self.vidlist[0]

    def since_item(self, item):
        indexOfItem = self.vidlist.index(item)
        toReturn = self.vidlist[0:indexOfItem]
        toReturn.reverse()
        return toReturn

    def is_ok(self, item, format):
        if format == "":
            format = "137+140"
        ydl_opts = {
            'skip_download': True,
            'logtostderr': True,
            'ignoreerrors': True,
            'extract_flat': False,
            'in_playlist': False,
            'format': format
        }
        with yt_dlp.YoutubeDL(ydl_opts) as ydl2:
            ydl2.download([item])
            video_exit_code = ydl2._download_retcode
        return video_exit_code == 0

resource_config = json.load(sys.stdin)

print("Got Config: "+json.dumps(resource_config), file=sys.stderr)

ydl_output = PlaylistProcessor()

ydl_opts_playlist = {
    'skip_download': True,
    'logtostderr': True,
    'ignoreerrors': True, 
    'extract_flat': True,
    'in_playlist': True,
}

exit_code = 0

with yt_dlp.YoutubeDL(ydl_opts_playlist) as ydl:
    info = ydl.extract_info(resource_config['source']['playlist'])
    if ydl._download_retcode:
        exit_code = ydl._download_retcode
    ydl_output.process_playlist_json(json.dumps(ydl.sanitize_info(info)), resource_config['source']['format_id'])

try:
    if ('version' in resource_config) and ('ref' in resource_config['version']):
        #Getting newer
        currentVer = resource_config['version']['ref']
        print("Returning newer than: " +currentVer, file=sys.stderr)
        newerItems = ydl_output.since_item(currentVer)
        print(outputFormatter(newerItems))
    else:
        #Getting latest
        print("Was not provided a version, returning latest", file=sys.stderr)
        print(json.dumps([{'ref': ydl_output.most_recent()}]))
except TypeError:
    #Get latest
    print(outputFormatter([ydl_output.most_recent()]))

sys.exit(exit_code)