Set some stuff up for werewolf 🐺

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()

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

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)

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)