mirror of
https://github.com/Astatin3/vidscout.git
synced 2026-06-08 16:18:02 -06:00
Astatin3
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import os
|
|
import sys
|
|
import json
|
|
|
|
import requests
|
|
from pytubefix import YouTube
|
|
from pytubefix.cli import on_progress
|
|
|
|
TBAip = "https://www.thebluealliance.com/api/v3"
|
|
# Free TBA Key!
|
|
headers = {"User-Agent": "Mozilla/5.0", "X-TBA-Auth-Key": "fzQY0pv6qwfwuII5Xx2bmP57BBSuE0maxKailYlrI0e1EdfKCq6F3Th9FFDqpW7f"}
|
|
RES = '1080p'
|
|
|
|
if len(sys.argv) < 2:
|
|
sys.exit("Usage: downloader.py <event code>")
|
|
|
|
evcode = sys.argv[1]
|
|
|
|
event_matches = requests.get(TBAip + "/event/" + evcode + "/matches", headers=headers).json()
|
|
|
|
video_urls = [None for _ in range(len(event_matches))]
|
|
|
|
for match in event_matches:
|
|
if match["comp_level"] != "qm": continue
|
|
match_urls = []
|
|
for video_url in match["videos"]:
|
|
if video_url["type"] != "youtube": continue
|
|
match_urls.append(video_url["key"])
|
|
if len(match_urls) != 0:
|
|
video_urls[match["match_number"]] = match_urls
|
|
|
|
if not os.path.exists(evcode):
|
|
os.makedirs(evcode)
|
|
|
|
for i, match in enumerate(video_urls):
|
|
if match is None: continue
|
|
# print(match)
|
|
for url in match:
|
|
if os.path.exists(os.path.join(evcode, str(i)+".mp4")):
|
|
continue
|
|
yt = YouTube("https://youtube.com/watch?v="+url, on_progress_callback=on_progress)
|
|
if yt.author != "FIRSTRoboticsCompetition": continue
|
|
|
|
|
|
for idx, stream in enumerate(yt.streams):
|
|
if stream.resolution == RES:
|
|
break
|
|
|
|
print(f"Downloading match {i}, {stream.resolution}")
|
|
|
|
# print(yt.streams[idx])
|
|
yt.streams[idx].download(output_path = evcode, filename = str(i)+".mp4")
|
|
|
|
# ys = yt.streams.get_highest_resolution()
|
|
# print(yt.title)
|