Add TOTP capabilities to Auth

hiboo/models.py

@@ -11,6 +11,7 @@ import sqlalchemy
 import datetime
 import json
 import uuid
+import pyotp
 
 
 def log(category, value=None, comment=None, user=None, profile=None,
@@ -150,6 +151,14 @@ class Auth(db.Model):
     def check_password(self, password):
         return hash.pbkdf2_sha256.verify(password, self.value)
 
+    def set_otp_key(self):
+        self.value = pyotp.random_base32()
+
+    def check_totp(self, totp):
+        Totp = pyotp.TOTP(self.value)
+        return Totp.verify(totp)
+
+
 
 class Service(db.Model):
     """ A service is a client application (SP or RP typically).