fixed pep 8 issues

This commit is contained in:
venusbb
2017-06-13 18:11:13 +01:00
parent 8188dfa7d3
commit 332fe14680

View File

@@ -1,9 +1,9 @@
import random from random import (SystemRandom)
from datetime import (datetime, timedelta) from datetime import (datetime, timedelta)
from sqlalchemy import func from sqlalchemy import func
from app import db from app import db
from app.models import (User, VerifyCode) from app.models import (User, VerifyCode)
from app import secrets
def _remove_values_for_keys_if_present(dict, keys): def _remove_values_for_keys_if_present(dict, keys):
for key in keys: for key in keys:
@@ -11,12 +11,7 @@ def _remove_values_for_keys_if_present(dict, keys):
def create_secret_code(): def create_secret_code():
L1 = [] return ''.join(map(str, [SystemRandom().randrange(10) for i in range(5)]))
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
def save_user_attribute(usr, update_dict={}): def save_user_attribute(usr, update_dict={}):