# ../core/players/weapons.py ''' $Rev: 585 $ $LastChangedBy: satoon101 $ $LastChangedDate: 2011-10-30 18:48:42 -0400 (Sun, 30 Oct 2011) $ ''' # ============================================================================= # >> IMPORTS # ============================================================================= # EventScripts Imports import es as _es import gamethread as _gamethread from playerlib import getPlayer as _getPlayer from weaponlib import getWeaponNameList as _getWeaponNameList # SPE import spe as _spe # GunGame Imports from gungame51.core import GunGameError # Weapons from gungame51.core.weapons.shortcuts import get_level_weapon as _level_weapon # ============================================================================= # >> GLOBALS # ============================================================================= list_pWeapons = _getWeaponNameList('#primary') list_sWeapons = _getWeaponNameList('#secondary') list_allWeapons = _getWeaponNameList() # ============================================================================= # >> CLASSES # ============================================================================= class PlayerWeapons(object): """Adds weapon methods to the BasePlayer class.""" @property def weapon(self): ''' Return the weapon name ''' return self.get_weapon() def get_weapon(self): return _level_weapon(self.level) def give_weapon(self): ''' Gives a player their current levels weapon. ''' error = None # Make sure player is on a team if self.team < 2: error = ('Unable to give player weapon (%s): ' % self.userid + 'is not on a team.') # Make sure player is alive elif _getPlayer(self.userid).isdead: error = ('Unable to give player weapon (%s): ' % self.userid + 'is not alive.') # Error ? if error: raise GunGameError(error) # Knife ? if self.weapon == 'knife': # Make them use their knife _es.server.queuecmd('es_xsexec %s "use weapon_knife"' % ( self.userid)) # If there is a level below the user's current level if self.level > 1: # Strip previous weapons self.strip_weapons(_level_weapon(self.level - 1)) else: self.strip() # Nade ? elif self.weapon == 'hegrenade': # Give them a grenade. given_weapon = _spe.giveNamedItem(self.userid, "weapon_hegrenade") # Make them use the grenade _es.server.queuecmd('es_xsexec %s "use weapon_hegrenade"' % ( self.userid)) # If there is a level below the user's current level if self.level > 1: # Strip previous weapons self.strip_weapons(_level_weapon(self.level - 1)) else: self.strip() else: # Player owns this weapon. if _spe.ownsWeapon(self.userid, "weapon_%s" % self.weapon): # Make them use it. If we don't do this, a very # strange bug comes up which prevents the player # from getting their current level's weapon after # being stripped, _es.server.queuecmd('es_xsexec %s "use weapon_%s"' % (self.userid, self.weapon)) return # Player DOES NOT own this weapon. else: # Retrieve a list of all weapon names in the player's # possession playerWeapons = _spe.getWeaponDict(self.userid) if playerWeapons: # See if there is a primary weapon in the list of weapons pWeapon = set(playerWeapons.keys()).intersection( list_pWeapons) # See if there is a primary weapon in the list of weapons sWeapon = set(playerWeapons.keys()).intersection( list_sWeapons) # Set up the weapon to strip weapToStrip = None # Strip secondary weapon ? if 'weapon_' + self.weapon in list_sWeapons and sWeapon: weapToStrip = sWeapon.pop() # Strip primary weapon ? elif 'weapon_' + self.weapon in list_pWeapons and pWeapon: weapToStrip = pWeapon.pop() if weapToStrip: # Make them drop the weapon _spe.dropWeapon(self.userid, weapToStrip) # Now remove it _spe.removeEntityByInstance(playerWeapons [weapToStrip]["instance"]) # Now give them the weapon and save the weapon instance given_weapon = _spe.giveNamedItem(self.userid, "weapon_%s" % self.weapon) # Retrieve the weapon instance of the weapon they "should" own weapon_check = _spe.ownsWeapon(self.userid, "weapon_%s" % self.weapon) # Make sure that the player owns the weapon we gave them if weapon_check != given_weapon: # Remove the given weapon since the player does not own it _spe.removeEntityByInstance(given_weapon) # If they don't have the right weapon, fire give_weapon() if not weapon_check: self.give_weapon() return _es.server.queuecmd('es_xsexec %s "use weapon_%s"' % (self.userid, self.weapon)) def give(self, weapon, useWeapon=False, strip=False): ''' Gives a player the specified weapon. Weapons given by this method will not be stripped by gg_dead_strip. Setting strip to True will make it strip the weapon currently held in the slot you are trying to give to. ''' # Format weapon weapon = 'weapon_%s' % str(weapon).replace('weapon_', '') # Check if weapon is valid if weapon not in list_pWeapons + list_sWeapons + [ 'weapon_hegrenade', 'weapon_flashbang', 'weapon_smokegrenade']: raise ValueError('Unable to give "%s": ' % weapon[7:] + 'is not a valid weapon') # Add weapon to strip exceptions so gg_dead_strip will not # strip the weapon if int(_es.ServerVar('gg_dead_strip')): self.stripexceptions.append(weapon[7:]) # Delay removing the weapon long enough for gg_dead_strip to fire _gamethread.delayed(0.10, self.stripexceptions.remove, (weapon[7:])) # If the player owns the weapon and the player is not being given a # second flashbang, stop here if (_spe.ownsWeapon(self.userid, weapon) and not (weapon == "weapon_flashbang" and _getPlayer(self.userid).getFB() < 2)): return # Strip the weapon ? if strip: # Retrieve a list of all weapon names in the player's possession playerWeapons = _spe.getWeaponDict(self.userid) if playerWeapons: # See if there is a primary weapon in the list of weapons pWeapon = set(playerWeapons.keys()).intersection(list_pWeapons) # See if there is a primary weapon in the list of weapons sWeapon = set(playerWeapons.keys()).intersection(list_sWeapons) stripWeapon = None # Holding a primary weapon ? if weapon in list_pWeapons and pWeapon: stripWeapon = pWeapon.pop() # Holding a secondary weapon ? elif weapon in list_sWeapons and sWeapon: stripWeapon = sWeapon.pop() # Strip the weapon if stripWeapon: # Make them drop the weapon _spe.dropWeapon(self.userid, stripWeapon) # Remove the weapon _spe.removeEntityByInstance(playerWeapons[stripWeapon][ "instance"]) # Give the player the weapon _spe.giveNamedItem(self.userid, weapon) if useWeapon: _es.server.queuecmd('es_xsexec %s "use %s"' % (self.userid, weapon)) def strip(self, levelStrip=False, exceptions=[]): ''' * Strips/removes all weapons from the player minus the knife and their current levels weapon. * If True is specified, then their level weapon is also stripped. * Exceptions can be entered in list format, and anything in the exceptions will not be stripped. ''' # Retrieve a dictionary of the player's weapons pWeapons = _spe.getWeaponDict(self.userid) if not pWeapons: return for weapon in pWeapons: if ((self.weapon == weapon[7:] and not levelStrip) or weapon in ('weapon_knife', 'weapon_c4') or weapon[7:] in exceptions): continue _spe.dropWeapon(self.userid, weapon) _spe.removeEntityByInstance(pWeapons[weapon]["instance"]) def strip_weapons(self, stripWeapons): ''' Strips a list of weapons from a player. (Used primarily for selective weapon removal when a player gets a new weapon) stripWeapons must be a list. ''' # Get the player's current held weapons playerWeapons = _spe.getWeaponDict(self.userid) # Format the stripWeapons list for all names to start with "weapon_" stripWeapons = [w if w.startswith("weapon_") else "weapon_%s" % w for w in stripWeapons] # Insure that the player owns the weapons by using 2 sets intersection remWeapons = set(stripWeapons).intersection( set(playerWeapons.keys())).difference(set(["weapon_knife"])) # Loop through any weapons to strip for stripWeapon in remWeapons: # Drop the weapon _spe.dropWeapon(self.userid, stripWeapon) # Remove the weapon _spe.removeEntityByInstance(playerWeapons[stripWeapon]["instance"])