warp-beacon 2.6.10__tar.gz → 2.6.11__tar.gz

{warp_beacon-2.6.10/warp_beacon.egg-info → warp_beacon-2.6.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warp_beacon
-Version: 2.6.10
+Version: 2.6.11
 Summary: Telegram bot for expanding external media links
 Home-page: https://github.com/sb0y/warp_beacon
 Author: Andrey Bagrintsev

warp_beacon-2.6.11/warp_beacon/__version__.py

@@ -0,0 +1,2 @@
+__version__ = "2.6.11"
+

{warp_beacon-2.6.10 → warp_beacon-2.6.11}/warp_beacon/scraper/youtube/youtube.py

@@ -78,6 +78,9 @@ class YoutubeScraper(YoutubeAbstract):
 			
 			video_stream = yt.streams.filter(adaptive=True, file_extension='mp4', only_video=True).order_by('resolution').desc().first()
 			audio_stream = yt.streams.filter(adaptive=True, file_extension='mp4', only_audio=True).order_by('abr').desc().first()
+			size_in_bytes = video_stream.filesize + audio_stream.filesize
+			if size_in_bytes > 2147483648: # 2 GiB
+				video_stream = yt.streams.filter(adaptive=True, file_extension='mp4', only_video=True, res='720p').order_by('resolution').desc().first()
 
 			local_video_file = video_stream.download(
 				output_path=self.DOWNLOAD_DIR,
@@ -161,7 +164,30 @@ class YoutubeScraper(YoutubeAbstract):
 	def _download_yt_dlp(self, url: str, thumbnail: Optional[io.BytesIO] = None, timeout: int = 60) -> list:
 		res = []
 		with self.build_yt_dlp(timeout) as ydl:
-			info = ydl.extract_info(url, download=True)
+			info = ydl.extract_info(url, download=False)
+			formats = info.get('formats', [])
+
+			dl_format = {}
+			for f in sorted(formats, key=lambda x: (x.get('height') or 0), reverse=True):
+				if f.get('vcodec') != 'none' and f.get('acodec') != 'none' and f.get('ext') == 'mp4':
+					dl_format = f
+					break
+		
+			filesize = dl_format.get('filesize', 0) or dl_format.get('filesize_approx', 0)
+			if filesize:
+				if filesize > 2147483648: # 2 GiB
+					logging.warning("Max resolution exceeding TG limits")
+					for f in sorted(formats, key=lambda x: (x.get('height') or 0), reverse=True):
+						if (f.get('vcodec') != 'none' and f.get('acodec') != 'none'
+								and (f.get('height') or 0) <= 720 and f.get('ext') == 'mp4'):
+							alt_filesize = f.get('filesize', 0) or f.get('filesize_approx', 0)
+							if alt_filesize: #and alt_filesize <= max_size_bytes:
+								dl_format = f
+								break
+			else:
+				logging.warning("Unknown filesize!")
+			ydl.params['format'] = dl_format['format_id']
+			ydl.download([url])
 			local_file = ydl.prepare_filename(info)
 			logging.debug("Temp filename: '%s'", local_file)
 			res.append({
@@ -172,4 +198,4 @@ class YoutubeScraper(YoutubeAbstract):
 				"media_type": JobType.VIDEO
 			})
 
-		return res
+		return res

{warp_beacon-2.6.10 → warp_beacon-2.6.11/warp_beacon.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warp_beacon
-Version: 2.6.10
+Version: 2.6.11
 Summary: Telegram bot for expanding external media links
 Home-page: https://github.com/sb0y/warp_beacon
 Author: Andrey Bagrintsev

warp_beacon-2.6.10/warp_beacon/__version__.py

@@ -1,2 +0,0 @@
-__version__ = "2.6.10"
-