soj7rddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z Gdd eZdS) N)Game)User) GameBoard) GameOutcome)GameTurn)RussianRouletteChamberStatusc eZdZdddddejfdededededed ef fd Zdd Zdd dd e fdZ d e fdZ d e fdZ dZ dde d dd efdZdeded e fdZded efdZdeded e fdZd efdZxZS)RussianRouletteNrowscolumnscylinderbullet_xbullet_y game_turnct|_|_||nfdt jD_|_|_t|tst|n|_ jdks jdkr dSdS)NcLg|] }dtjD!S)c&g|]}tjS)RRChamberStatus NOT_FIRED).0_s 6/app/src/model/game/russianroulette/RussianRoulette.py z7RussianRoulette.__init__...sHHHq*HHH)ranger)rrselfs rrz,RussianRoulette.__init__..s?MNHHE$,4G4GHHHrr ) super__init__r rrrrr isinstancerrset_bullet_position)rr rrrrr __class__s` rr!zRussianRoulette.__init__s   # HRWX\XaRbRb & % '1)X'F'F UHY   I  =B  $-2"5"5  $ $ & & & & &#6"5rreturnc tjd|jdz }tjd|jdz }||jdzks||jdzkr||_||_dSg)z* Sets the bullet position TrN)randomrandintr rrrrxys rr#z#RussianRoulette.set_bullet_position,sr  q$)a-00Aq$,"233ADIN""a4<1+<&<&< ! !  r other_boardc|dup|}|}||nd}|s|r||krdSdS|rdS|||krdSd}t|jD]E\}}t|D]0\}} | t jkr|||s|dz }1F|dkS)z Returns if the game is finished. The game is finished when all but one chamber is fired, except for the center :param other_board: The other board, in case of global game :return: GameTurn NrTFr') is_finishedget_fired_correct_countbullet_is_fired enumeraterrris_chamber_center) rr.other_is_finishedself_fired_correct_countother_fired_correct_countnot_fired_count row_indexrow chamber_indexchambers rr0zRussianRoulette.is_finished;s< (4/L;3J3J3L3L#'#?#?#A#A 5@5LK / / 1 1 1RS "! ##%% *BE^*^*^t5    ! ! 4  "'?B['['[4' 66 ) )NIs*3C.. ) )& wo777@V@V}AA7$q(O  ) !##rcF|tjkS)zF Returns if the bullet is fired :return: bool )get_bullet_chamber_statusrFIREDrs rr2zRussianRoulette.bullet_is_fired`s --//?3HHHrc<|j|j|jS)zT Returns the bullet chamber status :return: RRChamberStatus )rrrr@s rr>z)RussianRoulette.get_bullet_chamber_statushs }T]+DM::rcf|jtjur tjn tj|_dS)z Sets the turn N)rrOPPONENT CHALLENGERr@s rset_turnzRussianRoulette.set_turnps, $(>X5F#F#FH  HL] ruser_is_challengerc||s tjS|t|}|}||kr|r tjn tjS||kr|r tjn tjStjS|r-|jtj kr tjStjStjS)z Returns the outcome of the game :param user_is_challenger: If the outcome is being evaluated by the challenger :param other_board: The other board, in case of global game :return: GameTurn )r.) r0rNONEr1CHALLENGER_WON OPPONENT_WONDRAWr2rrrD)rrFr. self_points other_pointss r get_outcomezRussianRoulette.get_outcomeysK88 $# #  "6688K&>>@@L\))2DbK..+Jb|++0BbK,, Hb# #    ! ! 2~!444"//"11rusergamecT|jtjur ||jkS||jkS)z{ Returns if the user is the turn :param user: User :param game: Game :return: bool rrrD challengeropponent)rrOrPs r is_user_turnzRussianRoulette.is_user_turns. >X0 0 04?* *4=( (rcH|jtjkr|jS|jS)z[ Returns the user turn :param game: The game :return: User rR)rrPs r get_user_turnzRussianRoulette.get_user_turns$ >X0 0 0? "}rr,r-c:||jdzko ||jdzkS)z Returns if the chamber is the center :param x: Location x :param y: Location y :return: If is the center of the chamber r()r rr+s rr4z!RussianRoulette.is_chamber_centers&DIN"=qDLA,='==rcd}t|jD]/\}}t|D]\}}|tjkr|dz }0|r|dz}|S)zu Returns the number of chambers that have been fired and don't contain a bullet :return: int rr')r3rrr?r2)rfired_correct_countr9r:r;r<s rr1z'RussianRoulette.get_fired_correct_counts  ' 66 - -NIs*3C.. - -& wo333'1,' -    ! ! % 1 $ ""r)r%N)N)__name__ __module__ __qualname__rrDintlistr!r#boolr0r2rr>rErrNrrrUrWr4r1 __classcell__)r$s@rr r s&1'''' '  '  '''''''<    #$#$'8#$D#$#$#$#$JIIIII;?;;;;   JN" " "&" 5F" " " " " H ) )T )d ) ) ) ) $ 4    >3>3>4>>>>#########rr )r)src.model.Gamersrc.model.Userrsrc.model.game.GameBoardrsrc.model.game.GameOutcomersrc.model.game.GameTurnr;src.model.game.russianroulette.RussianRouletteChamberStatusrrr rrrrhs ......222222,,,,,, C#C#C#C#C#iC#C#C#C#C#r