diff --git a/helper_files/README.md b/helper_files/README.md index e8772f9..48cc87e 100644 --- a/helper_files/README.md +++ b/helper_files/README.md @@ -4,7 +4,7 @@ These are helper tools to work with Fabric. Examples include things like getting ## yt (YouTube) -`yt` is a command that uses the YouTube API to pull transcripts, get video duration, and other functions. It's primary function is to get a transcript from a video that can then be stitched (piped) into other Fabric Patterns. +`yt` is a command that uses the YouTube API to pull transcripts, pull user comments, get video duration, and other functions. It's primary function is to get a transcript from a video that can then be stitched (piped) into other Fabric Patterns. ```bash usage: yt [-h] [--duration] [--transcript] [url] @@ -15,9 +15,10 @@ positional arguments: url YouTube video URL options: - -h, --help show this help message and exit + -h, --help Show this help message and exit --duration Output only the duration --transcript Output only the transcript + --comments Output only the user comments ``` ## ts (Audio transcriptions) @@ -49,7 +50,7 @@ positional arguments: options: -h, --help show this help message and exit - +```` ## save `save` is a "tee-like" utility to pipeline saving of content, while keeping the output stream intact. Can optionally generate "frontmatter" for PKM utilities like Obsidian via the diff --git a/installer/client/cli/yt.py b/installer/client/cli/yt.py index 28a929d..b0449a0 100644 --- a/installer/client/cli/yt.py +++ b/installer/client/cli/yt.py @@ -18,36 +18,45 @@ def get_video_id(url): def get_comments(youtube, video_id): - # Fetch comments for the video comments = [] + try: - response = youtube.commentThreads().list( - part="snippet", + # Fetch top-level comments + request = youtube.commentThreads().list( + part="snippet,replies", videoId=video_id, textFormat="plainText", maxResults=100 # Adjust based on needs - ).execute() + ) - while response: + while request: + response = request.execute() for item in response['items']: - comment = item['snippet']['topLevelComment']['snippet']['textDisplay'] - comments.append(comment) - + # Top-level comment + topLevelComment = item['snippet']['topLevelComment']['snippet']['textDisplay'] + comments.append(topLevelComment) + + # Check if there are replies in the thread + if 'replies' in item: + for reply in item['replies']['comments']: + replyText = reply['snippet']['textDisplay'] + # Add incremental spacing and a dash for replies + comments.append(" - " + replyText) + + # Prepare the next page of comments, if available if 'nextPageToken' in response: - response = youtube.commentThreads().list( - part="snippet", - videoId=video_id, - textFormat="plainText", - pageToken=response['nextPageToken'], - maxResults=100 # Adjust based on needs - ).execute() + request = youtube.commentThreads().list_next( + previous_request=request, previous_response=response) else: - break + request = None + except HttpError as e: print(f"Failed to fetch comments: {e}") + return comments + def main_function(url, options): # Load environment variables from .env file load_dotenv(os.path.expanduser("~/.config/fabric/.env"))