mirror of
https://github.com/iv-org/mocks.git
synced 2024-11-26 23:50:48 +00:00
add mock generation script
This commit is contained in:
parent
cb16e0343c
commit
7a474e3692
56
generate.py
Normal file
56
generate.py
Normal 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))
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user