Mafiascum.net

Code: Select all

import random

class Player():
    def __init__(self, team, blocker):
        self.block = blocker
        self.team = team
        self.killing = False
        self.BP = False

        if self.team == "town":
            self.BP = True

    def __str__(self):
        return self.team

def main():
    wins = [0,0,0]
    n = int(input("How many times would you like it to run?\n"))
    scumA = True
    scumB = True
    
    players = initPlayers()
    done = ""
                
    for i in range(n):
        while(done == ""):

            if done == "":
                players = day(players)

            for player in players:
                print(str(player))
            print()
            
            teams = checkWinCon(players)
            done = teams[0]
            scumA = teams[1]
            scumB = teams[2]

            if done == "":
                players = night(players, scumA, scumB)

            for player in players:
                print(str(player))
            print()
            
            teams = checkWinCon(players)
            done = teams[0]
            scumA = teams[1]
            scumB = teams[2]
            
        end(done)
        
        if done == "Town":
            wins[0] += 1

        if done == "Scum A":
            wins[1] += 1

        if done == "Scum B":
            wins[2] += 1
        
        players = initPlayers()
        done = ""

    print("Town wins: " + str(wins[0]))
    print("Scum A wins: " + str(wins[1]))
    print("Scum B wins: " + str(wins[2]))

def initPlayers():
    players = []
    for i in range(4):
        players.append(Player("town", False))
        
    players.append(Player("scumA", True))
    players.append(Player("scumA", False))
    players.append(Player("scumB", True))
    players.append(Player("scumB", False))
    
    return players

def day(players):
    print("Day")
    random.seed()
    del players[random.randint(0,len(players)-1)]
    return players

def night(players, scumA, scumB):
    print("Night")
    random.seed()

    if scumA:
        killerA = findEnemy("scumA", players)
        players[killerA].killing = True

    if scumB:
        killerB = findEnemy("scumB", players)
        players[killerB].killing = True

    for player in players:
        if player.block and not player.killing:
            blocked = findEnemy(player.team, players)
            players[blocked].killing = False

    for player in players:
        if player.killing:
            target = findEnemy(player.team, players)
            players = kill(target, players)
            player.killing = False

    return players


def checkWinCon(players):
    scumA = 0
    scumB = 0
    Aexists = True
    Bexists = True

    for player in players:
        if player.team == "scumA":
            scumA += 1
        if player.team == "scumB":
            scumB += 1

    if scumA == 0:
        Aexists = False

    if scumB == 0:
        Bexists = False

    if scumA >= len(players)/2.0:
        return ["Scum A", 0, 0]

    if scumB >= len(players)/2.0:
        return ["Scum B", 0,0]

    if scumA == 0 and scumB == 0:
        print("sup")
        return ["Town", 0,0]

    return ["", Aexists, Bexists]

def end(team):
    print(team + " wins!\n")

def kill(index, players):
    if players[index].BP == True:
        players[index].BP = False
        print("checkBP")
    else:
        del players[index]
        print("checkNonBP")

    return players

def findEnemy(team, players):
    random.seed()
    enemy = Player("", False)

    while(enemy.team == team or enemy.team == ""):
        enemy = players[random.randint(0,len(players)-1)]

    return players.index(enemy)

main()