projects / yt-dlp.git / blame
| Commit | Line | Data |
|---|---|---|
| cb7cc448 BW |
1 | from .common import InfoExtractor |
| 2 | from ..utils import ( | |
| 3 | clean_html, | |
| 4 | format_field, | |
| 5 | int_or_none, | |
| 6 | strip_or_none, | |
| 7 | traverse_obj, | |
| 8 | unified_timestamp, | |
| 9 | ) | |
| 10 | ||
| 11 | ||
| 12 | class TruthIE(InfoExtractor): | |
| 13 | _VALID_URL = r'https?://truthsocial\.com/@[^/]+/posts/(?P<id>\d+)' | |
| 14 | _TESTS = [ | |
| 15 | { | |
| 16 | 'url': 'https://truthsocial.com/@realDonaldTrump/posts/108779000807761862', | |
| 17 | 'md5': '4a5fb1470c192e493d9efd6f19e514d3', | |
| 18 | 'info_dict': { | |
| 19 | 'id': '108779000807761862', | |
| 20 | 'ext': 'qt', | |
| 21 | 'title': 'Truth video #108779000807761862', | |
| 22 | 'description': None, | |
| 23 | 'timestamp': 1659835827, | |
| 24 | 'upload_date': '20220807', | |
| 25 | 'uploader': 'Donald J. Trump', | |
| 26 | 'uploader_id': 'realDonaldTrump', | |
| 27 | 'uploader_url': 'https://truthsocial.com/@realDonaldTrump', | |
| 28 | 'repost_count': int, | |
| 29 | 'comment_count': int, | |
| 30 | 'like_count': int, | |
| 31 | }, | |
| 32 | }, | |
| 33 | { | |
| 34 | 'url': 'https://truthsocial.com/@ProjectVeritasAction/posts/108618228543962049', | |
| 35 | 'md5': 'fd47ba68933f9dce27accc52275be9c3', | |
| 36 | 'info_dict': { | |
| 37 | 'id': '108618228543962049', | |
| 38 | 'ext': 'mp4', | |
| 39 | 'title': 'md5:debde7186cf83f60ff7b44dbb9444e35', | |
| 40 | 'description': 'md5:de2fc49045bf92bb8dc97e56503b150f', | |
| 41 | 'timestamp': 1657382637, | |
| 42 | 'upload_date': '20220709', | |
| 43 | 'uploader': 'Project Veritas Action', | |
| 44 | 'uploader_id': 'ProjectVeritasAction', | |
| 45 | 'uploader_url': 'https://truthsocial.com/@ProjectVeritasAction', | |
| 46 | 'repost_count': int, | |
| 47 | 'comment_count': int, | |
| 48 | 'like_count': int, | |
| 49 | }, | |
| 50 | }, | |
| 51 | ] | |
| 52 | ||
| 53 | def _real_extract(self, url): | |
| 54 | video_id = self._match_id(url) | |
| 55 | status = self._download_json(f'https://truthsocial.com/api/v1/statuses/{video_id}', video_id) | |
| 56 | uploader_id = strip_or_none(traverse_obj(status, ('account', 'username'))) | |
| 57 | return { | |
| 58 | 'id': video_id, | |
| 59 | 'url': status['media_attachments'][0]['url'], | |
| 60 | 'title': '', | |
| 61 | 'description': strip_or_none(clean_html(status.get('content'))) or None, | |
| 62 | 'timestamp': unified_timestamp(status.get('created_at')), | |
| 63 | 'uploader': strip_or_none(traverse_obj(status, ('account', 'display_name'))), | |
| 64 | 'uploader_id': uploader_id, | |
| 65 | 'uploader_url': format_field(uploader_id, None, 'https://truthsocial.com/@%s'), | |
| 66 | 'repost_count': int_or_none(status.get('reblogs_count')), | |
| 67 | 'like_count': int_or_none(status.get('favourites_count')), | |
| 68 | 'comment_count': int_or_none(status.get('replies_count')), | |
| 69 | } |