Set some stuff up for werewolf 🐺

2020-12-03 15:48:02 +01:00
parent 01553d5ffc
commit d03ae74937
5 changed files with 160 additions and 22 deletions
--- a/cogs/GamesCog.py
+++ b/cogs/GamesCog.py
@ -5,14 +5,14 @@ from funcs import logThis

 class GamesCog(commands.Cog):

-    def __init__(self,client):
+    def __init__(self,bot):
        """Runs game stuff."""
-        self.client = client
+        self.bot = bot

    # Checks user balance
    @commands.command(aliases = ["b"])
    async def balance(self, ctx):
-        response = self.client.money.checkBalance("#"+str(ctx.message.author.id))
+        response = self.bot.money.checkBalance("#"+str(ctx.message.author.id))
        if response == 1:
            new_message = ctx.message.author.display_name + " has " + str(response) + " GwendoBuck"
        else:
@ -25,19 +25,19 @@ class GamesCog(commands.Cog):
        commands = content.split(" ")
        amount = int(commands[-1])
        username = " ".join(commands[:-1])
-        if self.client.funcs.getID(username) == None:
+        if self.bot.funcs.getID(username) == None:
            async for member in ctx.guild.fetch_members(limit=None):
                if member.display_name.lower() == username.lower():
                    username = member.display_name
                    userID = "#" + str(member.id)
-                    self.client.database["users"].insert_one({"_id":userID,"user name":username,"money":0})
-        response = self.client.money.giveMoney("#"+str(ctx.message.author.id),username,amount)
+                    self.bot.database["users"].insert_one({"_id":userID,"user name":username,"money":0})
+        response = self.bot.money.giveMoney("#"+str(ctx.message.author.id),username,amount)
        await ctx.send(response)

    # Invest GwendoBucks in the stock market
    @commands.command(aliases=["i"])
    async def invest(self, ctx, *, content = "check"):
-        response = self.client.invest.parseInvest(content,"#"+str(ctx.message.author.id))
+        response = self.bot.invest.parseInvest(content,"#"+str(ctx.message.author.id))
        if response.startswith("**"):
            responses = response.split("\n")
            em = discord.Embed(title=responses[0],description="\n".join(responses[1:]),colour=0x00FF00)
@ -49,7 +49,7 @@ class GamesCog(commands.Cog):
    @commands.command()
    async def trivia(self, ctx, *, content = ""):
        if content == "":
-            question, answers, correctAnswer = self.client.trivia.triviaStart(str(ctx.message.channel.id))
+            question, answers, correctAnswer = self.bot.trivia.triviaStart(str(ctx.message.channel.id))
            if answers != "":
                results = "**"+question+"**\n"
                for x, answer in enumerate(answers):
@ -59,9 +59,9 @@ class GamesCog(commands.Cog):

                await asyncio.sleep(60)

-                self.client.trivia.triviaCountPoints(str(ctx.message.channel.id))
+                self.bot.trivia.triviaCountPoints(str(ctx.message.channel.id))

-                self.client.funcs.deleteGame("trivia questions",str(ctx.message.channel.id))
+                self.bot.funcs.deleteGame("trivia questions",str(ctx.message.channel.id))

                logThis("Time's up for the trivia question",str(ctx.message.channel.id))
                await ctx.send("Time's up The answer was \""+chr(correctAnswer)+") "+answers[correctAnswer-97]+"\". Anyone who answered that has gotten 1 GwendoBuck")
@ -69,7 +69,7 @@ class GamesCog(commands.Cog):
                await ctx.send(question)

        elif content in ["a","b","c","d"]:
-            response = self.client.trivia.triviaAnswer("#"+str(ctx.message.author.id),str(ctx.message.channel.id),content)
+            response = self.bot.trivia.triviaAnswer("#"+str(ctx.message.author.id),str(ctx.message.channel.id),content)
            if response.startswith("Locked in "):
                await ctx.message.add_reaction("👍")
            else:
@ -81,32 +81,37 @@ class GamesCog(commands.Cog):
    # Runs a game of blackjack
    @commands.command(aliases = ["bj"])
    async def blackjack(self, ctx, *, content = ""):
-        await self.client.blackjack.parseBlackjack(content,ctx)
+        await self.bot.blackjack.parseBlackjack(content,ctx)

    # Runs a game of Connect four
    @commands.command(aliases = ["fiar","connect4","connectfour","4iar","4inarow"])
    async def fourinarow(self, ctx, *, content = ""):
-        await self.client.gameLoops.fiar(ctx.message.channel,content,"#"+str(ctx.message.author.id))
+        await self.bot.gameLoops.fiar(ctx.message.channel,content,"#"+str(ctx.message.author.id))

    # Runs a game of Monopoly
    @commands.command(aliases = ["m","mon","mono"])
    async def monopoly(self, ctx, *, content = ""):
-        await self.client.gameLoops.runMonopoly(ctx.message.channel,content,"#"+str(ctx.message.author.id))
+        await self.bot.gameLoops.runMonopoly(ctx.message.channel,content,"#"+str(ctx.message.author.id))

    # Runs a game of Hangman
    @commands.command(aliases = ["hm"])
    async def hangman(self, ctx, *, content = "start"):
-        await self.client.gameLoops.runHangman(ctx.message.channel,"#"+str(ctx.message.author.id),content)
+        await self.bot.gameLoops.runHangman(ctx.message.channel,"#"+str(ctx.message.author.id),content)

    # Runs a game of Hex
    @commands.command(name="hex")
    async def hexCommand(self, ctx, *, content = ""):
-        await self.client.gameLoops.runHex(ctx.message.channel,content,"#"+str(ctx.message.author.id))
+        await self.bot.gameLoops.runHex(ctx.message.channel,content,"#"+str(ctx.message.author.id))

    # Runs a game of Love Letter
    @commands.command(aliases = ["ll"])
    async def loveletter(self, ctx, *, content = ""):
-        await self.client.gameLoops.runLoveletter(ctx.message.channel,content,"#"+str(ctx.message.author.id),ctx.message.author)
+        await self.bot.gameLoops.runLoveletter(ctx.message.channel,content,"#"+str(ctx.message.author.id),ctx.message.author)

-def setup(client):
-    client.add_cog(GamesCog(client))
+    # Runs a game of Werewolf
+    @commands.command()
+    async def werewolf(self,ctx, *,content = "start"):
+        await self.bot.werewolf.parseWerewolf(ctx,content.lower())
+
+def setup(bot):
+    bot.add_cog(GamesCog(bot))
--- a/funcs/funcs.py
+++ b/funcs/funcs.py
@ -30,6 +30,7 @@ class Funcs():
    def stopServer(self):
        self.bot.database["trivia questions"].delete_many({})
        self.bot.database["blackjack games"].delete_many({})
+        self.bot.database["werewolf games"].delete_many({})

        g = git.cmd.Git("")
        g.pull()
--- a/funcs/games/blackjack.py
+++ b/funcs/games/blackjack.py
@ -50,7 +50,6 @@ class Blackjack():
            else:
                for x in range(len(values)):
                    values[x] += int(cardValue)
-                    print(values[x])

        values.sort()

@ -91,9 +90,7 @@ class Blackjack():
        else:
            done = True

-        print(dealerHand)
        if self.calcHandValue(dealerHand) > 21:
-            print("yeet")
            self.bot.database["blackjack games"].update_one({"_id":channel},{"$set":{"dealer busted":True}})

        return done
--- a/funcs/games/games.py
+++ b/funcs/games/games.py
@ -6,6 +6,7 @@ from .gameLoops import GameLoops
 from .monopoly import Monopoly
 from .hangman import Hangman
 from .hex import HexGame
+from .werewolf import Werewolf

 class Games():
    def __init__(self, bot):
@ -19,3 +20,4 @@ class Games():
        bot.monopoly = Monopoly(bot)
        bot.hangman = Hangman(bot)
        bot.hex = HexGame(bot)
+        bot.werewolf = Werewolf(bot)
--- a/funcs/games/werewolf.py
+++ b/funcs/games/werewolf.py
@ -0,0 +1,133 @@
+import datetime
+import discord
+import asyncio
+from funcs import logThis
+from enum import Enum
+
+minimumPlayers = 1
+minimumPlayersWithAI = 5
+
+class Team(Enum):
+    Villager = 1
+    Werewolf = 2
+
+class Role():
+    def __init__(self,team,isWerewolf=False,isSeer=False,isSorcerer=False):
+        self.alive = True
+        self.team = team
+        self.isWerewolf = isWerewolf
+        self.isSeer = isSeer
+        self.isSorcerer = isSorcerer
+
+class VillagerRole(Role):
+    def __init__(self):
+        super().__init__(Team.Villager)
+
+class WerewolfRole(Role):
+    def __init__(self):
+        super.__init__(Team.Werewolf,isWerewolf=True)
+
+class SeerRole(Role):
+    def __init__(self):
+        super.__init__(Team.Villager,isSeer=True)
+
+class SorcererRole(Role):
+    def __init__(self):
+        super.__init__(Team.Werewolf,isSorcerer=True)
+
+class SorcererWerewolfRole(Role):
+    def __init__(self):
+        super.__init__(Team.Werewolf,isWerewolf=True,isSorcerer=True)
+
+class Werewolf():
+    def __init__(self,bot):
+        self.bot = bot
+
+    # Starts a game of werewolf
+    def werewolfStart(self,channel:str):
+        game = self.bot.database["werewolf games"].find_one({"_id":channel})
+
+        logThis("Trying to start a werewolf game in "+channel)
+
+        if game == None:
+
+            gameID = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
+
+            newGame = {"_id":channel,"users": {},"round":0,"turn":0,"turnOrder":[],"current message":None,"gameID":gameID}
+
+            self.bot.database["werewolf games"].insert_one(newGame)
+
+            return "Started a game of werewolf. Use \"!werewolf join\" to join the game", True
+        else:
+            logThis("There is already a werewolf game going on in "+channel)
+            return "There's already a werewolf game going on in this channel. Try again in a few minutes.", False
+
+    # Stops the game of werewolf
+    def werewolfStop(self,channel:str):
+        game = self.bot.database["werewolf games"].find_one({"_id":channel})
+
+        if game != None:
+            self.bot.database["werewolf games"].delete_one({"_id":channel})
+            return "Ended the werewolf game"
+        else:
+            return "There's no game going on right now"
+
+    async def werewolfLoop(self,ctx,round):
+        logThis(f"Starting loop {round} of Werewolf game")
+        channel = "#"+str(ctx.channel.id)
+        if round == 0:
+            await asyncio.sleep(120)
+            game = self.bot.database["werewolf games"].find_one({"_id":channel})
+            AINumber = 1
+            while len(game["users"].keys()) >= minimumPlayers and len(game["users"].keys()) < minimumPlayersWithAI:
+                user = {"role": None}
+                self.bot.database["werewolf games"].update_one({"_id":"#"+str(ctx.channel.id)},
+                {"$set":{"users."+str(AINumber):user}})
+                game = self.bot.database["werewolf games"].find_one({"_id":channel})
+                AINumber += 1
+            self.bot.database["werewolf games"].update_one({"_id":channel},{"$inc":{"round":1}})
+            await self.werewolfLoop(ctx,round+1)
+        else:
+            await asyncio.sleep(60)
+
+
+    async def werewolfJoin(self,ctx):
+        currentGames = list(self.bot.database["werewolf games"].find({}))
+        for game in currentGames:
+            if "#"+str(ctx.message.author.id) in game["users"].keys():
+                return f"{ctx.message.author.display_name} is already in a werewolf game", False
+
+        game = self.bot.database["werewolf games"].find_one({"_id":"#"+str(ctx.channel.id)})
+        if game["round"] == 0:
+            user = {"role": None}
+            self.bot.database["werewolf games"].update_one({"_id":"#"+str(ctx.channel.id)},
+                {"$set":{"users."+"#"+str(ctx.message.author.id):user}})
+            return "You joined the werewolf game", True
+        else:
+            return "It's too late to join", False
+
+    async def parseWerewolf(self,ctx,content):
+        if content == "start":
+            response, started = self.werewolfStart("#"+str(ctx.channel.id))
+            await ctx.send(response)
+            if started:
+                await self.werewolfLoop(ctx,0)
+
+        elif content == "stop":
+            await ctx.send(self.werewolfStop("#"+str(ctx.channel.id)))
+
+        elif content == "join":
+            response, joined = await self.werewolfJoin(ctx)
+            await ctx.send(response)
+            if joined:
+                game = self.bot.database["werewolf games"].find_one({"_id":"#"+str(ctx.channel.id)})
+                if game["current message"] != None:
+                    await ctx.channel.fetch_message(game["current message"]).delete()
+                users = ""
+                for x, user in enumerate(list(game["users"].keys())):
+                    users += str(x+1)+") "+self.bot.funcs.getName(user)+"\n"
+
+                if len(list(game["users"].keys())) < minimumPlayers:
+                    users += f"You need at least {minimumPlayers} players to play werewolf."
+                em = discord.Embed(title="Current Players",description=users,colour=0x00FF00)
+                await ctx.send(embed=em)