open-toontown/otp/login/AstronLoginScreen.py

17 lines
577 B
Python

from direct.directnotify import DirectNotifyGlobal
from direct.distributed.MsgTypes import *
from otp.login.LoginScreen import LoginScreen
class AstronLoginScreen(LoginScreen):
notify = DirectNotifyGlobal.directNotify.newCategory('AstronLoginScreen')
def handleWaitForLoginResponse(self, msgType, di):
if msgType == CLIENT_HELLO_RESP:
# Now we can start the heartbeat:
self.cr.startHeartbeat()
self.cr.astronLoginManager.handleRequestLogin(self.doneEvent)
else:
self.cr.handleMessageType(msgType, di)