Fox-V3/werewolf/votegroup.py

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 = "" 

    def __init__(self, game, channel):
        self.game = game
        self.channel = channel
        self.players = []
        self.vote_results = {}
        self.properties = {}  # Extra data for other options
        
        self.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)
            ]
        
    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):
        pass

    async def _at_day_start(self, data=None):
        pass
        
    async def _at_voted(self, data=None):
        pass
        
    async def _at_kill(self, data=None):
        if data["player"] in self.players:
            self.players.pop(data["player"])
        
    async def _at_hang(self, data=None):
        if data["player"] in self.players:
            self.players.pop(data["player"])

    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_player(self, player):
        """
        Add a player to player list
        """
        self.players.append(player)
    
    async def remove_player(self, player):
        """
        Remove a player from player list
        """
        if player.id in self.players:
            self.players.remove(player)

    async def vote(self, author, id):
        """
        Receive vote from game
        """
        
        self.vote_results[author.id] = id