Archives/fabric
2
0
Fork 0
mirror of https://github.com/danielmiessler/fabric synced 2024-11-08 07:11:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #376 from obswork/obswork/yt_metadata

Browse Source 
feat: add metadata flag to yt cli
This commit is contained in:
Daniel Miessler 2024-05-10 20:14:22 -07:00 committed by GitHub
commit d9d46bd662
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
installer/client/cli/yt.py
View File

@ -3,6 +3,7 @@ from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
from youtube_transcript_api import YouTubeTranscriptApi
from dotenv import load_dotenv
from datetime import datetime
import os
import json
import isodate
@ -79,12 +80,18 @@ def main_function(url, options):
# Get video details
video_response = youtube.videos().list(
id=video_id, part="contentDetails").execute()
id=video_id, part="contentDetails,snippet").execute()
# Extract video duration and convert to minutes
duration_iso = video_response["items"][0]["contentDetails"]["duration"]
duration_seconds = isodate.parse_duration(duration_iso).total_seconds()
duration_minutes = round(duration_seconds / 60)
# Set up metadata
metadata = {}
metadata['id'] = video_response['items'][0]['id']
metadata['title'] = video_response['items'][0]['snippet']['title']
metadata['channel'] = video_response['items'][0]['snippet']['channelTitle']
metadata['published_at'] = video_response['items'][0]['snippet']['publishedAt']
# Get video transcript
try:
@ -106,12 +113,15 @@ def main_function(url, options):
print(transcript_text.encode('utf-8').decode('unicode-escape'))
elif options.comments:
print(json.dumps(comments, indent=2))
elif options.metadata:
print(json.dumps(metadata, indent=2))
else:
# Create JSON object with all data
output = {
"transcript": transcript_text,
"duration": duration_minutes,
"comments": comments
"comments": comments,
"metadata": metadata
}
# Print JSON object
print(json.dumps(output, indent=2))
@ -126,6 +136,7 @@ def main():
parser.add_argument('--duration', action='store_true', help='Output only the duration')
parser.add_argument('--transcript', action='store_true', help='Output only the transcript')
parser.add_argument('--comments', action='store_true', help='Output the comments on the video')
parser.add_argument('--metadata', action='store_true', help='Output the video metadata')
parser.add_argument('--lang', default='en', help='Language for the transcript (default: English)')
args = parser.parse_args()