Fixed some issues with weapon restrictions on games other than CS:S/CS:GO.

Author: jordanbriere

addons/source-python/packages/source-python/weapons/restrictions.py

@@ -12,6 +12,7 @@
 # Source.Python Imports
 #   Core
 from core import AutoUnload
+from core import GAME_NAME
 from core import SOURCE_ENGINE
 #   Entities
 from entities.constants import INVALID_ENTITY_INDEX
@@ -172,9 +173,8 @@ def __init__(self):
         """Store all teams and their aliases for the current game."""
         super().__init__()
 
-        # Store all of the aliases except for spec/un
-        self.aliases = {
-            x: y for x, y in teams_by_name.items() if x not in ('un', 'spec')}
+        # Store all of the aliases
+        self.aliases = {**teams_by_name}
         self.update({x: set() for x in self.aliases.values()})
 
     def __getitem__(self, item):
@@ -389,17 +389,20 @@ def _on_weapon_bump(args):
         make_object(Player, args[0]), edict_from_pointer(args[1]).classname)
 
 
-@EntityPreHook(EntityCondition.is_player, 'buy_internal')
-def _on_weapon_purchase(args):
-    """Return whether the player is allowed to purchase the weapon."""
-    # TODO:
-    # In CS:GO it seems like the weapon isn't passed as a string anymore.
-    # Instead it's rather a pointer that might be NULL. If it's not NULL, the
-    # function sets it to some value:
-    #if ( a3 )
-    #        *(_DWORD *)a3 = v16;
-    return weapon_restriction_manager.on_player_purchasing_weapon(
-        make_object(Player, args[0]), args[1 if SOURCE_ENGINE != 'csgo' else 2])
+if GAME_NAME in ('css', 'csgo'):
+    @EntityPreHook(EntityCondition.is_player, 'buy_internal')
+    def _on_weapon_purchase(args):
+        """Return whether the player is allowed to purchase the weapon."""
+        # TODO:
+        # In CS:GO it seems like the weapon isn't passed as a string anymore.
+        # Instead it's rather a pointer that might be NULL. If it's not NULL,
+        # the function sets it to some value:
+        #if ( a3 )
+        #        *(_DWORD *)a3 = v16;
+        return weapon_restriction_manager.on_player_purchasing_weapon(
+            make_object(Player, args[0]),
+            args[1 if SOURCE_ENGINE != 'csgo' else 2]
+        )
 
 
 # =============================================================================