123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203 |
- import datetime as dt
- import functools
- import json
- import urllib.parse
- from .common import InfoExtractor
- from ..utils import (
- ExtractorError,
- float_or_none,
- int_or_none,
- make_archive_id,
- mimetype2ext,
- str_or_none,
- urljoin,
- )
- from ..utils.traversal import traverse_obj
- class Pr0grammIE(InfoExtractor):
- _VALID_URL = r'https?://pr0gramm\.com\/(?:[^/?#]+/)+(?P<id>[\d]+)(?:[/?#:]|$)'
- _TESTS = [{
- 'url': 'https://pr0gramm.com/new/video/5466437',
- 'info_dict': {
- 'id': '5466437',
- 'ext': 'mp4',
- 'title': 'pr0gramm-5466437 by g11st',
- 'tags': ['Neon Genesis Evangelion', 'Touhou Project', 'Fly me to the Moon', 'Marisad', 'Marisa Kirisame', 'video', 'sound', 'Marisa', 'Anime'],
- 'uploader': 'g11st',
- 'uploader_id': '394718',
- 'timestamp': 1671590240,
- 'upload_date': '20221221',
- 'like_count': int,
- 'dislike_count': int,
- 'age_limit': 0,
- 'thumbnail': r're:^https://thumb\.pr0gramm\.com/.*\.jpg',
- '_old_archive_ids': ['pr0grammstatic 5466437'],
- },
- }, {
- 'url': 'https://pr0gramm.com/new/3052805:comment28391322',
- 'info_dict': {
- 'id': '3052805',
- 'ext': 'mp4',
- 'title': 'pr0gramm-3052805 by Hansking1',
- 'tags': 'count:15',
- 'uploader': 'Hansking1',
- 'uploader_id': '385563',
- 'timestamp': 1552930408,
- 'upload_date': '20190318',
- 'like_count': int,
- 'dislike_count': int,
- 'age_limit': 0,
- 'thumbnail': r're:^https://thumb\.pr0gramm\.com/.*\.jpg',
- '_old_archive_ids': ['pr0grammstatic 3052805'],
- },
- }, {
- # Requires verified account
- 'url': 'https://pr0gramm.com/new/Gianna%20Michaels/5848332',
- 'info_dict': {
- 'id': '5848332',
- 'ext': 'mp4',
- 'title': 'pr0gramm-5848332 by erd0pfel',
- 'tags': 'count:18',
- 'uploader': 'erd0pfel',
- 'uploader_id': '349094',
- 'timestamp': 1694489652,
- 'upload_date': '20230912',
- 'like_count': int,
- 'dislike_count': int,
- 'age_limit': 18,
- 'thumbnail': r're:^https://thumb\.pr0gramm\.com/.*\.jpg',
- '_old_archive_ids': ['pr0grammstatic 5848332'],
- },
- }, {
- 'url': 'https://pr0gramm.com/top/5895149',
- 'info_dict': {
- 'id': '5895149',
- 'ext': 'mp4',
- 'title': 'pr0gramm-5895149 by algoholigSeeManThrower',
- 'tags': 'count:19',
- 'uploader': 'algoholigSeeManThrower',
- 'uploader_id': '457556',
- 'timestamp': 1697580902,
- 'upload_date': '20231018',
- 'like_count': int,
- 'dislike_count': int,
- 'age_limit': 0,
- 'thumbnail': 'https://thumb.pr0gramm.com/2023/10/18/db47bb3db5e1a1b3.jpg',
- '_old_archive_ids': ['pr0grammstatic 5895149'],
- },
- }, {
- 'url': 'https://pr0gramm.com/static/5466437',
- 'only_matching': True,
- }, {
- 'url': 'https://pr0gramm.com/new/rowan%20atkinson%20herr%20bohne/3052805',
- 'only_matching': True,
- }, {
- 'url': 'https://pr0gramm.com/user/froschler/dafur-ist-man-hier/5091290',
- 'only_matching': True,
- }]
- BASE_URL = 'https://pr0gramm.com'
- @functools.cached_property
- def _is_logged_in(self):
- return 'pp' in self._get_cookies(self.BASE_URL)
- @functools.cached_property
- def _maximum_flags(self):
- # We need to guess the flags for the content otherwise the api will raise an error
- # We can guess the maximum allowed flags for the account from the cookies
- # Bitflags are (msbf): pol, nsfp, nsfl, nsfw, sfw
- flags = 0b10001
- if self._is_logged_in:
- flags |= 0b01000
- cookies = self._get_cookies(self.BASE_URL)
- if 'me' not in cookies:
- self._download_webpage(self.BASE_URL, None, 'Refreshing verification information')
- if traverse_obj(cookies, ('me', {lambda x: x.value}, {urllib.parse.unquote}, {json.loads}, 'verified')):
- flags |= 0b00110
- return flags
- def _call_api(self, endpoint, video_id, query={}, note='Downloading API json'):
- data = self._download_json(
- f'https://pr0gramm.com/api/items/{endpoint}',
- video_id, note, query=query, expected_status=403)
- error = traverse_obj(data, ('error', {str}))
- if error in ('nsfwRequired', 'nsflRequired', 'nsfpRequired', 'verificationRequired'):
- if not self._is_logged_in:
- self.raise_login_required()
- raise ExtractorError(f'Unverified account cannot access NSFW/NSFL ({error})', expected=True)
- elif error:
- message = traverse_obj(data, ('msg', {str})) or error
- raise ExtractorError(f'API returned error: {message}', expected=True)
- return data
- @staticmethod
- def _create_source_url(path):
- return urljoin('https://img.pr0gramm.com', path)
- def _real_extract(self, url):
- video_id = self._match_id(url)
- video_info = traverse_obj(
- self._call_api('get', video_id, {'id': video_id, 'flags': self._maximum_flags}),
- ('items', 0, {dict}))
- source = video_info.get('image')
- if not source or not source.endswith('mp4'):
- self.raise_no_formats('Could not extract a video', expected=bool(source), video_id=video_id)
- metadata = self._call_api('info', video_id, {'itemId': video_id}, note='Downloading tags')
- tags = traverse_obj(metadata, ('tags', ..., 'tag', {str}))
- # Sorted by "confidence", higher confidence = earlier in list
- confidences = traverse_obj(metadata, ('tags', ..., 'confidence', ({int}, {float})))
- if confidences:
- tags = [tag for _, tag in sorted(zip(confidences, tags), reverse=True)]
- formats = traverse_obj(video_info, ('variants', ..., {
- 'format_id': ('name', {str}),
- 'url': ('path', {self._create_source_url}),
- 'ext': ('mimeType', {mimetype2ext}),
- 'vcodec': ('codec', {str}),
- 'width': ('width', {int_or_none}),
- 'height': ('height', {int_or_none}),
- 'bitrate': ('bitRate', {float_or_none}),
- 'filesize': ('fileSize', {int_or_none}),
- })) if video_info.get('variants') else [{
- 'ext': 'mp4',
- 'format_id': 'source',
- **traverse_obj(video_info, {
- 'url': ('image', {self._create_source_url}),
- 'width': ('width', {int_or_none}),
- 'height': ('height', {int_or_none}),
- }),
- }]
- subtitles = {}
- for subtitle in traverse_obj(video_info, ('subtitles', lambda _, v: v['language'])):
- subtitles.setdefault(subtitle['language'], []).append(traverse_obj(subtitle, {
- 'url': ('path', {self._create_source_url}),
- 'note': ('label', {str}),
- }))
- return {
- 'id': video_id,
- 'title': f'pr0gramm-{video_id} by {video_info.get("user")}',
- 'tags': tags,
- 'formats': formats,
- 'subtitles': subtitles,
- 'age_limit': 18 if traverse_obj(video_info, ('flags', {0b110.__and__})) else 0,
- '_old_archive_ids': [make_archive_id('Pr0grammStatic', video_id)],
- **traverse_obj(video_info, {
- 'uploader': ('user', {str}),
- 'uploader_id': ('userId', {str_or_none}),
- 'like_count': ('up', {int}),
- 'dislike_count': ('down', {int}),
- 'timestamp': ('created', {int}),
- 'upload_date': ('created', {int}, {dt.date.fromtimestamp}, {lambda x: x.strftime('%Y%m%d')}),
- 'thumbnail': ('thumb', {lambda x: urljoin('https://thumb.pr0gramm.com', x)}),
- }),
- }
|