Scrape images, video, and post forwarding information for Telegram channel posts

snscrape/modules/telegram.py

@@ -9,7 +9,7 @@ import re
 import snscrape.base
 import typing
 import urllib.parse
-
+import base64
 
 _logger = logging.getLogger(__name__)
 _SINGLE_MEDIA_LINK_PATTERN = re.compile(r'^https://t\.me/[^/]+/\d+\?single$')
@@ -30,6 +30,9 @@ class TelegramPost(snscrape.base.Item):
 	date: datetime.datetime
 	content: str
 	outlinks: list
+	images: list
+	video: str
+	forwarded: str
 	linkPreview: typing.Optional[LinkPreview] = None
 
 	outlinksss = snscrape.base._DeprecatedProperty('outlinksss', lambda self: ' '.join(self.outlinks), 'outlinks')
@@ -90,17 +93,38 @@ class TelegramChannelScraper(snscrape.base.Scraper):
 				_logger.warning(f'Possibly incorrect URL: {rawUrl!r}')
 			url = rawUrl.replace('//t.me/', '//t.me/s/')
 			date = datetime.datetime.strptime(dateDiv.find('time', datetime = True)['datetime'].replace('-', '', 2).replace(':', ''), '%Y%m%dT%H%M%S%z')
+			images = []
+			video = None
+			forwarded = None
 			if (message := post.find('div', class_ = 'tgme_widget_message_text')):
-				content = message.text
+				content = message.get_text(separator="\n")
+
+				for video_tag in post.find_all('video'):
+					video = video_tag['src']	
+
+				if (forward_tag := post.find('a', class_ = 'tgme_widget_message_forwarded_from_name')):
+					forwarded = forward_tag['href'].split('t.me/')[1].split('/')[0]			
+
 				outlinks = []
 				for link in post.find_all('a'):
 					if any(x in link.parent.attrs.get('class', []) for x in ('tgme_widget_message_user', 'tgme_widget_message_author')):
 						# Author links at the top (avatar and name)
 						continue
 					if link['href'] == rawUrl or link['href'] == url:
+						style = link.attrs.get('style', '')
 						# Generic filter of links to the post itself, catches videos, photos, and the date link
-						continue
+						if style != '':
+							image = re.findall('url\(\'(.*?)\'\)', style)
+							if len(image) == 1:
+								images.append(image[0])
+							continue
 					if _SINGLE_MEDIA_LINK_PATTERN.match(link['href']):
+						style = link.attrs.get('style', '')
+						image = re.findall('url\(\'(.*?)\'\)', style)
+						if len(image) == 1:
+							images.append(image[0])
+							# resp = self._get(image[0])
+							# encoded_string = base64.b64encode(resp.content)
 						# Individual photo or video link
 						continue
 					href = urllib.parse.urljoin(pageUrl, link['href'])
@@ -109,6 +133,8 @@ class TelegramChannelScraper(snscrape.base.Scraper):
 			else:
 				content = None
 				outlinks = []
+				images = []
+				video = None
 			linkPreview = None
 			if (linkPreviewA := post.find('a', class_ = 'tgme_widget_message_link_preview')):
 				kwargs = {}
@@ -125,7 +151,7 @@ class TelegramChannelScraper(snscrape.base.Scraper):
 					else:
 						_logger.warning(f'Could not process link preview image on {url}')
 				linkPreview = LinkPreview(**kwargs)
-			yield TelegramPost(url = url, date = date, content = content, outlinks = outlinks, linkPreview = linkPreview)
+			yield TelegramPost(url = url, date = date, content = content, outlinks = outlinks, linkPreview = linkPreview, images = images, video = video, forwarded = forwarded)
 
 	def get_items(self):
 		r, soup = self._initial_page()