View source for Scripts:ClanTkProtection

This script is a modified version of the default tkpunish script.

Clanmembers is a list of current members that will be protected
This is the only code changed besides the declare of the list.

<pre>
def onPlayerKilled(victim, attacker, weapon, assists, object):
       
        string1 = attacker.getName()
        #Is the attacker protected against punishments? If so, dont handle the event and return
        for ncount in clanmembers:
            if string1 == ncount: return
</pre>

<pre>
# tk punish system


import bf2
import host
from bf2 import g_debug

TK_PUNISH_TIME = 20

TK_PUNISH_COMMANDID = 100
TK_FORGIVE_COMMANDID = 101

hasPendingTks = {}
updateTimer = None

clanmembers = ["|TAG|Bill","|TAG|John","|TAG|Blue"]#clan members to protect

class TkData:
	def __init__(self):
		self.punished = 0
		self.pending = []
		self.lastTKedBy = None


def init():
	if g_debug: print "initializing tk-punish script"

	host.registerHandler('PlayerConnect', onPlayerConnect, 1)
	host.registerHandler('PlayerKilled', onPlayerKilled)
	host.registerHandler('ClientCommand', onClientCommand)

	# Connect already connected players if reinitializing
	for p in bf2.playerManager.getPlayers():
		onPlayerConnect(p)
	
	checkEnable()


def checkEnable():
	global updateTimer
	
	if not bf2.serverSettings.getTKPunishEnabled():
		if updateTimer:
			updateTimer.destroy()
			updateTimer = None
		return False
	else:
		if not updateTimer:
			#if g_debug: print "Started timer"
			updateTimer = bf2.Timer(onUpdate, 10, 1)
			updateTimer.setRecurring(10)
			
		return True
	

def onUpdate(data):
	if not checkEnable(): return
	global hasPendingTks

	#if len(hasPendingTks) > 0:
	#	if g_debug: print "update: Has pending tks!"
	#else:
	#	if g_debug: print "update:No pending tks."
		

	currentTime = host.timer_getWallTime()

	newPending = {}
	for attacker in hasPendingTks.iterkeys():
		newList = []
		#if g_debug: print "attacker pending list:", len(attacker.tkData.pending)
		for tk in attacker.tkData.pending:
			tkDate = tk[0]
			tkVictim = tk[1]
			#if g_debug: print "check %d %s" % (tkDate, tkVictim.getName())

			# if there is a pending TK with time expired
			if tkDate + TK_PUNISH_TIME < currentTime:
				if bf2.serverSettings.getTKPunishByDefault():
					attacker.tkData.punished += 1
					checkPunishLimit(attacker)
				#	if g_debug: print "Default punished ", attacker.getName()
				else:
				#	if g_debug: print "Timeouted pending tk for ", attacker.getName()
					pass

				# remove player from global list, if it was the last tk
				if len(attacker.tkData.pending) == 1:
					pass
				else:
					newPending[attacker] = 1
			
			else:
				newList += [tk]
				newPending[attacker] = 1
		
		attacker.tkData.pending = newList

	hasPendingTks = newPending
	
						
def checkPunishLimit(player):
	#if g_debug: print "player %s has %d punishes" % (player.getName(), player.tkData.punished)

	if player.tkData.punished >= bf2.serverSettings.getTKNumPunishToKick():
		if g_debug: print "%s reached punish limit" % player.getName()
		if g_debug: print "Banning player!"
		
		result = host.rcon_invoke("admin.banPlayer " + str(player.index) + " Round")
		
	


# event actions	

def onPlayerConnect(player):
	player.tkData = TkData()

def onClientCommand(command, issuer, args):
	if not checkEnable(): return
	
	#if g_debug: print "received client command"
	if command == TK_PUNISH_COMMANDID:
		executePunishAction(issuer, True)
	elif command == TK_FORGIVE_COMMANDID:
		executePunishAction(issuer, False)
		

def onPlayerKilled(victim, attacker, weapon, assists, object):
       
        string1 = attacker.getName()
        #Is the attacker protected against punishments? If so, dont handle the event and return
        for ncount in clanmembers:
            if string1 == ncount: return
        
	if not checkEnable(): return

	if not victim or not attacker or victim == attacker or victim.getTeam() != attacker.getTeam(): return
	
	# ok, we have a teamkill
	
	currentTime = host.timer_getWallTime()
	attacker.tkData.pending += [(currentTime, victim)]
	hasPendingTks[attacker] = 1
		
	victim.tkData.lastTKedBy = attacker # can only punish / forgive the last TK'er
	
	bf2.gameLogic.sendClientCommand(victim.index, 100, (2, victim.index, attacker.index, TK_PUNISH_TIME))


def executePunishAction(victim, punish):
	if not checkEnable() or not victim or not victim.isValid(): return
	attacker = victim.tkData.lastTKedBy
	if attacker == None or not attacker.isValid(): return

	global hasPendingTks

	if punish:
		if g_debug: print "%s tkPunished %s" % (victim.getName(), attacker.getName())
	else:
		if g_debug: print "%s tkForgive %s" % (victim.getName(), attacker.getName())

	currentTime = host.timer_getWallTime()
	
	newList = []
	for tk in attacker.tkData.pending:
		tkDate = tk[0]
		tkVictim = tk[1]
		
		# if the attacker had a pending tk, and time has not run out
		if tkVictim == victim and tkDate + TK_PUNISH_TIME > currentTime:
			if punish:
				attacker.tkData.punished += 1
				bf2.gameLogic.sendClientCommand(-1, 100, (0, victim.index, attacker.index)) # 100 = tkpunish event, 0 = punish
				if g_debug: print "%s punished %s, now has %d punishes" % (victim.getName(), attacker.getName(), attacker.tkData.punished)
			else:
				bf2.gameLogic.sendClientCommand(-1, 100, (1, victim.index, attacker.index)) # 100 = tkpunish event, 1 = forgive
				if g_debug: print "%s forgave %s" % (victim.getName(), attacker.getName())
		else:
			newList += [tk]
			
	attacker.tkData.pending = newList
	
	if len(attacker.tkData.pending) == 0:
		del hasPendingTks[attacker]
	
	checkPunishLimit(attacker)
</pre>