Added Player.team_name and Player.player_class_name attributes (#340).

Author: jordanbriere

addons/source-python/data/source-python/teams/cstrike.ini

@@ -5,3 +5,13 @@ un = 0
 spec = 1
 t = 2
 ct = 3
+
+[classes]
+Phoenix Connexion = 1
+Elite Crew = 2
+Artic Avengers = 3
+Guerilla Warfare = 4
+Seal Team 6 = 5
+GSG-9 = 6
+SAS = 7
+GIGN = 8

addons/source-python/data/source-python/teams/tf.ini

@@ -5,3 +5,14 @@ un = 0
 spec = 1
 red = 2
 blue = 3
+
+[classes]
+Scout = 1
+Sniper = 2
+Soldier = 3
+Demoman = 4
+Medic = 5
+Heavy = 6
+Pyro = 7
+Spy = 8
+Engineer = 9

addons/source-python/packages/source-python/players/_base.py

@@ -68,6 +68,10 @@
 from players.helpers import playerinfo_from_index
 from players.helpers import index_from_userid
 from players.helpers import uniqueid_from_playerinfo
+from players.teams import classes_by_name
+from players.teams import classes_by_number
+from players.teams import teams_by_name
+from players.teams import teams_by_number
 from players.voice import mute_manager
 #   Weapons
 from weapons.default import NoWeaponManager
@@ -264,6 +268,32 @@ def set_team(self, value):
 
     team = property(get_team, set_team)
 
+    def get_team_name(self):
+        """Return the name of the player's team."""
+        return teams_by_number.get(self.team, 'Unknown')
+
+    def set_team_name(self, team):
+        """Set the player's team."""
+        team = teams_by_name.get(team, None)
+        if team is None:
+            raise ValueError('Invalid team name.')
+        self.team = team
+
+    team_name = property(get_team_name, set_team_name)
+
+    def get_player_class_name(self):
+        """Return the name of the player's class."""
+        return classes_by_number.get(self.player_class, 'Unknown')
+
+    def set_player_class_name(self, player_class):
+        """Set the player's class."""
+        player_class = classes_by_name.get(player_class, None)
+        if player_class is None:
+            raise ValueError('Invalid player class name.')
+        self.player_class = player_class
+
+    player_class_name = property(get_player_class_name, set_player_class_name)
+
     def get_language(self):
         """Return the player's language.
 

addons/source-python/packages/source-python/players/teams.py

@@ -47,3 +47,9 @@
 
 # Set the managers as a frozenset, since it should never be modified
 team_managers = frozenset(team_managers)
+
+# Store the game specific player classes by name
+classes_by_name = _team_data.get('classes', {})
+
+# Store the game specific player classes by number
+classes_by_number = {number: name for name, number in classes_by_name.items()}