paradizelost/youtube-dl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Style fixes

Browse Source
This commit is contained in:
palewire 2022-08-08 11:39:35 -07:00
parent 13077c0c96
commit 1f5cc92752
No known key found for this signature in database
GPG Key ID: A5AD4A9AD42D69AB
1 changed files with 49 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
youtube_dl/extractor/truth.py
View File

@ -3,38 +3,43 @@ from __future__ import unicode_literals
from .common import InfoExtractor from .common import InfoExtractor
from ..utils import clean_html, int_or_none, unified_timestamp from ..utils import (
clean_html,
int_or_none,
unified_timestamp,
strip_or_none
)
class TruthIE(InfoExtractor): class TruthIE(InfoExtractor):
"""Extract videos from posts on Donald Trump's truthsocial.com.""" """Extract videos from posts on Donald Trump's truthsocial.com."""
_VALID_URL = r"https://truthsocial\.com/@[^/]+/posts/(?P<id>[0-9]+)" _VALID_URL = r'https://truthsocial\.com/@[^/]+/posts/(?P<id>[0-9]+)'
_TESTS = [ _TESTS = [
{ {
"url": "https://truthsocial.com/@realDonaldTrump/posts/108779000807761862", 'url': 'https://truthsocial.com/@realDonaldTrump/posts/108779000807761862',
"md5": "4a5fb1470c192e493d9efd6f19e514d3", 'md5': '4a5fb1470c192e493d9efd6f19e514d3',
"info_dict": { 'info_dict': {
"id": "108779000807761862", 'id': '108779000807761862',
"ext": "qt", 'ext': 'qt',
"title": "Untitled", 'title': '0d8691160c73d663',
"timestamp": 1659835827, 'timestamp': 1659835827,
"upload_date": "20220807", 'upload_date': '20220807',
"uploader": "Donald J. Trump", 'uploader': 'Donald J. Trump',
"uploader_id": "realDonaldTrump", 'uploader_id': 'realDonaldTrump',
}, },
}, },
{ {
"url": "https://truthsocial.com/@ProjectVeritasAction/posts/108618228543962049", 'url': 'https://truthsocial.com/@ProjectVeritasAction/posts/108618228543962049',
"md5": "fd47ba68933f9dce27accc52275be9c3", 'md5': 'fd47ba68933f9dce27accc52275be9c3',
"info_dict": { 'info_dict': {
"id": "108618228543962049", 'id': '108618228543962049',
"ext": "mp4", 'ext': 'mp4',
"title": "RETRACTO #368: Utah NPR Affiliate RETRACTS False Claim Live On Air Following Veritas' Reporting on Curtis Campaign “Nothing I ever do will suffice for these people. They are engaged in conspiracy theories. They are doing precisely the thing they project that I do. Which is they don’t believe in facts, they don’t believe in logic, and they don’t believe in rationality.” - James O’Keefe", 'title': """RETRACTO #368: Utah NPR Affiliate RETRACTS False Claim Live On Air Following Veritas' Reporting on Curtis Campaign \n“Nothing I ever do will suffice for these people. They are engaged in conspiracy theories. They are doing precisely the thing they project that I do. Which is they don’t believe in facts, they don’t believe in logic, and they don’t believe in rationality.” - James O’Keefe""",
"timestamp": 1657382637, 'timestamp': 1657382637,
"upload_date": "20220709", 'upload_date': '20220709',
"uploader": "Project Veritas Action", 'uploader': 'Project Veritas Action',
"uploader_id": "ProjectVeritasAction", 'uploader_id': 'ProjectVeritasAction',
}, },
}, },
] ]
@ -44,36 +49,33 @@ class TruthIE(InfoExtractor):
video_id = self._match_id(url) video_id = self._match_id(url)
# Get data from API # Get data from API
api_url = "https://truthsocial.com/api/v1/statuses/" + video_id api_url = 'https://truthsocial.com/api/v1/statuses/' + video_id
status = self._download_json(api_url, video_id) status = self._download_json(api_url, video_id)
# Pull out video # Pull out video
attachments = status["media_attachments"] url = status['media_attachments'][0]['url']
video = attachments[0]
url = video["url"]
# Pull out metadata # Pull out metadata
title = clean_html(status.get("content")).replace("\n", "") or "Untitled" title = strip_or_none(clean_html(status.get('content'))) or self._generic_title(url)
account = status.get("account") or {} timestamp = unified_timestamp(status.get('created_at'))
timestamp = unified_timestamp(status.get("created_at")) account = status.get('account') or {}
uploader = account.get("display_name") uploader = strip_or_none(account.get('display_name'))
uploader_id = account.get("username") uploader_id = strip_or_none(account.get('username'))
uploader_url = ( uploader_url = ('https://truthsocial.com/@' + uploader_id) if uploader_id else None
"https://truthsocial.com/@" + uploader_id if uploader_id else None repost_count = int_or_none(status.get('reblogs_count'))
) like_count = int_or_none(status.get('favourites_count'))
repost_count = int_or_none(status.get("reblogs_count")) comment_count = int_or_none(status.get('replies_count'))
like_count = int_or_none(status.get("favourites_count"))
comment_count = int_or_none(status.get("replies_count"))
# Return the stuff
return { return {
"id": video_id, 'id': video_id,
"url": url, 'url': url,
"title": title, 'title': title,
"timestamp": timestamp, 'timestamp': timestamp,
"uploader": uploader, 'uploader': uploader,
"uploader_id": uploader_id, 'uploader_id': uploader_id,
"uploader_url": uploader_url, 'uploader_url': uploader_url,
"repost_count": repost_count, 'repost_count': repost_count,
"like_count": like_count, 'like_count': like_count,
"comment_count": comment_count, 'comment_count': comment_count,
} }