56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import math
|
|
import os
|
|
import re
|
|
from shlex import join
|
|
from tempfile import TemporaryDirectory
|
|
import discord
|
|
import pixiv
|
|
import send_message
|
|
|
|
REFRESH_TOKEN = os.getenv("PIXIV_REFRESH_TOKEN")
|
|
if REFRESH_TOKEN is None:
|
|
print("Pixiv refresh token is missing, please set PIXIV_REFRESH_TOKEN")
|
|
exit(1)
|
|
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
|
|
if DISCORD_TOKEN is None:
|
|
print("Discord token is missing, please set DISCORD_TOKEN")
|
|
exit(1)
|
|
|
|
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
|
|
client = discord.Client(intents=intents)
|
|
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
print(f"We have logged in as {client.user}")
|
|
|
|
|
|
@client.event
|
|
async def on_message(message: discord.Message):
|
|
if message.author == client.user:
|
|
return
|
|
|
|
if (
|
|
match := re.search(
|
|
r"(?P<url>https?://(?:www\.)?pixiv.net/\w+/artworks/(?P<id>\d+))",
|
|
message.content,
|
|
)
|
|
) is not None:
|
|
with TemporaryDirectory() as tmp_dir:
|
|
async with message.channel.typing():
|
|
title = pixiv.dowload_pixiv_images(
|
|
int(match.group("id")), tmp_dir, REFRESH_TOKEN
|
|
)
|
|
files = os.listdir(tmp_dir)
|
|
files.reverse()
|
|
await send_message.send_message_with_embed(
|
|
message, files, title, tmp_dir, match.group("url")
|
|
)
|
|
await send_message.send_message(message, files, tmp_dir)
|
|
|
|
|
|
client.run(DISCORD_TOKEN)
|