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.
Fox-V3/werewolf/Werewolf.py

103 lines
2.6 KiB

import asyncio
import discord
from discord.ext import commands
from redbot.core import Config
from datetime import datetime,timedelta
from .game import Game
class Werewolf:
"""
Base to host werewolf on a server
"""
def __init__(self, bot):
self.bot = bot
self.config = Config.get_conf(self, identifier=87101114101119111108102, force_registration=True)
default_global = {}
default_guild = {
}
self.config.register_global(**default_global)
self.config.register_guild(**default_guild)
self.games = {} # Active games stored here, id is per server
@commands.group()
async def ww(self, ctx: commands.Context):
"""
Base command for this cog. Check help for the commands list.
"""
if ctx.invoked_subcommand is None:
await ctx.send_help()
7 years ago
@ww.command()
7 years ago
async def join(self, ctx, game_code):
"""
Joins a game of Werewolf or start a new one
"""
7 years ago
game = self._get_game(ctx.guild, game_code)
7 years ago
if not game:
ctx.send("Please provide a role code to get started!")
return
7 years ago
await game.join(ctx.author, ctx.channel)
@ww.command()
async def quit(self, ctx):
"""
Quit a game of Werewolf
"""
7 years ago
game = self._get_game(ctx.guild)
7 years ago
await game.quit(ctx.author, ctx.channel)
@ww.command()
async def vote(self, ctx, id):
"""
Vote for a player by ID
"""
game = self._get_game(guild)
if not game:
ctx.send("No game running, cannot vote")
7 years ago
# Game handles response now
channel = ctx.channel
if channel is game.village_channel:
await game.vote(ctx.author, id, channel)
if channel in (c for id,c in game.secret_channels.items()):
await game.vote(ctx.author, id, channel)
7 years ago
7 years ago
def _get_game(self, guild, game_code = None):
7 years ago
if guild.id not in self.games:
7 years ago
if not game_code:
7 years ago
return None
7 years ago
self.games[guild.id] = Game(game_code)
7 years ago
return self.games[guild.id]
async def _game_start(self, game):
await game.start()
7 years ago
async def on_message(self, message):
if message.author.id == self.bot.user.id:
return
author = message.author
channel = message.channel
guild = message.guild
7 years ago
7 years ago
if channel is game.village_channel: