school_api.session.redisstorage

Complexity

Total Complexity

9

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	RedisStorage.expires_time()	0	2	1
A	RedisStorage.delete()	0	3	1
A	RedisStorage.get()	0	6	2
A	RedisStorage.key_name()	0	2	1
A	RedisStorage.set()	0	6	2
A	RedisStorage.__init__()	0	4	2

# -*- coding: utf-8 -*-
from __future__ import absolute_import, unicode_literals
import json
from school_api.session import SessionStorage
from school_api.utils import to_text


class RedisStorage(SessionStorage):

    def __init__(self, redis):
        for method_name in ('get', 'set', 'delete'):
            assert hasattr(redis, method_name)
        self.redis = redis

    def key_name(self, key):
        return 'school:{0}'.format(key)

    def get(self, key, default=None):
        key = self.key_name(key)
        value = self.redis.get(key)
        if value is None:
            return default
        return json.loads(to_text(value))

    def set(self, key, value, ttl=None):
        if value is None:
            return
        key = self.key_name(key)
        value = json.dumps(value)
        self.redis.set(key, value, ex=ttl)

    def delete(self, key):
        key = self.key_name(key)
        self.redis.delete(key)

    def expires_time(self, key):
        return self.redis.ttl(self.key_name(key))