Add a limit to the number of active 2fa codes that we create. At the moment that is set to 10.

This commit is contained in:
Rebecca Law
2017-02-15 16:18:05 +00:00
parent 9de88c50ba
commit 52342afe3f
5 changed files with 44 additions and 3 deletions

View File

@@ -82,6 +82,14 @@ def delete_user_verify_codes(user):
db.session.commit()
def count_user_verify_codes(user):
query = db.session.query(
func.count().label('count')
).filter(VerifyCode.user == user,
VerifyCode.expiry_datetime <= datetime.utcnow()).one()
return query.count
def get_user_by_id(user_id=None):
if user_id:
return User.query.filter_by(id=user_id).one()