You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.4 KiB
101 lines
2.4 KiB
import asyncio
|
|
|
|
import discord
|
|
|
|
import random
|
|
|
|
|
|
class VoteGroup:
|
|
"""
|
|
Base VoteGroup class for werewolf game
|
|
Handles secret channels and group decisions
|
|
"""
|
|
|
|
allignment = 0 # 1: Town, 2: Werewolf, 3: Neutral
|
|
channel_id = ""
|
|
|
|
action_list = [
|
|
(self._at_game_start, 0), # (Action, Priority)
|
|
(self._at_day_start, 0),
|
|
(self._at_voted, 0),
|
|
(self._at_kill, 0),
|
|
(self._at_hang, 0),
|
|
(self._at_day_end, 0),
|
|
(self._at_night_start, 2),
|
|
(self._at_night_end, 0)
|
|
]
|
|
|
|
|
|
def __init__(self, game, members):
|
|
self.game = game
|
|
self.members = members
|
|
self.channel = None
|
|
self.vote_results = {}
|
|
self.properties = {} # Extra data for other options
|
|
|
|
|
|
async def on_event(self, event, data):
|
|
"""
|
|
See Game class for event guide
|
|
"""
|
|
|
|
await action_list[event][0](data)
|
|
|
|
async def _at_game_start(self, data=None):
|
|
if self.channel_id:
|
|
self.channel = await self.game.register_channel(self.channel_id)
|
|
|
|
async def _at_day_start(self, data=None):
|
|
pass
|
|
|
|
async def _at_voted(self, data=None):
|
|
pass
|
|
|
|
async def _at_kill(self, data=None):
|
|
pass
|
|
|
|
async def _at_hang(self, data=None):
|
|
pass
|
|
|
|
async def _at_day_end(self, data=None):
|
|
pass
|
|
|
|
async def _at_night_start(self, data=None):
|
|
if self.channel is None:
|
|
return
|
|
|
|
await self.game.generate_targets(self.channel)
|
|
|
|
async def _at_night_end(self, data=None):
|
|
if self.channel is None:
|
|
return
|
|
|
|
target = None
|
|
vote_list = list(self.vote_results.values())
|
|
|
|
if vote_list:
|
|
target = max(set(vote_list), key=vote_list.count)
|
|
|
|
if target:
|
|
# Do what you voted on
|
|
pass
|
|
|
|
async def register_member(self, member):
|
|
"""
|
|
Add a member to member list
|
|
"""
|
|
self.members.append(member)
|
|
|
|
async def remove_member(self, member):
|
|
"""
|
|
Remove a member from member list
|
|
"""
|
|
if member in self.members:
|
|
self.members.remove(member)
|
|
|
|
async def vote(self, author, id):
|
|
"""
|
|
Receive vote from game
|
|
"""
|
|
|
|
self.vote_results[author.id] = id |