drunkman.py

"""
Python emulation of the famous card game drunkman
"""

from random import shuffle
from collections import deque
import csv
from datetime import datetime
from pathlib import Path

deck = ['6', '6', '6', '6', '7', '7', '7', '7', '8', '8', '8', '8',
        '9', '9', '9', '9', '10', '10', '10', '10', 'J', 'J', 'J', 'J',
        'Q', 'Q', 'Q', 'Q', 'K', 'K', 'K', 'K', 'A', 'A', 'A', 'A']


def beats(card1, card2):
    """
    Checks which card is bigger
    Returns True if card1 is bigger, False is card2 is bigger
    """

    cards = ('6', '7', '8', '9', '10', 'J', 'Q', 'K', 'A')

    if card1 == '6' and card2 == 'A':
        return True
    if card1 == 'A' and card2 == '6':
        return False
    if cards.index(card1) > cards.index(card2):
        return True

    return False


def main(game):
    """
    implements main logic, returns number of steps taken
    filename - the file to record the game progress 
    """
    shuffle(deck)
    player1 = deque(deck[:18])
    player2 = deque(deck[18:])

    log = []

    log.append(f'Deck: {deck}')
    log.append(f'Player 1: {player1}')
    log.append(f'Player 2: {player2}')

    winner = 'None'

    step = 0

    while player1 and player2:
        step += 1
        cards1 = [player1.popleft()]
        cards2 = [player2.popleft()]

        while cards1[-1] == cards2[-1]:
            log.append(f'Spor on {cards1[-1]} and {cards2[-1]}')
            try:
                hidden1 = player1.popleft()
                hidden2 = player2.popleft()
            except IndexError:
                log.append('Player out of cards')
                break
            else:
                cards1.append(hidden1)
                log.append(f'Player 1 has put {hidden1} as hidden card')
                cards2.append(hidden2)
                log.append(f'Player 2 has put {hidden2} as hidden card')

            try:
                open1 = player1.popleft()
                open2 = player2.popleft()
            except IndexError:
                log.append('Using hidden cards for argue')
                print(f'Game {game}: Using hidden cards for argue')
            else:
                cards1.append(open1)
                log.append(f'Player 1 has put {open1} as open card')
                cards2.append(open2)
                log.append(f'Player 2 has put {open2} as open card')

        if beats(cards1[-1], cards2[-1]):
            player1.extend(cards1)
            player1.extend(cards2)
            log.append(f'One the step {step} Player 1 got {cards2[-1]} with {cards1[-1]},'
                       f'the bank taken {cards1, cards2}')
            log.append(f'Players 1 cards: {player1}')
            log.append(f'Players 2 cards: {player2}')
        else:
            player2.extend(cards2)
            player2.extend(cards1)
            log.append(f'On the step {step} Player 2 got {cards1[-1]} with {cards2[-1]}, '
                       f'the bank taken {cards1, cards2}')
            log.append(f'Players 1 cards: {player1}')
            log.append(f'Players 2 cards: {player2}')

        if step > 10000:
            print(f'Game {game}: Looped')
            log.append(f'Game {game}: Looped')
            with open(f'drunk/game_{game}.txt', 'w') as fl:
                fl.write('\n'.join(log))
            return step, winner

    if player1:
        print(f'Game {game}: Player 1 wins! Number of steps {step}')
        winner = '1'
    else:
        print(f'Game {game}: Player 2 wins! Number of steps {step}')
        winner = '2'

    with open(f'drunk/game_{game}.txt', 'w') as fl:
        fl.write('\n'.join(log))

    return step, winner


if __name__ == '__main__':
    steps = []
    start = datetime.now()
    path = Path().cwd() / 'drunk'
    for i in range(1000):
        steps.append(main(i))
        # if i % 10 == 0:
        #     name = path / f'{i}.zip'
        #     with ZipFile(name, 'w', compression=ZIP_DEFLATED, compresslevel=5) as zip_file:
        #         for one in path.glob('*.txt'):
        #             zip_file.write(one)
        #             one.unlink()

    finish = datetime.now()
    steps = ((x[0], x[1]) for x in steps)

    results_winners = {}
    results_steps = {}

    for one in steps:
        results_winners[one[1]] = results_winners.get(one[1], 0) + 1
        results_steps[one[0]] = results_steps.get(one[0], 0) + one[0]

    print(results_winners)
    # print(f"avarage length of the game {sum(not_cycled) / len(not_cycled)}")
    # print(f"minimal steps count: {min(not_cycled)}")
    # print(f"maximum steps count: {max(not_cycled)}")
    with open('drunker.csv', 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(steps)

    print(f"Time taken {finish - start}")