2
0

login_login_2fa.py 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. from .tool.func import *
  2. async def login_login_2fa():
  3. with get_db_connect() as conn:
  4. curs = conn.cursor()
  5. # email 2fa
  6. # pw 2fa
  7. # q_a 2fa
  8. if not (flask.session and 'login_id' in flask.session):
  9. return redirect(conn, '/user')
  10. ip = ip_check()
  11. if ip_or_user(ip) == 0:
  12. return redirect(conn, '/user')
  13. if (await ban_check(None, 'login'))[0] == 1:
  14. return await re_error(conn, 0)
  15. if flask.request.method == 'POST':
  16. if await captcha_post(conn, flask.request.form.get('g-recaptcha-response', flask.request.form.get('g-recaptcha', ''))) == 1:
  17. return await re_error(conn, 13)
  18. user_agent = flask.request.headers.get('User-Agent', '')
  19. user_id = flask.session['login_id']
  20. user_pw = flask.request.form.get('pw', '')
  21. curs.execute(db_change('select data from user_set where name = "2fa_pw" and id = ?'), [user_id])
  22. user_1 = curs.fetchall()
  23. if user_1:
  24. curs.execute(db_change('select data from user_set where name = "2fa_pw_encode" and id = ?'), [user_id])
  25. user_1 = user_1[0][0]
  26. user_2 = curs.fetchall()[0][0]
  27. pw_check_d = pw_check(conn, user_pw, user_1, user_2, user_id)
  28. if pw_check_d != 1:
  29. return await re_error(conn, 10)
  30. flask.session['id'] = user_id
  31. ua_plus(conn,
  32. user_id,
  33. ip,
  34. user_agent,
  35. get_time()
  36. )
  37. flask.session.pop('b_id', None)
  38. return redirect(conn, '/user')
  39. else:
  40. return await render_template(
  41. await get_lang('login'),
  42. '''
  43. <form method="post">
  44. <input class="__ON_INPUT__" placeholder="''' + await get_lang('2fa_password') + '''" name="pw" type="password">
  45. <hr class="main_hr">
  46. ''' + await captcha_get(conn) + '''
  47. <button class="__ON_BUTTON__" type="submit">''' + await get_lang('login') + '''</button>
  48. ''' + await http_warning() + '''
  49. </form>
  50. ''',
  51. 0,
  52. [['user', await get_lang('return')]]
  53. )