school_api.session.memorystorage

Complexity

Total Complexity

10

Size/Duplication

Total Lines	46
Duplicated Lines	0 %

Importance

Changes

0

6 Methods

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

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


class MemoryStorage(SessionStorage):
    ''' 非持久化缓存 不推荐使用'''

    def __init__(self):
        self._data = {}

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

    def get(self, key, default=None):
        key = self.key_name(key)
        value = None
        data = self._data.get(key, default)
        if data:
            if data['expires_at'] is True or data['expires_at'] > time.time():
                # 为True时：永不过期， 大于当前时间为：不到过期时间
                value = data['value']
            else:
                self.delete(key)
        return value

    def set(self, key, value, ttl=None):

        if value is None:
            return

        key = self.key_name(key)
        expires_at = not ttl or time.time() + ttl
        data = {'value': value, 'expires_at': expires_at}
        self._data[key] = data

    def delete(self, key):
        key = self.key_name(key)
        self._data.pop(key, None)

    def expires_time(self, key):
        data = self._data.get(self.key_name(key))
        return data['expires_at'] - time.time()