updated states

README.md

@@ -16,7 +16,13 @@ WaitingForPlayers --> StartGame: Start && ready
 StartGame : Assign Player Roles
 StartGame : Show Introduction
 
-StartGame --> MainPhaseNight : all ready
+StartGame --> MainPhaseNight : No Cupin
+StartGame --> CupinSelect: Cupin
+
+state "Inform Loved Ones" as InformLovedOnes
+
+CupinSelect --> InformLovedOnes: Cupin selected
+InformLovedOnes --> MainPhaseNight: ack
 state MainPhaseNight {
  !include https://gitea.d1v3.de/matthias/pywerewolf/raw/branch/master/README.md!TheNight
 }
@@ -33,11 +39,7 @@ MainPhaseDay --> MainPhaseNight: Some Werewolfes left
 ```plantuml
 @startuml(id=TheNight)
 hide empty description
-[*] --> NightPhaseCupin: 
-
-NightPhaseCupin --> InformLovedOnes: Cupin selected
-InformLovedOnes --> NightPhaseMain: ack
-NightPhaseCupin --> NightPhaseMain : Cupin already selected or No Cupin
+[*] --> NightPhaseMain 
 ' state NightPhaseCupinEnd <<join>>
 ' NightPhaseCupin --> WaitToContinueCupin: !Cupin
 ' NightPhaseCupin --> SelectLovedOnes: Cupin

pywerwolf/gamelogic.py

@@ -149,6 +149,22 @@ class Game(object):
                     next=GameState.start_game,
                 )
             ],
+            GameState.start_game: [
+                TransitionObject(
+                    name = "StartGame 2 MainPhaseNight",
+                    inputType=GameMasterStartGame,
+                    condition=[self.players_ready, self.no_cupin],
+                    transition=self.assign_roles,
+                    next=GameState.night_phase,
+                ),
+                TransitionObject(
+                    name = "StartGame 2 CupinSelect",
+                    inputType=GameMasterStartGame,
+                    condition=[self.has_cupin],
+                    transition=self.assign_roles,
+                    next=GameState.night_phase,
+                )
+            ],
         }
 
         if game_id is None: