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.getAllowed(userID) == 0:
		userToken.enqueue(serverPackets.loginBanned())
		return

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

	# 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"]):
		log.debug("!!!! UPDATING CACHED STATS !!!!")
		# 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 != None:
			token.enqueue(serverPackets.userPanel(userID))
			token.enqueue(serverPackets.userStats(userID))

	# TODO: Enqueue all if we've changed game mode, (maybe not needed because it's cached)
	#glob.tokens.enqueueAll(serverPackets.userPanel(userID))
	#glob.tokens.enqueueAll(serverPackets.userStats(userID))

	# Send osu!direct alert if needed
	# NOTE: Remove this when osu!direct will be fixed
	if userToken.actionID == actions.osuDirect and userToken.osuDirectAlert == False:
		userToken.osuDirectAlert = True
		userToken.enqueue(serverPackets.sendMessage("FokaBot", userToken.username, "Sup! osu!direct works, kinda. To download a beatmap, you have to click the \"View listing\" button (the last one) instead of \"Download\". However, if you are on the stable (fallback) branch, it should work also with the \"Download\" button. We'll fix that bug as soon as possibleTM."))


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