Nemcio/CyberPanel
1
0
Fork 0
mirror of https://github.com/usmannasir/cyberpanel.git synced 2025-11-13 08:46:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

complete https://feedback.cyberpanel.net/b/requests/p/2fa-support-in-login

Browse Source
This commit is contained in:
Usman Nasir
2020-08-10 23:55:46 +05:00
parent e047da10b8
commit 6810ff1b87
9 changed files with 231 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
loginSystem/views.py
View File

@@ -91,8 +91,32 @@ def verifyLogin(request):
json_data = json.dumps(data)
return HttpResponse(json_data)
if admin.twoFA:
try:
twoinit = request.session['twofa']
except:
request.session['twofa'] = 0
data = {'userID': admin.pk, 'loginStatus': 2, 'error_message': "None"}
json_data = json.dumps(data)
response.write(json_data)
return response
if hashPassword.check_password(admin.password, password):
if admin.twoFA:
if request.session['twofa'] == 0:
import pyotp
totp = pyotp.TOTP(admin.secretKey)
del request.session['twofa']
logging.writeToFile(str(totp.now()))
if totp.verify(data['twofa']):
data = {'userID': 0, 'loginStatus': 0, 'error_message': "Invalid verification code."}
json_data = json.dumps(data)
response.write(json_data)
return response
request.session['userID'] = admin.pk
ipAddr = request.META.get('REMOTE_ADDR')