33 lines
1007 B
Python
33 lines
1007 B
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
import copy
|
|
import datetime
|
|
|
|
from keystone.common import kvs
|
|
from keystone import exception
|
|
from keystone import token
|
|
|
|
|
|
class Token(kvs.Base, token.Driver):
|
|
# Public interface
|
|
def get_token(self, token_id):
|
|
token = self.db.get('token-%s' % token_id)
|
|
if (token and (token['expires'] is None
|
|
or token['expires'] > datetime.datetime.now())):
|
|
return token
|
|
else:
|
|
raise exception.TokenNotFound(token_id=token_id)
|
|
|
|
def create_token(self, token_id, data):
|
|
data_copy = copy.deepcopy(data)
|
|
if 'expires' not in data:
|
|
data_copy['expires'] = self._get_default_expire_time()
|
|
self.db.set('token-%s' % token_id, data_copy)
|
|
return copy.deepcopy(data_copy)
|
|
|
|
def delete_token(self, token_id):
|
|
try:
|
|
return self.db.delete('token-%s' % token_id)
|
|
except KeyError:
|
|
raise exception.TokenNotFound(token_id=token_id)
|