mirror of
https://github.com/artiemis/artemis.git
synced 2026-02-14 08:31:55 +00:00
271 lines
9.1 KiB
Python
271 lines
9.1 KiB
Python
from __future__ import annotations
|
|
from functools import cached_property
|
|
|
|
import html
|
|
import random
|
|
import re
|
|
from typing import Any, Literal, Optional
|
|
|
|
import discord
|
|
import pendulum
|
|
from aiohttp import ClientSession
|
|
from humanize import intcomma
|
|
from yt_dlp.utils import random_user_agent
|
|
|
|
import utils
|
|
from utils.common import ArtemisError
|
|
|
|
|
|
class Route:
|
|
method: str
|
|
path: str
|
|
url: str
|
|
|
|
BASE = "https://old.reddit.com"
|
|
|
|
def __init__(self, method, path):
|
|
self.method = method
|
|
self.path = path
|
|
self.url = self.BASE + self.path
|
|
|
|
|
|
class Reddit:
|
|
def __init__(self, session: ClientSession):
|
|
self.session: ClientSession = session
|
|
|
|
@staticmethod
|
|
def _gen_session_id() -> str:
|
|
id_length = 16
|
|
rand_max = 1 << (id_length * 4)
|
|
return "%0.*x" % (id_length, random.randrange(rand_max))
|
|
|
|
async def _request(self, route: Route, **kwargs) -> dict[str, Any]:
|
|
headers = {"User-Agent": random_user_agent()}
|
|
cookies = {
|
|
"reddit_session": self._gen_session_id(),
|
|
"_options": "%7B%22pref_quarantine_optin%22%3A%20true%7D",
|
|
}
|
|
async with self.session.request(
|
|
route.method, route.url, headers=headers, cookies=cookies, **kwargs
|
|
) as r:
|
|
data = await r.json()
|
|
return data
|
|
|
|
async def subreddit(
|
|
self,
|
|
name: str = "all",
|
|
sort: Literal["hot", "new"] = "hot",
|
|
include_stickied_and_pinned: bool = False,
|
|
) -> list[Post]:
|
|
route = Route("GET", f"/r/{name}/{sort}.json")
|
|
data = await self._request(route)
|
|
|
|
if data.get("reason"):
|
|
raise ArtemisError(f"This subreddit is inaccessible.\nReason: `{data['reason']}`")
|
|
if not data.get("data") or not data["data"]["children"]:
|
|
raise ArtemisError(f"Subreddit `{name}` not found.")
|
|
|
|
posts = [Post(post["data"]) for post in data["data"]["children"]]
|
|
|
|
if not include_stickied_and_pinned:
|
|
posts = [post for post in posts if not post.stickied and not post.pinned]
|
|
|
|
return posts
|
|
|
|
async def post(self, pid: str):
|
|
route = Route("GET", f"/{pid}.json?limit=1")
|
|
try:
|
|
data = await self._request(route)
|
|
post_data = data[0]["data"]["children"][0]["data"]
|
|
return Post(post_data)
|
|
except Exception:
|
|
return None
|
|
|
|
async def random(self, subreddit: str = "all", *args, **kwargs) -> Post:
|
|
posts = await self.subreddit(subreddit, *args, **kwargs)
|
|
return random.choice(posts)
|
|
|
|
async def random_image(self, subreddit: str = "all", *args, **kwargs) -> str:
|
|
posts = await self.subreddit(subreddit, *args, **kwargs)
|
|
images = [post for post in posts if post.image or post.gallery]
|
|
post = random.choice(images)
|
|
|
|
if post.gallery:
|
|
return post.gallery[0]
|
|
return post.image
|
|
|
|
|
|
class Post:
|
|
ICON = "https://www.redditstatic.com/desktop2x/img/favicon/android-icon-192x192.png"
|
|
GOLD_ICON = "https://www.redditstatic.com/gold/awards/icon/gold_64.png"
|
|
|
|
def __init__(self, data: dict):
|
|
self.data = data
|
|
|
|
self.title = html.unescape(self.data["title"])
|
|
self.body = self.data.get("selftext")
|
|
self.thumbnail = self.data.get("thumbnail", "")
|
|
|
|
self.over_18 = self.data.get("over_18")
|
|
self.stickied = self.data.get("stickied")
|
|
self.pinned = self.data.get("pinned")
|
|
self.spoiler = self.data.get("spoiler")
|
|
self.score = self.data.get("score", "N/A")
|
|
self.num_comments = self.data.get("num_comments", "N/A")
|
|
self.gilded = self.data.get("gilded")
|
|
self.awards = self.data.get("all_awardings")
|
|
|
|
self.permalink = "https://reddit.com" + self.data.get("permalink", "")
|
|
self.subreddit = self.data.get("subreddit")
|
|
self.subreddit_prefixed = f"r/{self.subreddit}"
|
|
self.created_at = pendulum.from_timestamp(self.data["created_utc"], "UTC")
|
|
|
|
@cached_property
|
|
def image(self) -> Optional[str]:
|
|
image = self.data.get("url_overridden_by_dest", "")
|
|
|
|
if not self.body:
|
|
if self.data.get("secure_media") or self.data.get("media_embed"):
|
|
return None
|
|
if re.search(r"(i\.redd\.it\/[^\/]+\.gifv)|gifv|webm", image):
|
|
return None
|
|
elif re.search(r"jpg|png|webp|gif|jfif|jpeg|imgur", image):
|
|
return image
|
|
return None
|
|
|
|
@cached_property
|
|
def video(self) -> Optional[str]:
|
|
media = self.data.get("media") or self.data.get("secure_media")
|
|
if not media:
|
|
return None
|
|
reddit_video = media.get("reddit_video")
|
|
if not reddit_video:
|
|
return None
|
|
playlist = reddit_video.get("dash_url") or reddit_video.get("hls_url")
|
|
if not playlist:
|
|
return None
|
|
return playlist
|
|
|
|
@cached_property
|
|
def preview(self) -> Optional[str]:
|
|
try:
|
|
preview = self.data["preview"]["images"][0]["source"]["url"]
|
|
return html.unescape(preview)
|
|
except Exception:
|
|
return None
|
|
|
|
@cached_property
|
|
def gallery(self) -> list[str]:
|
|
if not self.data.get("gallery_data") or not self.data.get("media_metadata"):
|
|
return []
|
|
|
|
images: list[str] = []
|
|
metadata = self.data["media_metadata"]
|
|
|
|
for image in self.data["gallery_data"]["items"]:
|
|
media_id = image["media_id"]
|
|
try:
|
|
url = html.unescape(metadata[media_id]["s"]["u"])
|
|
except Exception:
|
|
url = html.unescape(metadata[media_id]["s"]["gif"])
|
|
images.append(url)
|
|
|
|
return images
|
|
|
|
def get_warnings(self, nsfw: bool) -> str | None:
|
|
warnings = []
|
|
if self.spoiler:
|
|
warnings.append("SPOILER")
|
|
if nsfw:
|
|
warnings.append("NSFW")
|
|
if self.data.get("secure_media") or self.data.get("media_embed"):
|
|
warnings.append("UNSUPPORTED MEDIA")
|
|
|
|
if warnings:
|
|
return f"`❗ {', '.join(warnings)}`" # type: ignore
|
|
return None
|
|
|
|
def is_nsfw(self, message: discord.Message):
|
|
return self.over_18 and message.guild and not message.channel.is_nsfw()
|
|
|
|
async def to_embed(self, message: discord.Message) -> list[discord.Embed]:
|
|
COLOUR = discord.Colour(0xFF4500)
|
|
SPOILER_IMG_URL = "https://derpicdn.net/img/2016/5/22/1160541/medium.png"
|
|
NSFW_IMG_URL = "https://upload.wikimedia.org/wikipedia/commons/thumb/7/7a/Znaczek_TV_-_dozwolone_od_lat_18.svg/150px-Znaczek_TV_-_dozwolone_od_lat_18.svg.png"
|
|
|
|
files = []
|
|
icon_url = None
|
|
embed = discord.Embed(title=utils.trim(self.title, 256), url=self.permalink, colour=COLOUR)
|
|
embeds = [embed]
|
|
|
|
nsfw = self.is_nsfw(message)
|
|
|
|
if self.image:
|
|
embed.set_image(url=self.image)
|
|
|
|
if self.body:
|
|
body = html.unescape(utils.trim(self.body.strip(), 4096))
|
|
images = re.findall(r"https.*(?:png|jpg|jpeg|webp|gif)\S*", body)
|
|
for idx, url in enumerate(images[:10]):
|
|
if idx == 0:
|
|
embed.set_image(url=url)
|
|
else:
|
|
embeds.append(discord.Embed(color=COLOUR).set_image(url=url))
|
|
|
|
if self.spoiler or nsfw:
|
|
body = f"||{body}||"
|
|
|
|
embed.description = body
|
|
|
|
if self.gallery:
|
|
for idx, url in enumerate(self.gallery[:10]):
|
|
if idx == 0:
|
|
embed.set_image(url=url)
|
|
else:
|
|
embeds.append(discord.Embed(colour=COLOUR).set_image(url=url))
|
|
|
|
if not embed.image:
|
|
if self.preview:
|
|
embed.set_image(url=self.preview)
|
|
elif self.thumbnail and "http" in self.thumbnail:
|
|
embed.set_thumbnail(url=self.thumbnail)
|
|
|
|
if nsfw:
|
|
if embed.thumbnail:
|
|
embed.set_thumbnail(url=NSFW_IMG_URL)
|
|
elif embed.image:
|
|
for idx, embed in enumerate(embeds):
|
|
embed.set_image(url=NSFW_IMG_URL)
|
|
|
|
if self.spoiler:
|
|
if embed.image and not files:
|
|
for idx, embed in enumerate(embeds):
|
|
embed.set_image(url=SPOILER_IMG_URL)
|
|
|
|
warnings = self.get_warnings(nsfw)
|
|
if warnings:
|
|
if embed.description:
|
|
embed.description = f"{warnings}\n\n{embed.description}"
|
|
else:
|
|
embed.description = warnings
|
|
|
|
if self.gilded:
|
|
icon_url = self.GOLD_ICON
|
|
elif self.awards:
|
|
sorted_awards = sorted(self.awards, key=lambda x: int(x.get("count")), reverse=True)
|
|
icon_url = sorted_awards[0]["icon_url"]
|
|
|
|
upvotes = f"{intcomma(self.score)} upvote{'s' if self.score != 1 else ''}"
|
|
comments = f"{intcomma(self.num_comments)} comment{'s' if self.num_comments != 1 else ''}"
|
|
|
|
embed.set_author(
|
|
name=self.subreddit_prefixed,
|
|
icon_url=self.ICON,
|
|
url=f"https://reddit.com/r/{self.subreddit}",
|
|
)
|
|
|
|
embeds[-1].set_footer(text=f"{upvotes} and {comments}", icon_url=icon_url)
|
|
embeds[-1].timestamp = self.created_at
|
|
|
|
return embeds
|