main.py

import os
import discord
from util.functions import *
from informations.handtrap import Urara, Warasi
from informations.season5 import tier
from discord.ext import commands
from dotenv import load_dotenv

# load .env
load_dotenv()

# cludetype.io에서 토큰 값 받아오기
TOKEN = os.environ.get('TOKEN')
PREFIX = os.environ.get("PREFIX")
HOST_URL = os.environ.get('HOST_URL')

# Intents 명시
intents = discord.Intents.default()
intents.members = True

bot = commands.Bot(command_prefix=PREFIX, intents=intents)


@bot.event
async def on_ready():
    print(f'{bot.user.name}이 연결 되었습니다.')
    await bot.change_presence(status=discord.Status.online, activity=None)


@bot.command()
async def 안녕(ctx):
    await ctx.send('{}님, 반갑습니다.'.format(ctx.author.mention))


@bot.command()
async def 명령어(ctx):
    embed = discord.Embed(title="TG 하이퍼 라이브러리언 명렁어 목록", description="마스터 듀얼에 도움이 되는 명렁어들이 있습니다!", color=0x32A4FF)
    embed.set_thumbnail(url="https://uploads3.yugioh.com/card_images/3946/detail/5736.jpg?1385135416")
    embed.add_field(name="!안녕", value="TG 하이퍼 라이브러리언과 인사합니다", inline=False)
    embed.add_field(name="!티어덱", value="마스터듀얼 티어덱을 조회합니다.", inline=False)
    embed.add_field(name="!어드민", value="서버 어드민을 확인합니다.", inline=False)
    embed.add_field(name="!우라라 가이드", value="우라라 명령어에 대한 가이드를 확인합니다.", inline=False)
    embed.add_field(name="!와라시 가이드", value="와라시 명령어에 대한 가이드를 확인합니다.", inline=False)
    embed.add_field(name="!서치 [카드이름]", value="카드 이름을 입력하면 카드를 서치합니다.", inline=False)
    embed.set_footer(text="Summoned by 강형준#5876",
                     icon_url="https://uploads3.yugioh.com/card_images/3946/detail/5736.jpg?1385135416")
    await ctx.send(embed=embed)


@bot.command('서치')
async def search(ctx, *args):
    url = HOST_URL + '/card/'
    param = makeStringWithSpace(*args)
    res = get(url, param)
    json_object = res.json()

    if json_object.get('status') == 404:
        string = "해당하는 데이터를 찾지 못했습니다."
    else:
        string = ''
        left_list = ["이름", "레벨", "종족", "공격력", "방어력", "속성", "효과"]
        right_list = ["card_name", "card_level", "card_species", "card_atk", "card_def", "card_attribute", "card_effect"]
        for i in range(len(left_list)):
            string += addContent(left_list[i], json_object.get(right_list[i]))

    await ctx.send(backtick(string))

# 우라라 리스트 명령어
@bot.command()
async def 와라시(ctx, arg):
    string = ''
    if arg == '가이드':
        string += '!와라시 [조회하고 싶은 덱] : 와라시 타이밍을 조회, ex)!와라시 드라이트론\n' \
                  '!와라시 리스트 : 와라시를 사용할 수 있는 목록을 불러옵니다.'

    elif arg == '리스트':
        for k in Warasi.keys():
            string += k + ' '

    else:
        if not arg in list(Warasi.keys()):
            string += '다시 입력해 주세요!'
        else:
            Warasi_list = Warasi.get(arg)
            # 와라시는 순위가 없음
            for i in range(len(Warasi_list)):
                string += Warasi_list[i] + '\n'

    await ctx.send(backtick(string))


@bot.command()
async def 어드민(ctx):
    await ctx.send(backtick("서버의 어드민은 {} 입니다.".format(ctx.guild.owner)))


@bot.command()
async def 우라라(ctx, arg):
    if arg == '가이드':
        string = '!우라라 [조회하고 싶은 덱]' + ': 우라라 타이밍을 조회, ex)!우라라 스파이랄\n' +\
                 '!우라라 주의사항' + ': 하루 우라라를 사용할 때의 주의 사항\n' +\
                 '!우라라 리스트' + ': 우라라를 사용할 수 있는 목록을 불러옵니다.\n' \
                 '해당 명령어는 \'마스터듀얼 갤러리\'의 \'마스터듀얼정보글\' 님의 게시글을 참고했습니다!'

    elif arg == '주의사항':
        string = '여기 적혀져 있는 순위는 모든 상황에 국한된 것이 아닙니다!\n' \
                 '상대 덱의 용병 카드나 패 필드/묘지/제외 존에 있는 카드의 영향을 무시한\n' \
                 '보편적으로 효과가 있는 상황에서의 순위입니다! 그 점 참고 해 주세요!\n\n'

        string += '주의사항 : 1턴에 1번 제약이 없는 효과에 우라라 던지지 않기\n' \
                  '주의사항 : 하루 우라라의 효과를 무시할 수 있는 카드거나 효과 무효를 무시하는 카드가 있는데 우라라 던지지 않기'

    elif arg == '리스트':
        string = '우라라 타이밍 조회가 가능한 덱 리스트 :\n'
        for k in Urara.keys():
            string += k + ' '

    else:
        string = ''

        if not arg in list(Urara.keys()):
            string += '다시 입력해 주세요!'
        else:
            Urara_list = Urara.get(arg)

            for i in range(len(Urara_list)):
                # 우라라 타이밍 리스트가 3개가 넘어가면 부가 설명으로 전환
                if i >= 3:
                    string += '부가 설명 : ' + Urara_list[i] + '\n'
                else:
                    string += str((i + 1)) + '순위 : ' + Urara_list[i] + '\n'

    await ctx.send(backtick(string))


@bot.command()
async def 티어덱(ctx):
    string = ''
    for i in range(len(tier)):
        string += f'{i + 1}티어 : '
        for j in range(len(tier[i])):
            string += tier[i][j] + ' '
        string += '\n'

    await ctx.send(backtick(string))


bot.run(TOKEN)