mirror of
https://github.com/TheFunny/TelegramTwitterMediaBot.git
synced 2026-09-28 00:02:12 +00:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a38c1b05b4
|
||
|
|
b3c9b1ace7
|
||
|
|
6a7ab11148
|
@@ -14,7 +14,7 @@ if WEBHOOK:
|
|||||||
WEBHOOK_CERT = os.getenv("WEBHOOK_CERT", "cert/cert.pem")
|
WEBHOOK_CERT = os.getenv("WEBHOOK_CERT", "cert/cert.pem")
|
||||||
WEBHOOK_SECRET_TOKEN = os.getenv("WEBHOOK_SECRET_TOKEN")
|
WEBHOOK_SECRET_TOKEN = os.getenv("WEBHOOK_SECRET_TOKEN")
|
||||||
|
|
||||||
x_url_regex = re.compile(r"^(?:https?://)(?:www\.|mobile\.|)(?:x|twitter)\.com/.+/status/(\d+)")
|
x_url_regex = re.compile(r"^(?:https?://)(?:www\.|mobile\.|)(?:x|twitter)\.com/(.+)/status/(\d+)")
|
||||||
x_media_regex = re.compile(r"^(?:https?://)(pbs|video)\.twimg\.com/(.*)")
|
x_media_regex = re.compile(r"^(?:https?://)(pbs|video)\.twimg\.com/(.*)")
|
||||||
x_tco_regex = re.compile(r"(?:https?://)t\.co/.+$", re.M)
|
x_tco_regex = re.compile(r"(?:https?://)t\.co/.+$", re.M)
|
||||||
|
|
||||||
|
|||||||
@@ -9,7 +9,8 @@ from telegram.ext import (
|
|||||||
filters,
|
filters,
|
||||||
InlineQueryHandler,
|
InlineQueryHandler,
|
||||||
PicklePersistence,
|
PicklePersistence,
|
||||||
MessageHandler, CommandHandler
|
MessageHandler,
|
||||||
|
CommandHandler
|
||||||
)
|
)
|
||||||
|
|
||||||
import common
|
import common
|
||||||
@@ -32,7 +33,8 @@ async def reply_media(update: Update, context: ContextTypes.DEFAULT_TYPE) -> Non
|
|||||||
async with TGTweet(context.bot_data['client'], url) as tweet:
|
async with TGTweet(context.bot_data['client'], url) as tweet:
|
||||||
media = list(tweet.pm_media_generator)
|
media = list(tweet.pm_media_generator)
|
||||||
message_sent = await update.effective_message.reply_media_group(
|
message_sent = await update.effective_message.reply_media_group(
|
||||||
media, caption=tweet.message_text,
|
media,
|
||||||
|
caption=tweet.message_text,
|
||||||
reply_to_message_id=update.message.message_id
|
reply_to_message_id=update.message.message_id
|
||||||
)
|
)
|
||||||
if 'forward_channel_id' in context.user_data:
|
if 'forward_channel_id' in context.user_data:
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ from telegram import (
|
|||||||
from common import x_url_regex, x_media_regex, x_tco_regex
|
from common import x_url_regex, x_media_regex, x_tco_regex
|
||||||
|
|
||||||
twimg_url = 'https://pbs.twimg.com/'
|
twimg_url = 'https://pbs.twimg.com/'
|
||||||
vx_api_url = 'https://api.vxtwitter.com/status/'
|
vx_api_url = 'https://api.vxtwitter.com/status/{0}/status/{1}'
|
||||||
|
|
||||||
message_raw_text = """{url}
|
message_raw_text = """{url}
|
||||||
<a href="{author_url}">{author}</a>: {text}
|
<a href="{author_url}">{author}</a>: {text}
|
||||||
@@ -108,25 +108,25 @@ class TGTweet(Tweet):
|
|||||||
def __init__(self, session: ClientSession, url: str):
|
def __init__(self, session: ClientSession, url: str):
|
||||||
self._session: ClientSession = session
|
self._session: ClientSession = session
|
||||||
self._url: str = url
|
self._url: str = url
|
||||||
self._id: str = self._tweet_id
|
self._api_param: tuple[str] = self._tweet_id
|
||||||
assert self._id
|
assert self._api_param
|
||||||
|
|
||||||
async def __aenter__(self):
|
async def __aenter__(self):
|
||||||
self._tweet: dict = await self._fetch_tweet(self._id)
|
self._tweet: dict = await self._fetch_tweet(self._api_param)
|
||||||
super().__init__(*self._init_properties)
|
super().__init__(*self._init_properties)
|
||||||
return self
|
return self
|
||||||
|
|
||||||
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
async def _fetch_tweet(self, tweet_id: str) -> dict:
|
async def _fetch_tweet(self, api_param: tuple[str]) -> dict:
|
||||||
return await fetch_json(self._session, vx_api_url + tweet_id)
|
return await fetch_json(self._session, vx_api_url.format(*api_param))
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _tweet_id(self) -> str | None:
|
def _tweet_id(self) -> tuple[str] | None:
|
||||||
match = x_url_regex.match(self._url)
|
match = x_url_regex.match(self._url)
|
||||||
if match:
|
if match:
|
||||||
return match.group(1).strip('/')
|
return match.groups()
|
||||||
return None
|
return None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
@@ -147,12 +147,13 @@ class TGTweet(Tweet):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def _init_properties(self) -> tuple:
|
def _init_properties(self) -> tuple:
|
||||||
|
id = self._tweet['tweetID']
|
||||||
author = self._tweet['user_name']
|
author = self._tweet['user_name']
|
||||||
author_id = self._tweet['user_screen_name']
|
author_id = self._tweet['user_screen_name']
|
||||||
text = self._tweet_text
|
text = self._tweet_text
|
||||||
media = self._tweet_media
|
media = self._tweet_media
|
||||||
sensitive = self._tweet['possibly_sensitive']
|
sensitive = self._tweet['possibly_sensitive']
|
||||||
return self._id, author, author_id, text, media, sensitive
|
return id, author, author_id, text, media, sensitive
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def message_text(self) -> str:
|
def message_text(self) -> str:
|
||||||
@@ -168,7 +169,7 @@ class TGTweet(Tweet):
|
|||||||
for i, tweet_media in enumerate(self.media):
|
for i, tweet_media in enumerate(self.media):
|
||||||
if tweet_media.type == "image":
|
if tweet_media.type == "image":
|
||||||
yield InlineQueryResultPhoto(
|
yield InlineQueryResultPhoto(
|
||||||
id=self.id,
|
id=str(i),
|
||||||
photo_url=tweet_media.url,
|
photo_url=tweet_media.url,
|
||||||
thumbnail_url=tweet_media.thumb,
|
thumbnail_url=tweet_media.thumb,
|
||||||
title=self.url,
|
title=self.url,
|
||||||
@@ -177,7 +178,7 @@ class TGTweet(Tweet):
|
|||||||
)
|
)
|
||||||
elif tweet_media.type == "video":
|
elif tweet_media.type == "video":
|
||||||
yield InlineQueryResultVideo(
|
yield InlineQueryResultVideo(
|
||||||
id=self.id,
|
id=str(i),
|
||||||
video_url=tweet_media.url,
|
video_url=tweet_media.url,
|
||||||
mime_type="video/mp4",
|
mime_type="video/mp4",
|
||||||
thumbnail_url=tweet_media.thumb,
|
thumbnail_url=tweet_media.thumb,
|
||||||
|
|||||||
Reference in New Issue
Block a user