Imposter.py

from os import system, listdir
from platform import system as os
from random import sample, choice, randint

class Player:
    playerCount = 0
    currentWord = ''
    playerList  = []

    def __init__(self, playerName: str):
        self.name = playerName
        self.isImposter = False
        self.score = 0
        self.playerId = Player.playerCount
        Player.playerCount += 1
        Player.playerList.append(self)

    def __del__(self):
        print(f"{self.name} has been deleted.")
    
    def setImposter(self) -> None:
        self.isImposter = True
        return
    
    def clearImposter(self) -> None:
        self.isImposter = False
        return

    def reset(self) -> None:
        self.isImposter = False
        return
    
    def getId(self) -> int:
        return self.playerId
    
    def getWord(self) -> str:
        if self.isImposter:
            return "Imposter"
        else:
            return Player.currentWord
    
    def getName(self) -> str:
        return self.name
    
    @classmethod
    def setWord(cls, newWord: str) -> None:
        cls.currentWord = newWord
    
    @classmethod
    def getNameList(cls) -> list[str]:
        returnList = []
        for x in cls.playerList:
            returnList.append(x.getName())
        return returnList
    
    @classmethod
    def clearImposters(cls) -> None:
        for player in cls.playerList:
            player.clearImposter()
        return
    
    @classmethod
    def clearPlayers(cls) -> None:
        cls.playerList.clear()

def clearScreen() -> None:
    if os() == 'Windows':
        system('cls')
    else:
        system('clear')

def conjunction(nameList: list[str]) -> str:
    if len(nameList) == 0:
        return ''
    elif len(nameList) == 1:
        return nameList[0]
    elif len(nameList) == 2:
        return ' and '.join(nameList)
    else:
        return ', '.join(nameList[:-1]) + ', and ' + nameList[len(nameList) - 1]
    
def allImposter() -> bool:
    for lines in open('settings.txt', 'r').read().split('\n'):
        if lines == 'ALLIMPOSTERS = TRUE':
            if randint(1, 100) == 100:
                return True
    return False

def main() -> None:
    # Variables
    inputPlayers  = ''
    imposterCount = 0
    inputCount    = ''
    dictList      = [x[:-4] for x in listdir('dictionaries') if x[-4:] == '.txt']
    index         = 0
    indexSelector = ''
    userIndex     = 0
    currentFile   = ''
    wordCount     = 0
    imposters     = []
    wordList      = []
    votedImposter = []
    continueSame  = True
    continueNew   = True
    
    # Continue with new players loop
    while continueNew:

        # Get the player list
        clearScreen()
        inputPlayers = ''
        print('Enter "!" when finished (minimum 2 players required).')
        while True:
            inputPlayers = input('Enter player name: ')
            if inputPlayers == '!':
                if len(Player.playerList) >= 2:
                    break
                else:
                    print(f'You need at least 2 players! Currently have {len(Player.playerList)}.')
            elif inputPlayers in Player.getNameList():
                print('You cannot have repeat names')
            elif inputPlayers != '':
                Player(inputPlayers.strip())

        # Continue with same players loop
        continueSame = True
        while continueSame:

            # Get the imposter count
            clearScreen()
            imposterCount = -1
            while imposterCount <= 0 or imposterCount > len(Player.playerList) / 2:
                try:
                    imposterCount = int(input(f'How many imposters do you want (player count: {len(Player.playerList)}): '))
                    if imposterCount <= 0:
                        print('You need more than 0 imposters!')
                    if imposterCount > len(Player.playerList) / 2:
                        print('You need less than half the player count as imposter!')
                except:
                    print('Not a valid number!')

            # Get the category
            clearScreen()
            index         = 0
            indexSelector = ''
            userIndex     = -1
            print('Select a category.')
            for x in dictList:
                wordCount = len(open('dictionaries/' + x + '.txt', 'r').read().split('\n'))
                print(f" {index}) {x} ({wordCount})")
                index += 1
            while userIndex <= -1 or userIndex >= index:
                indexSelector = input(' >> ')
                try:
                    userIndex = int(indexSelector)
                except:
                    userIndex = -1
            currentFile = dictList[userIndex]

            # Select the word
            clearScreen()
            wordList = open('dictionaries/' + currentFile + '.txt', 'r').read().split('\n')
            Player.setWord(choice(wordList))

            # Set the imposters
            if allImposter():
                for x in Player.playerList:
                    x.setImposter()
            else:
                for x in sample(Player.playerList, imposterCount):
                    x.setImposter()
                
            
            # Give players the word
            for x in Player.playerList:
                clearScreen()
                print(x.getName())
                input('Press enter to reveal the word...')
                print(x.getWord())
                input('Press enter continue...')

            # Select first player
            clearScreen()
            print(f'{choice(Player.playerList).getName()} goes first.')
            input('Press Enter to Vote')

            # Vote Imposter
            votedImposter = []
            for x in range(imposterCount):
                clearScreen()
                index              = 0
                indexSelector      = ''
                userIndex          = -1
                remainingImposters = [y for y in Player.getNameList() if y not in votedImposter]
                print('Vote who is imposter.')
                for y in remainingImposters:
                    print(f" {index}) {y}")
                    index += 1
                while userIndex <= -1 or userIndex >= index:
                    indexSelector = input(' >> ')
                    try:
                        userIndex = int(indexSelector)
                    except:
                        userIndex = -1
                votedImposter.append(remainingImposters[userIndex])

            # Reveal imposter
            clearScreen()
            input(f'{conjunction(votedImposter)} {"was" if imposterCount == 1 else "were"} voted imposter...')
            input(f'{conjunction([x.getName() for x in Player.playerList if x.isImposter])}, what is the word? ')

            # Ask user if they want to continue
            clearScreen()
            indexSelector = ''
            userIndex     = -1
            print('Select an option.')
            print(" 0) Continue with New Players")
            print(" 1) Continue with Same Players")
            print(" 2) Exit")
            while userIndex <= -1 or userIndex >= 3:
                indexSelector = input(' >> ')
                try:
                    userIndex = int(indexSelector)
                except:
                    userIndex = -1
            if userIndex == 0:
                Player.clearPlayers()
                continueSame = False
            elif userIndex == 1:
                Player.clearImposters()
            elif userIndex == 2:
                continueSame = False
                continueNew  = False


if __name__ == "__main__":
    main()