mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 10:02:46 +00:00
87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: qabot
|
||
@Author：虎
|
||
@file： file_cache.py
|
||
@date：2023/9/11 15:58
|
||
@desc: 文件缓存
|
||
"""
|
||
import datetime
|
||
import math
|
||
import os
|
||
import time
|
||
|
||
from diskcache import Cache
|
||
from django.core.cache.backends.base import BaseCache
|
||
|
||
|
||
class FileCache(BaseCache):
|
||
def __init__(self, dir, params):
|
||
super().__init__(params)
|
||
self._dir = os.path.abspath(dir)
|
||
self._createdir()
|
||
self.cache = Cache(self._dir)
|
||
|
||
def _createdir(self):
|
||
old_umask = os.umask(0o077)
|
||
try:
|
||
os.makedirs(self._dir, 0o700, exist_ok=True)
|
||
finally:
|
||
os.umask(old_umask)
|
||
|
||
def add(self, key, value, timeout=None, version=None):
|
||
expire = timeout if isinstance(timeout, int) or isinstance(timeout,
|
||
float) or timeout is None else timeout.total_seconds()
|
||
return self.cache.add(self.get_key(key, version), value=value, expire=expire)
|
||
|
||
def set(self, key, value, timeout=None, version=None):
|
||
expire = timeout if isinstance(timeout, int) or isinstance(timeout,
|
||
float) or timeout is None else timeout.total_seconds()
|
||
return self.cache.set(self.get_key(key, version), value=value, expire=expire)
|
||
|
||
def get(self, key, default=None, version=None):
|
||
return self.cache.get(self.get_key(key, version), default=default)
|
||
|
||
@staticmethod
|
||
def get_key(key, version):
|
||
if version is None:
|
||
return f"default:{key}"
|
||
return f"{version}:{key}"
|
||
|
||
def delete(self, key, version=None):
|
||
return self.cache.delete(self.get_key(key, version))
|
||
|
||
def touch(self, key, timeout=None, version=None):
|
||
expire = timeout if isinstance(timeout, int) or isinstance(timeout,
|
||
float) else timeout.total_seconds()
|
||
|
||
return self.cache.touch(self.get_key(key, version), expire=expire)
|
||
|
||
def ttl(self, key, version=None):
|
||
"""
|
||
获取key的剩余时间
|
||
:param key: key
|
||
:return: 剩余时间
|
||
@param version:
|
||
"""
|
||
value, expire_time = self.cache.get(self.get_key(key, version), expire_time=True)
|
||
if value is None:
|
||
return None
|
||
return datetime.timedelta(seconds=math.ceil(expire_time - time.time()))
|
||
|
||
def clear_by_application_id(self, application_id):
|
||
delete_keys = []
|
||
for key in self.cache.iterkeys():
|
||
value = self.cache.get(key)
|
||
if (hasattr(value,
|
||
'application') and value.application is not None and value.application.id is not None and
|
||
str(
|
||
value.application.id) == application_id):
|
||
delete_keys.append(key)
|
||
for key in delete_keys:
|
||
self.cache.delete(key)
|
||
|
||
def clear_timeout_data(self):
|
||
for key in self.cache.iterkeys():
|
||
self.get(key)
|