EZPPFarm/pep.py
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add README

Browse Source
This commit is contained in:
Nyo
2016-06-11 18:43:27 +02:00
parent 93508624c4
commit 9d2fcf250f
5 changed files with 68 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
events/loginEvent.py
View File

@@ -46,6 +46,11 @@ def handle(tornadoRequest):
# Banned
raise exceptions.loginBannedException()
# 2FA check
if userHelper.check2FA(userID, requestIP) == True:
log.warning("Need 2FA check for user {}".format(loginData[0]))
raise exceptions.need2FAException()
# No login errors!
# Log user IP
userHelper.IPLog(userID, requestIP)
@@ -167,11 +172,14 @@ def handle(tornadoRequest):
except exceptions.banchoMaintenanceException:
# Bancho is in maintenance mode
responseData += serverPackets.notification("Our bancho server is in maintenance mode. Please try to login again later.")
responseData += serverPackets.loginError()
responseData += serverPackets.loginFailed()
except exceptions.banchoRestartingException:
# Bancho is restarting
responseData += serverPackets.notification("Bancho is restarting. Try again in a few minutes.")
responseData += serverPackets.loginError()
responseData += serverPackets.loginFailed()
except exceptions.need2FAException:
# User tried to log in from unknown IP
responseData += serverPackets.needVerification()
finally:
# Console and discord log
msg = "Bancho login request from {} for user {} ({})".format(requestIP, loginData[0], "failed" if err == True else "success")