[discord.py] 1. 간단하게 봇과 놀기

chaejm55·2021년 3월 5일

디스코드봇

목록 보기

2/18

0. 들어가기 전에

봇이 태어나고 hello world를 말한지 3일, 이젠 아기가 엄마 아빠 하듯 새로운 말을 배울 시기가 된 것이다. 그래서 이번 포스팅에서는 간단하게 봇과 놀 수 있도록 random 라이브러리를 사용한 명령어를 작성할 것이다.

1. 주사위 굴리기

아마 파이썬을 조금이라도 맛 봤다면 다 작성 해봤을 코드이다. 바로 코드를 보자.

@bot.command()
async def dice(ctx):
    randnum = random.randint(1, 6)  # 1이상 6이하 랜덤 숫자를 뽑음
    await ctx.send(f'주사위 결과는 {randnum} 입니다.')

실행을 시켜보면 이렇게 주사위 숫자를 출력한다.

2. 광물캐기

주사위는 너무 밋밋한 면이 있다.
그래서 이번에는 random.choices()를 이용하여 ~~로직이 아닌 공개된~~ 가중치를 부여하여 광물을 캐는 미니게임을 작성한다. 한번에 5개를 캐서 결과를 출력하는 방식의 게임이다.

@bot.command()
async def mining(ctx):
    minerals = ['다이아몬드', '루비', '에메랄드', '자수정', '철', '석탄']
    weights = [1, 3, 6, 15, 25, 50]
    results = random.choices(minerals, weights=weights, k=5)  # 광물 5개를 가중치에 따라 뽑음
    await ctx.send(', '.join(results) + ' 광물들을 획득하였습니다.')

실행을 하면 가중치에따라 결과가 나오는 모습을 확인할 수 있다.

3. 가위바위보

봇에 의해 정해지는 결과를 보는 건 조금 심심한 느낌이다.
그래서 다음 명령어는 가위바위보를 작성해보겠다. 입력을 받아야 하기 때문에 함수형태가 약간 달라진다. 아래 코드를 통해 살펴보자.

@bot.command()
async def game(ctx, user: str):  # user:str로 !game 다음에 나오는 메시지를 받아줌
    rps_table = ['가위', '바위', '보']
    bot = random.choice(rps_table)
    result = rps_table.index(user) - rps_table.index(bot)  # 인덱스 비교로 결과 결정
    if result == 0:
        await ctx.send(f'{user} vs {bot}  비겼습니다.')
    elif result == 1 or result == -2:
        await ctx.send(f'{user} vs {bot}  유저가 이겼습니다.')
    else:
        await ctx.send(f'{user} vs {bot}  봇이 이겼습니다.')