add logger

This commit is contained in:
2024-03-22 22:18:43 +08:00
parent 76e0ffe58c
commit c3faecf640
2 changed files with 8 additions and 1 deletions
+6 -1
View File
@@ -8,7 +8,7 @@ from telegram import (
InputMediaVideo
)
from common import x_url_regex, x_media_regex, x_tco_regex
from common import x_url_regex, x_media_regex, x_tco_regex, logger
twimg_url = 'https://pbs.twimg.com/'
vx_api_url = 'https://api.vxtwitter.com/status/{0}/status/{1}'
@@ -19,6 +19,7 @@ message_raw_text = """{url}
async def fetch_json(session: ClientSession, url: str) -> dict:
logger.info(f"Fetching {url}")
async with session.get(url) as response:
assert response.status == 200, f"Failed to fetch {url}, status code {response.status}"
return await response.json()
@@ -30,6 +31,9 @@ class TweetMedia:
self._thumb: str = thumb
self._type: str = media_type
def __str__(self):
return f"url: {self.url}, thumb: {self.thumb}, type: {self.type}"
@property
def _uri(self) -> str | None:
match = x_media_regex.match(self._url)
@@ -114,6 +118,7 @@ class TGTweet(Tweet):
async def __aenter__(self):
self._tweet: dict = await self._fetch_tweet(self._api_param)
super().__init__(*self._init_properties)
logger.info(f"Media: {self.media}")
return self
async def __aexit__(self, exc_type, exc_val, exc_tb):