redis_cache.py

Go to the documentation of this file.

# SPDX-FileCopyrightText: 2015 Eric Larson
#
# SPDX-License-Identifier: Apache-2.0
 
from __future__ import division
 
from datetime import datetime
from pip._vendor.cachecontrol.cache import BaseCache
 
 
class RedisCache(BaseCache):
 
    def __init__(self, conn):
        self.conn = conn
 
    def get(self, key):
        return self.conn.get(key)
 
    def set(self, key, value, expires=None):
        if not expires:
            self.conn.set(key, value)
        elif isinstance(expires, datetime):
            expires = expires - datetime.utcnow()
            self.conn.setex(key, int(expires.total_seconds()), value)
        else:
            self.conn.setex(key, expires, value)
 
    def delete(self, key):
        self.conn.delete(key)
 
    def clear(self):
        """Helper for clearing all the keys in a database. Use with
        caution!"""
        for key in self.conn.keys():
            self.conn.delete(key)
 
    def close(self):
        """Redis uses connection pooling, no need to close the connection."""
        pass