flyapps/fir_ser/common/cache/state.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# project: 12月 
# author: NinEveN
# date: 2021/12/22
import logging
import time

from django.core.cache import cache

logger = logging.getLogger(__name__)


class CacheBaseState(object):

    def __init__(self, key, value=time.time(), timeout=3600 * 24):
        self.key = f"CacheBaseState_{self.__class__.__name__}_{key}"
        self.value = value
        self.timeout = timeout
        self.active = False

    def get_state(self):
        return cache.get(self.key)

    def del_state(self):
        return cache.delete(self.key)

    def __enter__(self):
        if cache.get(self.key):
            return False
        else:
            cache.set(self.key, self.value, self.timeout)
            self.active = True
        return True

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.active:
            cache.delete(self.key)
        logger.info(f"cache base state __exit__ {exc_type}, {exc_val}, {exc_tb}")


class MigrateStorageState(CacheBaseState):
    ...


class CleanSignDataState(CacheBaseState):
    ...


class CleanAppSignDataState(CacheBaseState):
    ...


class CleanErrorBundleIdSignDataState(CacheBaseState):
    ...