discord_emoji_downloader.py

# coding: UTF-8
# title: Discord 서버 커스텀 이모지 다운로드
# author: SaidBySolo

"""
MIT License

Copyright (c) 2020 SaidBySolo

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
"""
from utils import Downloader, clean_title
import requests


@Downloader.register
class DownloaderDiscordEmoji(Downloader):
    type = "discord"

    def init(self):
        pass

    def read(self):
        token_guild_id_list = self.url.replace("discord_", "", 1).split(
            "/"
        )  # 값을 어떻게 받을지 몰라서 일단 나눴어요. discord_이메일/비밀번호/서버아이디 또는 discord_토큰/서버아이디 이런식으로 받게 해놨어요.

        if len(token_guild_id_list) == 2:
            token = token_guild_id_list[0]
            guild_id = token_guild_id_list[1]
        elif len(token_guild_id_list) == 3:
            email = token_guild_id_list[0]
            password = token_guild_id_list[1]
            guild_id = token_guild_id_list[2]

            response = self.post_account_info(email, password)
            account_info = response.json()
            if response.status_code == 400:
                if account_info.get("captcha_key"):
                    return self.Invalid(
                        "먼저 웹 또는 디스코드 앱에서 로그인하신후 캡차를 인증해주세요."
                    )  # 메세지 박스 return하니까 멈춰서 raise로 해놨어요
                else:
                    return self.Invalid("이메일 또는 비밀번호가 잘못되었습니다. 확인후 다시 시도해주세요.")
            else:
                if not account_info["token"]:
                    return self.Invalid("토큰을 받아오지 못했어요. 2단계인증을 사용중이신경우 토큰을 이용해 요청해주세요.")
                else:
                    token = account_info["token"]
        else:
            return self.Invalid("인자값이 더 많이왔어요.")

        guild_info_response = self.get_emoji_list(token, int(guild_id))  # 토큰과 함께 get요청함

        if guild_info_response.status_code != 200:
            return self.Invalid("정상적인 토큰이 아니거나 서버를 찾을수없어요. 맞는 토큰인지, 해당 서버에 접속해있는지 확인해주세요.")
        else:
            guild_info = guild_info_response.json()

        if guild_info["emojis"]:
            base_url = "https://cdn.discordapp.com/emojis/"
            for emoji in guild_info["emojis"]:  # 이모지 리스트로 가져옴
                if emoji["animated"] is True:  # 만약 gif면 gif 다운로드
                    param = emoji["id"] + ".gif"
                else:  # 아닐경우 png로
                    param = emoji["id"] + ".png"

                self.title = clean_title(
                    f'{guild_info["name"]}({guild_info["id"]})'  # 폴더 이름은 서버 이름, id
                )
                self.urls.append(base_url + param + "?v=1")  # 인자 합치기
        else:
            return self.Invalid("해당 서버에는 이모지가 없어요")

    def get_emoji_list(self, token: str, guild_id: int) -> dict:
        response = requests.get(
            f"https://discordapp.com/api/v6/guilds/{guild_id}",
            headers={"Authorization": token},
        )
        if response.status_code == 401:
            response = requests.get(
                f"https://discordapp.com/api/v6/guilds/{guild_id}",
                headers={"Authorization": f"Bot {token}"},
            )

        return response

    def post_account_info(self, email: str, password: str) -> dict:
        response = requests.post(
            "https://discordapp.com/api/v8/auth/login",
            json={
                "email": email,
                "password": password,
                "undelete": False,
                "captcha_key": None,
                "login_source": None,
                "gift_code_sku_id": None,
            },
        )

        return response