Merge pull request #39 from wieden-kennedy/master

Allow numeric codes (should also close issue #38)

docs/settings.rst

@@ -38,6 +38,9 @@ django-nopassword settings
 
 .. attribute:: NOPASSWORD_TWILIO_AUTH_TOKEN
     Account secret for Twilio
+
+.. attribute:: NOPASSWORD_NUMERIC_CODES
+    A boolean flag if set to True, codes will contain numeric characters only (0-9). Default: False
 
 Django settings used in django-nopassword
 +++++++++++++++++++++++++++++++++++++++++

nopassword/backends/__init__.py

@@ -1,3 +1,7 @@
 # -*- coding: utf-8 -*-
 from .email import EmailBackend
-from .sms import TwilioBackend
+
+try:
+    from .sms import TwilioBackend
+except ImportError:
+    pass

nopassword/models.py

@@ -65,4 +65,8 @@ def generate_code(cls, length=20):
         m = getattr(hashlib, hash_algorithm)()
         m.update(getattr(settings, 'SECRET_KEY', None).encode('utf-8'))
         m.update(os.urandom(16))
-        return m.hexdigest()[:length]
+        if getattr(settings, 'NOPASSWORD_NUMERIC_CODES', False):
+            hashed = str(int(m.hexdigest(), 16))[-length:]
+        else:
+            hashed = m.hexdigest()[:length]
+        return hashed

tests/test_models.py

@@ -35,6 +35,12 @@ def test_shorter_code(self):
         self.code = LoginCode.create_code_for_user(self.user)
         self.assertEqual(len(self.code.code), 8)
 
+    @override_settings(NOPASSWORD_NUMERIC_CODES=True)
+    def test_numeric_code(self):
+        self.code = LoginCode.create_code_for_user(self.user)
+        self.assertEqual(len(self.code.code), 20)
+        self.assertTrue(self.code.code.isdigit())
+
     def test_next_value(self):
         self.code = LoginCode.create_code_for_user(self.user, next='/secrets/')
         self.assertEqual(self.code.next, '/secrets/')