|
13 | 13 | from src.sample_coach_agent import SampleCoachAgent |
14 | 14 | from src.sample_player_agent import SamplePlayerAgent |
15 | 15 | from src.sample_trainer_agent import SampleTrainerAgent |
| 16 | +import traceback |
16 | 17 |
|
17 | 18 |
|
18 | 19 | console_logging_level = logging.INFO |
@@ -51,8 +52,7 @@ def GetAction(self, state: pb2.State): |
51 | 52 | return self.GetTrainerActions(state) |
52 | 53 | except Exception as e: |
53 | 54 | self.logger.error(f"Error in GetAction: {e}") |
54 | | - import traceback |
55 | | - traceback.print_exc() |
| 55 | + self.logger.error(traceback.format_exc()) |
56 | 56 | return pb2.PlayerActions() |
57 | 57 |
|
58 | 58 | def GetPlayerActions(self, state: pb2.State): |
@@ -81,16 +81,31 @@ def GetTrainerActions(self, state: pb2.State): |
81 | 81 | return pb2.TrainerActions(actions=self.agent.get_actions()) |
82 | 82 |
|
83 | 83 | def SetServerParams(self, server_params: pb2.ServerParam): |
84 | | - self.logger.debug(f"Server params received unum {server_params.register_response.uniform_number}") |
85 | | - self.agent.set_server_params(server_params) |
| 84 | + try: |
| 85 | + self.logger.debug(f"Server params received unum {server_params.register_response.uniform_number}") |
| 86 | + self.agent.set_server_params(server_params) |
| 87 | + except Exception as e: |
| 88 | + self.logger.error(f"Error in GetAction: {e}") |
| 89 | + self.logger.error(traceback.format_exc()) |
| 90 | + return pb2.PlayerActions() |
86 | 91 |
|
87 | 92 | def SetPlayerParams(self, player_params: pb2.PlayerParam): |
88 | | - self.logger.debug(f"Player params received unum {player_params.register_response.uniform_number}") |
89 | | - self.agent.set_player_params(player_params) |
| 93 | + try: |
| 94 | + self.logger.debug(f"Player params received unum {player_params.register_response.uniform_number}") |
| 95 | + self.agent.set_player_params(player_params) |
| 96 | + except Exception as e: |
| 97 | + self.logger.error(f"Error in GetAction: {e}") |
| 98 | + self.logger.error(traceback.format_exc()) |
| 99 | + return pb2.PlayerActions() |
90 | 100 |
|
91 | 101 | def SetPlayerType(self, player_type: pb2.PlayerType): |
92 | | - self.logger.debug(f"Player type received unum {player_type.register_response.uniform_number}") |
93 | | - self.agent.set_player_types(player_type) |
| 102 | + try: |
| 103 | + self.logger.debug(f"Player type received unum {player_type.register_response.uniform_number}") |
| 104 | + self.agent.set_player_types(player_type) |
| 105 | + except Exception as e: |
| 106 | + self.logger.error(f"Error in GetAction: {e}") |
| 107 | + self.logger.error(traceback.format_exc()) |
| 108 | + return pb2.PlayerActions() |
94 | 109 |
|
95 | 110 | class GameHandler(pb2_grpc.GameServicer): |
96 | 111 | def __init__(self, shared_lock, shared_number_of_connections) -> None: |
@@ -158,8 +173,7 @@ def Register(self, register_request: pb2.RegisterRequest, context): |
158 | 173 | return register_response |
159 | 174 | except Exception as e: |
160 | 175 | main_logger.error(f"Error in Register: {e}") |
161 | | - import traceback |
162 | | - traceback.print_exc() |
| 176 | + main_logger.error(traceback.format_exc()) |
163 | 177 | return pb2.RegisterResponse() |
164 | 178 |
|
165 | 179 | def SendByeCommand(self, register_response: pb2.RegisterResponse, context): |
|
0 commit comments