add mock generation script

This commit is contained in:
lenny 2023-04-01 01:16:55 +02:00
parent cb16e0343c
commit 7a474e3692
No known key found for this signature in database
GPG Key ID: 68EA109BBE53E6E7

56
generate.py Normal file
View File

@ -0,0 +1,56 @@
import subprocess
import urllib.request
import typing
import json
import tempfile
import os
default_context = json.loads('''
{
"context":{
"client":{
"hl":"en",
"gl":"US",
"deviceMake":"",
"deviceModel":"",
"clientName":"WEB",
"clientVersion":"2.20230217.01.00"
}
}
}
''')
# URL of the script
url = "https://github.com/iv-org/youtube-utils/raw/master/scripts/yt-api-helper.sh"
temp_dir = tempfile.TemporaryDirectory()
script_path = os.path.join(temp_dir.name, "yt-api-helper.sh")
urllib.request.urlretrieve(url, script_path)
os.chmod(script_path, 0o755)
def run_yt_helper_script(*args: [str]) -> str:
output = subprocess.run([script_path, *args], check=True, stdout=subprocess.PIPE)
return output.stdout.decode()
def read_videos_metadata() -> typing.List[dict]:
with open("./video/_videos.json", "r") as file:
file_contents = file.read()
data = json.loads(file_contents)
return data["files"]
def write_string_to_file(filename: str, string: str):
with open(filename, 'w') as file:
file.write(string)
files = read_videos_metadata()
for file in files:
default_context.update({"videoId": file["video_id"]})
context = json.dumps(default_context)
video = json.loads(run_yt_helper_script("-c", "web", "-e", file['endpoint'], "--data", context))
del video['responseContext']
write_string_to_file("video/"+file["name"], json.dumps(video, indent=2))