pep.py/events/changeActionEvent.py

from objects import glob
from constants import clientPackets
from constants import serverPackets
from helpers import userHelper
from helpers import logHelper as log
from constants import actions

def handle(userToken, packetData):
	# Get usertoken data
	userID = userToken.userID
	username = userToken.username

	# Make sure we are not banned
	if userHelper.isBanned(userID):
		userToken.enqueue(serverPackets.loginBanned())
		return

	# Send restricted message if needed
	if not userToken.restricted:
		if userHelper.isRestricted(userID):
			userToken.setRestricted()

	# Change action packet
	packetData = clientPackets.userActionChange(packetData)

	# If we are not in spectate status but we're spectating someone, stop spectating
	#if userToken.spectating != 0 and userToken.actionID != actions.watching and userToken.actionID != actions.idle and userToken.actionID != actions.afk:
	#	userToken.stopSpectating()

	# If we are not in multiplayer but we are in a match, part match
	#if userToken.matchID != -1 and userToken.actionID != actions.multiplaying and userToken.actionID != actions.multiplayer and userToken.actionID != actions.afk:
	#	userToken.partMatch()

	# Update cached stats if our pp changedm if we've just submitted a score or we've changed gameMode
	if (userToken.actionID == actions.PLAYING or userToken.actionID == actions.MULTIPLAYING) or (userToken.pp != userHelper.getPP(userID, userToken.gameMode)) or (userToken.gameMode != packetData["gameMode"]):
		# Always update game mode, or we'll cache stats from the wrong game mode if we've changed it
		userToken.gameMode = packetData["gameMode"]
		userToken.updateCachedStats()

	# Always update action id, text and md5
	userToken.actionID = packetData["actionID"]
	userToken.actionText = packetData["actionText"]
	userToken.actionMd5 = packetData["actionMd5"]
	userToken.actionMods = packetData["actionMods"]

	# Enqueue our new user panel and stats to us and our spectators
	recipients = [userID]
	if len(userToken.spectators) > 0:
		recipients += userToken.spectators

	for i in recipients:
		if i == userID:
			# Save some loops
			token = userToken
		else:
			token = glob.tokens.getTokenFromUserID(i)

		if token is not None:
			# Force our own packet
			force = True if token.userID == userID else False
			token.enqueue(serverPackets.userPanel(userID, force))
			token.enqueue(serverPackets.userStats(userID, force))

	# Console output
	log.info("{} changed action: {} [{}][{}]".format(username, str(userToken.actionID), userToken.actionText, userToken.actionMd5))