Changeset 6.1.25

Timestamp:

08/20/2016 08:46:48 PM (8 years ago)

Author:

David Owen <dsowen@fugue88.ws>

branch-nick:

state

revision id:

dsowen@fugue88.ws-20160820204648-w2xyagebfu9y0f4i

Message:

Objectify state

File:

• main.py (modified) (4 diffs)

main.py

@@ -40 +40 @@
 
 
+class AssembleTeam(object):
+    pass
+
+
 def main(player_count):
     spy_count = spies_for_players(player_count)
@@ -70 +74 @@
         team_size = mission_size(player_count, mission)
         print("Mission {} needs {} members".format(mission, team_size))
-        return "ASSEMBLE_TEAM"
+        return AssembleTeam()
 
     def assemble_team():
@@ -98 +102 @@
         else:
             team_failures += 1
-            return "ASSEMBLE_TEAM"
+            return AssembleTeam()
 
     def perform_mission():
@@ -129 +133 @@
         if isinstance(state, AdvanceMission):
             state = advance_mission()
-        elif state == "ASSEMBLE_TEAM":
+        elif isinstance(state, AssembleTeam):
             state = assemble_team()
         elif state == "PERFORM_MISSION":