Merge pull request #6 from JamesRitchie/default_settings

Provide default settings

rest_framework_expiring_authtoken/models.py

@@ -4,11 +4,12 @@
     ExpiringToken
 """
 
-from django.conf import settings
 from django.utils import timezone
 
 from rest_framework.authtoken.models import Token
 
+from rest_framework_expiring_authtoken.settings import token_settings
+
 
 class ExpiringToken(Token):
 
@@ -20,6 +21,6 @@ class Meta(object):
     def expired(self):
         """Return boolean indicating token expiration."""
         now = timezone.now()
-        if self.created < now - settings.EXPIRING_TOKEN_LIFESPAN:
+        if self.created < now - token_settings.EXPIRING_TOKEN_LIFESPAN:
             return True
         return False

rest_framework_expiring_authtoken/settings.py

@@ -0,0 +1,29 @@
+"""
+Provides access to settings.
+
+Returns defaults if not set.
+"""
+from datetime import timedelta
+
+from django.conf import settings
+
+
+class TokenSettings(object):
+
+    """Provides settings as defaults for working with tokens."""
+
+    @property
+    def EXPIRING_TOKEN_LIFESPAN(self):
+        """
+        Return the allowed lifespan of a token as a TimeDelta object.
+
+        Defaults to 30 days.
+        """
+        try:
+            val = settings.EXPIRING_TOKEN_LIFESPAN
+        except AttributeError:
+            val = timedelta(days=30)
+
+        return val
+
+token_settings = TokenSettings()

tests/settings.py

@@ -31,5 +31,3 @@
 }
 
 ROOT_URLCONF = 'tests.urls'
-
-EXPIRING_TOKEN_LIFESPAN = timedelta(days=30)