holodex.py

#!/usr/bin/env python3
import requests
import sys
import argparse
from datetime import datetime

def main(argv):
    url = "https://holodex.net/api/v2/live?"
    req_body = url
    parser = argparse.ArgumentParser(description='fetch current live vtubers from holodex')
    parser.add_argument('--source',
                        help='where you want to watch the stream: holodex, youtube')
    parser.add_argument('--org',
                        help='filter by organization e.g Hololive, Nijisanji')
    parser.add_argument('-l', type=int,
                        help='limit the output by an integer')
    parser.add_argument('--key',
                        help='API key')
    parser.add_argument('--status',
                        help='status of the video: new,upcoming,live,past,missing')
    parser.add_argument('--topic',
                        help='topic of the stream (e.g game name, singing')
    args = parser.parse_args()
    r = requests.get(req_body,headers={'X-APIKEY': '%s' % args.key}, params = {'org': args.org, 'limit': args.l, 'status' : args.status, 'topic' : args.topic})
    data = r.json()
    if args.source == 'youtube':
        watch = "https://youtube.com/watch/"
    else:
        watch = "https://holodex.net/watch/"
    for stream in data:
        print('title: ' + stream['title'])
        try:
            print('topic: ' + stream['topic_id'])
        except KeyError:
            pass
        try:
            print('name: ' + stream['channel']['english_name'])
        except KeyError:
            print('name: ' + stream['channel']['name'])
        print('starts: ' + str(datetime.strptime(stream['available_at'], '%Y-%m-%dT%H:%M:%S.%fZ')))
        print('url: ' + watch+stream['id']+'\n')
        
if __name__ == "__main__":
   main(sys.argv[1:])