Browse Source

yt comments includes reply threads. Readme updated.

pull/264/head
raisindetre 8 months ago
parent
commit
e6df0f93f0
  1. 7
      helper_files/README.md
  2. 41
      installer/client/cli/yt.py

7
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

41
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"))

Loading…
Cancel
Save