mirror of
https://github.com/Xtao-Labs/iShotaBot.git
synced 2024-11-22 15:36:55 +00:00
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
from typing import Dict
|
|
|
|
from httpx import AsyncClient
|
|
|
|
from .model import FixTweet, FixTweetMedia, User
|
|
|
|
|
|
class FixTwitterError(Exception):
|
|
def __init__(self, status_code: int, message: str):
|
|
self.status_code = status_code
|
|
self.message = message
|
|
|
|
|
|
class FixTwitterClient:
|
|
def __init__(self):
|
|
self.client = AsyncClient(timeout=60.0)
|
|
self.url = "https://api.fxtwitter.com/"
|
|
|
|
@staticmethod
|
|
def gen_tweet(data: Dict) -> FixTweet:
|
|
tweet = FixTweet(**data)
|
|
if medias := data.get("media"):
|
|
if all_media := medias.get("all"):
|
|
tweet.medias = [FixTweetMedia(**i) for i in all_media]
|
|
if quote := data.get("quote"):
|
|
tweet.quoted = FixTwitterClient.gen_tweet(quote)
|
|
if retweet := data.get("retweet"):
|
|
tweet.retweeted = FixTwitterClient.gen_tweet(retweet)
|
|
return tweet
|
|
|
|
async def tweet_detail(self, tid: int) -> FixTweet:
|
|
url = f"{self.url}X/status/{tid}"
|
|
response = await self.client.get(url)
|
|
if response.status_code != 200:
|
|
raise FixTwitterError(response.status_code, response.text)
|
|
data = response.json()
|
|
if data.get("code", 200) != 200:
|
|
raise FixTwitterError(data.get("code"), data.get("message"))
|
|
return self.gen_tweet(data["tweet"])
|
|
|
|
async def user_by_screen_name(self, username: str) -> User:
|
|
url = f"{self.url}{username}"
|
|
response = await self.client.get(url)
|
|
if response.status_code != 200:
|
|
raise FixTwitterError(response.status_code, response.text)
|
|
data = response.json()
|
|
if data.get("code", 200) != 200:
|
|
raise FixTwitterError(data.get("code"), data.get("message"))
|
|
return User(**data["user"])
|
|
|
|
|
|
fix_twitter_client = FixTwitterClient()
|