From ad7f099aefc17d04a2a04deb7fd3055adc8cd84a Mon Sep 17 00:00:00 2001
From: Todd Willey <todd@ansolabs.com>
Date: Tue, 27 Jul 2010 01:03:05 -0400
Subject: Flag for SessionToken ttl setting.

---
 nova/compute/model.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'nova/compute')

diff --git a/nova/compute/model.py b/nova/compute/model.py
index bae93b6c1..212830d3c 100644
--- a/nova/compute/model.py
+++ b/nova/compute/model.py
@@ -235,6 +235,7 @@ class SessionToken(datastore.BasicModel):
 
     def __init__(self, session_token):
         self.token = session_token
+        self.default_ttl = FLAGS.auth_token_ttl
         super(SessionToken, self).__init__()
 
     @property
@@ -243,7 +244,7 @@ class SessionToken(datastore.BasicModel):
 
     def default_state(self):
         now = datetime.datetime.utcnow()
-        diff = datetime.timedelta(hours=1)
+        diff = datetime.timedelta(seconds=self.default_ttl)
         expires = now + diff
         return {'user': None, 'session_type': None, 'token': self.token,
                 'expiry': expires.strftime(utils.TIME_FORMAT)}
@@ -282,7 +283,7 @@ class SessionToken(datastore.BasicModel):
     def update_expiry(self, **kwargs):
         """updates the expirty attribute, but doesn't save"""
         if not kwargs:
-            kwargs['hours'] = 1
+            kwargs['seconds'] = self.default_ttl
         time = datetime.datetime.utcnow()
         diff = datetime.timedelta(**kwargs)
         expires = time + diff
@@ -293,6 +294,13 @@ class SessionToken(datastore.BasicModel):
         expires = utils.parse_isotime(self['expiry'])
         return expires <= now
 
+    def ttl(self):
+        """number of seconds remaining before expiration"""
+        now = datetime.datetime.utcnow()
+        expires = utils.parse_isotime(self['expiry'])
+        delta = expires - now
+        return (delta.seconds + (delta.days * 24 * 3600))
+
 
 if __name__ == "__main__":
     import doctest
-- 
cgit