darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-02 09:26:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

Generate 2FA secret code cryptographically install of using random number

Browse Source
This commit is contained in:
venusbb
2017-06-09 16:19:30 +01:00
parent 43aade9ab5
commit a0640bb803
2 changed files with 64 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/dao/users_dao.py
View File

@@ -3,7 +3,7 @@ from datetime import (datetime, timedelta)
from sqlalchemy import func
from app import db
from app.models import (User, VerifyCode)
from app import secrets
def _remove_values_for_keys_if_present(dict, keys):
for key in keys:
@@ -11,6 +11,13 @@ def _remove_values_for_keys_if_present(dict, keys):
def create_secret_code():
'''
L1 = []
for i in range(0, 5):
L1.append(secrets.randbelow(10)) #return cryptographically strong random number using secrets module
L2 = ''.join(map(str, L1))
return L2
'''
return ''.join(map(str, random.sample(range(9), 5)))