Init

.vscode/launch.json

@@ -0,0 +1,17 @@
+{
+  // Use IntelliSense to learn about possible attributes.
+  // Hover to view descriptions of existing attributes.
+  // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+  "version": "0.2.0",
+  "configurations": [
+    {
+      "name": "Python: Current File",
+      "type": "python",
+      "request": "launch",
+      "program": "main.py",
+      "envFile": "${workspaceFolder}/.env",
+      "console": "integratedTerminal",
+      "justMyCode": true
+    }
+  ]
+}

.vscode/settings.json

@@ -0,0 +1,8 @@
+{
+  "[python]": {
+    "editor.defaultFormatter": "ms-python.black-formatter"
+  },
+  "python.formatting.provider": "none",
+  "python.analysis.autoImportCompletions": true,
+  "python.analysis.typeCheckingMode": "basic"
+}

main.py

@@ -0,0 +1,83 @@
+import math
+import os
+import re
+from shlex import join
+from tempfile import TemporaryDirectory
+import discord
+import pixiv
+
+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():
+                pixiv.dowload_pixiv_images(
+                    int(match.group("id")), tmp_dir, REFRESH_TOKEN
+                )
+                files = os.listdir(tmp_dir)
+                to_send_files = []
+                to_send_embed = []
+                files.reverse()
+                for file in files:
+                    joined = os.path.join(tmp_dir, file)
+                    embed = discord.Embed(url=match.group("url"))
+                    embed.set_image(url=f"attachment://{file}")
+                    to_send_files.append(discord.File(joined, file))
+                    to_send_embed.append(embed)
+                if len(to_send_embed) > 4:
+                    parts = math.ceil(len(to_send_embed) / 4)
+                    i = 1
+                    while len(to_send_embed) > 4:
+                        splited_file = to_send_files[:4]
+                        to_send_files = to_send_files[4:]
+                        splited_embed = to_send_embed[:4]
+                        to_send_embed = to_send_embed[4:]
+                        for embed in splited_embed:
+                            embed.title = f"Part {i}/{parts}"
+                        if i == 1:
+                            await message.reply(
+                                files=splited_file, embeds=splited_embed
+                            )
+                        else:
+                            await message.channel.send(
+                                files=splited_file, embeds=splited_embed
+                            )
+
+                        i = i + 1
+                    for embed in to_send_embed:
+                        embed.title = f"Part {i}/{parts}"
+                    await message.channel.send(
+                        files=to_send_files, embeds=to_send_embed
+                    )
+
+
+client.run(DISCORD_TOKEN)

pixiv.py

@@ -0,0 +1,20 @@
+import pixivpy3
+
+
+def dowload_pixiv_images(
+    illust_id: int,
+    dest_folder: str,
+    refresh_token: str,
+):
+    api = pixivpy3.AppPixivAPI()
+    api.auth(refresh_token=refresh_token)
+    # get origin url
+    json_result = api.illust_detail(illust_id)
+    illust = json_result.illust
+    pages = illust["meta_pages"]
+    print(f"Dowloading image to {dest_folder}")
+    for page in pages:
+        url = page["image_urls"]["original"]
+        print(url)
+        api.download(url=url, path=dest_folder)
+    print("Dowload finished !")

requirements.txt

@@ -0,0 +1,2 @@
+pixivpy3
+discord.py