-
Notifications
You must be signed in to change notification settings - Fork 0
/
Yachubot.py
395 lines (330 loc) · 16.7 KB
/
Yachubot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
from Yachu import *
from MultiYachu import MultiYachu
from discord.ext import commands
from asyncio import TimeoutError
from ext_nalgang import *
from config import *
from logger import put_log
game = discord.Game("도움말은 !야추도움")
bot = commands.Bot(command_prefix='!', activity=game)
bot.player_one = None
bot.player_two = None
bot.channel_now = None
TIME_OUT = 90.0
def make_free(bot):
bot.player_one = None
bot.player_two = None
bot.channel_now = None
return
def is_free(bot):
return (bot.player_one is None) and (bot.player_two is None)
@bot.check # only available in whitelisted channel
async def channel_whitelist(ctx):
put_log(f'{ctx.message.content} called by {ctx.author.display_name} in {ctx.channel.name} at {time.ctime()}\n')
return ctx.channel.id in CHANNEL_WHITELIST
@bot.event
async def on_ready():
make_free(bot)
put_log(f'bot started at {time.ctime()}\n')
if ALERT_AT_START:
for channel_id in CHANNEL_WHITELIST:
await bot.get_channel(channel_id).send('야추봇이 시작되었습니다! **주의: 봇이 5초 이상 멈춰있으면 아무 입력이나 넣어주세요')
print("READY")
@bot.command(name='야추연습')
async def start(ctx: discord.ext.commands.Context):
if not is_free(bot):
await ctx.send("다른 분과 이미 야추중이에요. 기다려주세요")
return
else:
await ctx.send(f'{ctx.author.mention}님과 야추 연습을 시작합니다.')
bot.player_one = ctx.author
bot.channel_now = ctx.channel
await single_play(ctx.author, bot.channel_now, betpoint=0)
@bot.command(name='야추도움')
async def print_help(ctx):
await ctx.send('```\n'
'!야추도움 - 이 도움말을 봅니다.\n'
'!야추연습 - 야추봇과 야추 주사위 연습게임을 시작합니다.\n'
'!야추베팅 (걸 점수) - 날갱점수를 걸고 총점 200점에 도전합니다. 승리하면 두 배로 돌려받습니다.\n'
'!야추대결 (@대결상대) (걸 점수) - 날갱점수를 걸고 상대와 대결합니다.\n'
'!야추규칙 - 안읽고 하면 책임안짐ㅎㅎ...ㅋㅋ...ㅈㅅ!!'
'```'
)
@bot.command(name='야추규칙')
async def rule(ctx):
await ctx.send('```\n'
'1. 입출력 과정에 멈춤 현상이 일어날 경우, 아무 메시지나 반응 이모지를 전송하면 해결할 수 있습니다.\n'
'2. 잘못된 입력이 들어오면 자동으로 게임이 종료될 수 있으며, 포인트 역시 사라질 수 있습니다.\n'
'3. 합당한 귀책사유가 있는 경우를 제외하고, 버그로 인한 포인트 손실은 책임지지 않습니다. \n'
'3.1 특히, 이용자 단순 실수, 입출력 과정의 프리징, 서버 다운 등은 해결할 수 없으므로 책임지지 않습니다.\n'
'4. 버그제보 및 코드 리뷰, 기능 추가는 언제든 환영입니다! => https://github.com/SeminKim/YachtDiscord\n'
'```'
)
return
'''@bot.command(name='점수조회')
async def point_query(ctx):
await ctx.send(ng_getpoint(ctx.author.id, ctx.guild.id))'''
@bot.command(name='야추베팅')
async def bet(ctx, point: int):
if not is_free(bot):
await ctx.send("다른 분과 이미 야추중이에요. 기다려주세요")
return
if point < 0:
await ctx.send('되겠냐고ㅋㅋㅋ')
return
if not USE_NG_API:
await ctx.send('포인트 기능이 비활성화되어 사용할 수 없습니다.')
return
limit = ng_getpoint(ctx.author.id, ctx.guild.id)
if point > limit: # 생각해보니 게임 전 베팅액을 미리 차감해야할 것 같음. 게임하면서 동시에 다른 채널에서 뒤로 빼돌리는 문제가 있음
drip_msg: discord.Message = await ctx.send(f'{limit}점 가지고 있어서 {point}점 베팅할 수 없어요. ㅋㅋㅋㅋ거지쉑')
time.sleep(1)
await drip_msg.edit(content=f'{limit}점 가지고 있어서 {point}점 베팅할 수 없어요ㅠ.ㅠ')
return
await ctx.send(f'{ctx.author.mention}님이 {point}점을 걸고 야추 200점에 도전합니다.')
bot.player_one = ctx.author
bot.channel_now = ctx.channel
await single_play(ctx.author, bot.channel_now, betpoint=point)
@bot.command(name='야추대결')
async def vs(ctx, user: discord.member.Member, point: int):
if is_free(bot):
if not USE_NG_API:
await ctx.send('포인트 기능이 비활성화되어 사용할 수 없습니다.')
return
a, b = ng_getpoint(ctx.author.id, ctx.guild.id), ng_getpoint(user.id, ctx.guild.id)
if a < point or b < point or point < 0:
await ctx.send(f'거 상도덕은 지키면서 삽시다. 어떻게 {point}점을 걸겠다고 그러십니까?\n(현재 각각 {a}점, {b}점 보유)')
else:
bot.player_one = ctx.author
bot.player_two = user
await ctx.send(f'{ctx.author.mention}님이 {user.mention}님과 {point}점을 걸고 대결을 신청합니다.')
await ctx.send(f'{user.display_name}님, 동의하시면 1분 내에 \"나는야 야추왕\"이라고 정확히 적어주세요')
def check(m):
if user.id != m.author.id: return False
if m.content != '나는야 야추왕': return False
return True
try:
await bot.wait_for('message', check=check, timeout=TIME_OUT)
except TimeoutError:
await ctx.send('도전 거절!')
make_free(bot)
return
await ctx.send(f'{ctx.author.display_name}님과 {user.display_name}님의 대결이 시작됩니다!')
await vs_playing(ctx.author, user, ctx.channel, point)
else:
await ctx.send("다른 분과 이미 야추중이에요. 기다려주세요")
return
async def play_one_turn(yachu: Yachu,
player: discord.Member,
chan: discord.TextChannel,
board_msg: discord.Message,
roll_msg: discord.Message,
phase_msg: discord.Message,
betpoint: int):
dice_index_list = ['1️⃣', '2️⃣', '3️⃣', '4️⃣', '5️⃣', '🆗']
save_index_list = ['1️⃣', '2️⃣', '3️⃣', '4️⃣', '5️⃣', '6️⃣', '🇦', '🇧', '🇨', '🇩', '🇪', '🇫', '🆗']
def check(reaction, user):
return user == player and str(reaction.emoji) == '🆗'
while yachu.phase < 3:
# 먼저 주사위를 굴리고(phase 1증가), 무엇을 다시 굴릴 지 질문 메시지 전송. 메시지 화면은 board, dice, phase, ask 순.
await roll_msg.edit(content=yachu.rollDice())
await phase_msg.edit(content='\n{}번 다시 굴릴 수 있습니다'.format(3 - yachu.phase))
if yachu.phase == 3: break # 굴릴 기회가 없으면 저장하러 보냄
ask_msg = await chan.send('다시 굴릴 주사위 번호를 고르고 :ok:버튼을 눌러주세요.')
for button in dice_index_list: # Button은 이모지
await ask_msg.add_reaction(button)
# Reaction 기다린 다음 OK 이모지인지, 유저는 동일인인지 체크. 타임아웃시 종료
try:
reaction_type, user = await bot.wait_for('reaction_add', timeout=TIME_OUT, check=check)
except TimeoutError:
await chan.send('Timeout!')
return False
# 캐시된 메시지를 새걸로 업데이트하고, Reaction 중 dice index 찾아서 unlock
ask_msg: discord.Message = discord.utils.get(bot.cached_messages, id=ask_msg.id)
yachu.lockAll()
for reaction in ask_msg.reactions: # reaction:discord.Reaction
if (reaction.emoji not in dice_index_list) or (reaction.emoji == '🆗'): continue
async for user in reaction.users():
if user != player: continue
yachu.unlock(dice_index_list.index(reaction.emoji) + 1)
break
await ask_msg.delete()
# 만약 다 잠겨있으면, 루프를 종료하고 점수 저장하도록 넘어감
if yachu.isAllLocked():
break
# 점수 저장하는 부분
ask_msg = await chan.send(content='저장할 항목을 고르고 :ok:버튼을 눌러주세요')
for button in save_index_list:
await ask_msg.add_reaction(button)
try:
reaction_type, user = await bot.wait_for('reaction_add', timeout=TIME_OUT, check=check)
except TimeoutError:
await chan.send('Timeout!')
return False
# 어느 칸에 저장할지 결정, 처리
ind = -1
breaker = False
ask_msg = discord.utils.get(bot.cached_messages, id=ask_msg.id)
for reaction in ask_msg.reactions: # type:discord.Reaction
if (reaction.emoji not in save_index_list) or (reaction.emoji == '🆗'): continue
if breaker: break
async for user in reaction.users():
if user != player: continue
ind = save_index_list.index(reaction.emoji)
if yachu.isAvailable(ind):
breaker = True
break
else:
ind = -1
if ind == -1:
await chan.send('잘못된 응답입니다. 야추 게임을 종료합니다.')
return False
await ask_msg.delete()
yachu.setScore(ind)
# 현재 스코어보드 전송
await board_msg.edit(embed=yachu.getScoreBoard(name=player.display_name))
return True
async def single_play(player: discord.Member, chan: discord.TextChannel, betpoint=0):
yachu = Yachu()
board_msg = await chan.send(embed=yachu.getScoreBoard())
roll_msg = await chan.send('로딩 중...')
phase_msg = await chan.send('로딩 중...')
i = 12
while i > 0:
if not await play_one_turn(yachu=yachu,
player=player,
chan=chan,
board_msg=board_msg,
roll_msg=roll_msg,
phase_msg=phase_msg,
betpoint=betpoint):
await ng_addpoint(chan, player, -1 * betpoint)
make_free(bot)
return
i -= 1
# 결과 및 점수정산
await chan.send(embed=yachu.getScoreBoard())
await chan.send(f'{yachu.getTotal()}점을 얻으셨습니다!')
if betpoint > 0:
if yachu.score[14] < 200:
await ng_addpoint(chan, player, -1 * betpoint)
await chan.send('안타깝게도 200점을 획득하지는 못하셨네요. 다음 기회에!')
else:
await ng_addpoint(chan, player, betpoint)
await chan.send(f'200점 이상을 획득하셨네요. {betpoint}점을 획득하였습니다!')
make_free(bot)
return
async def vs_playing(player_one: discord.Member,
player_two: discord.Member,
chan: discord.TextChannel,
betpoint: int):
multiyachu = MultiYachu()
i = 12
board_msg_one = await chan.send(embed=multiyachu.player_one.getScoreBoard(name=player_one.display_name))
board_msg_two = await chan.send(embed=multiyachu.player_two.getScoreBoard(name=player_two.display_name))
roll_msg = await chan.send('로딩 중...')
phase_msg = await chan.send('로딩 중...')
while i > 0:
player_indicator_msg = await chan.send(f'{player_one.mention}님의 차례입니다.')
if not await play_one_turn(yachu=multiyachu.player_one,
player=player_one,
chan=chan,
board_msg=board_msg_one,
roll_msg=roll_msg,
phase_msg=phase_msg,
betpoint=betpoint):
await ng_movepoint(chan, sender=player_one, receiver=player_two, point=betpoint)
make_free(bot)
return
await player_indicator_msg.delete()
player_indicator_msg = await chan.send(f'{player_two.mention}님의 차례입니다.')
if not await play_one_turn(yachu=multiyachu.player_two,
player=player_two,
chan=chan,
board_msg=board_msg_two,
roll_msg=roll_msg,
phase_msg=phase_msg,
betpoint=betpoint):
await ng_movepoint(chan, sender=player_two, receiver=player_one, point=betpoint)
make_free(bot)
return
i -= 1
await player_indicator_msg.delete()
await chan.send(multiyachu.getScoreBoard())
a, b = multiyachu.getFinalScore()
await chan.send(f'{player_one.display_name}님이 {a}점, {player_two.display_name}님이 {b}점 획득하셨네요!')
if a > b:
await ng_movepoint(chan, sender=player_two, receiver=player_one, point=betpoint)
elif a < b:
await ng_movepoint(chan, sender=player_one, receiver=player_two, point=betpoint)
else:
await chan.send('어케비겼누ㄷ.ㄷ')
result = [ng_getpoint(player_one.id, chan.guild.id), ng_getpoint(player_two.id, chan.guild.id)]
await chan.send(
f'게임 결과에 따라, 각각 {result[0]}점, {result[1]}점이 되셨습니다!'
)
make_free(bot)
# LEGACY CODE
''' yachus = [Yachu(), Yachu()]
multiyachu = MultiYachu(yachus[0], yachus[1], player_one.display_name, player_two.display_name)
for i in range(12):
for j in range(2):
if j == 0:
await ctx.send(f'{ctx.author.display_name}님의 차례입니다')
else:
await ctx.send(f'{user.display_name}님의 차례입니다')
await ctx.send(multiyachu.getScoreBoard())
while yachus[j].phase < 3:
await ctx.send(yachus[j].rollDice())
if yachus[j].phase == 3:
def func1(msg):
this = int(msg.content)
if yachus[j].isAvailable(this):
return this
else:
raise ReAsk
if j == 0:
ind = await question(ctx, '저장할 칸 선택', func1, ctx.author, betpoint)
else:
ind = await question(ctx, '저장할 칸 선택', func1, user, betpoint)
else:
def func2(msg):
this = int(msg.content)
if this == 0: return this
if yachus[j].isAvailable(this):
return this
else:
raise ReAsk
if j == 0:
ind = await question(ctx, '저장할 칸 선택, 다시 굴리려면 0', func2, ctx.author, betpoint)
else:
ind = await question(ctx, '저장할 칸 선택, 다시 굴리려면 0', func2, user, betpoint)
if ind == -1: return
if ind == 0:
yachus[j].unlockAll()
def func3(msg):
temp = msg.content.split()
if len(temp) == 1 and temp[0] == '0': return
for ind in temp:
ind = int(ind)
yachus[j].lock(ind)
return
if j == 0:
if await question(ctx, '고정할 주사위 선택 - ex) 1 2 4, 없으면 0', func3, ctx.author,
betpoint) == -1: return
if j == 1:
if await question(ctx, '고정할 주사위 선택 - ex) 1 2 4, 없으면 0', func3, user, betpoint) == -1: return
else:
yachus[j].setScore(ind)
break
await ctx.send(multiyachu.getScoreBoard())
a, b = multiyachu.getFinalScore()
await ctx.send(f'{ctx.author.display_name}님이 {a}점, {user.display_name}님이 {b}점 획득하셨네요!')
if a > b:
ng_movepoint(ctx, sender=user, receiver=ctx.author, point=betpoint)
elif a < b:
ng_movepoint(ctx, sender=ctx.author, receiver=user, point=betpoint)
else:
await ctx.send('어케비겼누ㄷ.ㄷ')
await ctx.send(f'게임 결과에 따라, 각각 {ng_getpoint(ctx.author)}점, {ng_getpoint(user)}점이 되셨습니다!')
return'''