Achievement-Promotion/bot.py

import os
import asyncio
from typing import Any

import disnake
from disnake.ext import commands
from dotenv import load_dotenv
from linked_roles import RoleMetadataType, LinkedRolesOAuth2, RoleMetadataRecord

load_dotenv()
bot = commands.InteractionBot(
    status=disnake.Status.dnd,
    activity=disnake.Activity(type=disnake.ActivityType.watching, name="Achievements"),
    intents=disnake.Intents.default()
)

def list_chunk(lst, n):
    return [lst[i:i+n] for i in range(0, len(lst), n)]

async def async_list(values: list) -> Any:
    for value in values:
        yield value
        await asyncio.sleep(0)

@bot.slash_command(name="register", description="Linked Role의 기본적인 연동을 설정합니다.")
@commands.is_owner()
async def _addConnection(inter: disnake.ApplicationCommandInteraction):
    await inter.response.defer(ephemeral=True)
    records = [
        RoleMetadataRecord(
            key="honor_roll",
            name="Honor roll",
            description="Perfect the tutorial",
            type=RoleMetadataType.boolean_equal,
            name_localizations={
                "ko": "우등생"
            },
            description_localizations={
                "ko": "튜토리얼 스테이지를 퍼펙트로 클리어해야 합니다."
            }
        ),
        RoleMetadataRecord(
            key="new_day",
            name="New Day",
            description="Get through the morning",
            type=RoleMetadataType.boolean_equal,
            name_localizations={
                "ko": "밝아오는 새로운 아침"
            },
            description_localizations={
                "ko": "모든 스테이지를 클리어해야 합니다."
            }
        ),
        RoleMetadataRecord(
            key="go_to_bed",
            name="Go to bed",
            description="Get all perfect scores",
            type=RoleMetadataType.boolean_equal,
            name_localizations={
                "ko": "자러 갈 시간"
            },
            description_localizations={
                "ko": "모든 스테이지를 퍼펙트로 클리어해야 합니다."
            }
        ),
        RoleMetadataRecord(
            key="percentage",
            name="% of Achievements",
            description="% of cleared achievements",
            type=RoleMetadataType.interger_greater_than_or_equal,
            name_localizations={
                "ko": "% 도전 과제 달성률"
            },
            description_localizations={
                "ko": "%의 도전 과제를 달성해야 합니다."
            }
        )
    ]
    async with LinkedRolesOAuth2(client_id=os.getenv("CLIENT_ID"), token=os.getenv("BOT_TOKEN")) as client:
        try:
            result = await client.register_role_metadata(records=tuple(records), force=True)
        except Exception as e:
            await inter.edit_original_message(e)
        else:
            await inter.edit_original_message(content=f"> :star2: 역할 추가가 완료되었습니다!\n{result}")

bot.run(os.getenv("BOT_TOKEN"))