Init
This commit is contained in:
parent
899d5792d6
commit
0f6b756667
17
.vscode/launch.json
vendored
Normal file
17
.vscode/launch.json
vendored
Normal file
@ -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
|
||||
}
|
||||
]
|
||||
}
|
8
.vscode/settings.json
vendored
Normal file
8
.vscode/settings.json
vendored
Normal file
@ -0,0 +1,8 @@
|
||||
{
|
||||
"[python]": {
|
||||
"editor.defaultFormatter": "ms-python.black-formatter"
|
||||
},
|
||||
"python.formatting.provider": "none",
|
||||
"python.analysis.autoImportCompletions": true,
|
||||
"python.analysis.typeCheckingMode": "basic"
|
||||
}
|
83
main.py
Normal file
83
main.py
Normal file
@ -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)
|
20
pixiv.py
Normal file
20
pixiv.py
Normal file
@ -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 !")
|
2
requirements.txt
Normal file
2
requirements.txt
Normal file
@ -0,0 +1,2 @@
|
||||
pixivpy3
|
||||
discord.py
|
Loading…
Reference in New Issue
Block a user