memegen.routes._cache

Complexity

Total Complexity

16

Size/Duplication

Total Lines	71
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
B	Cache._skip_cache()	0	19	7
A	Cache.name()	0	3	2
A	Cache.__init__()	0	4	1
A	Cache.add()	0	9	3
A	Cache.get()	0	16	3

import yorm
from yorm.types import List, Object
from profanityfilter import ProfanityFilter
import log

profanity_filter = ProfanityFilter()
profanity_filter.remove_word("damn")


@yorm.attr(items=List.of_type(Object))
@yorm.sync("data/cache/{self.name}.yml", auto_resolve=True)
class Cache:

    SIZE = 100

    def __init__(self, filtered=True):
        self.items = []
        self.disabled = False
        self.filtered = filtered

    @property
    def name(self):
        return 'filtered' if self.filtered else 'unfiltered'

    def add(self, **kwargs):
        if self._skip_cache(kwargs):
            return

        log.info("Caching: %s", kwargs)

        self.items.insert(0, kwargs)
        while len(self.items) > self.SIZE:
            self.items.pop()

    def get(self, index):
        log.info("Getting cache index: %s", index)

        try:
            data = self.items[index]
        except IndexError:
            data = {}

        if not isinstance(data, dict):
            log.error("Invalid cache")
            self.items = []
            data = {}

        log.info("Retrieved cache: %s", data)

        return data

    def _skip_cache(self, kwargs):
        if self.disabled:
            log.debug("Caching disabled")
            return True

        if kwargs in self.items:
            log.debug("Already cached: %s", kwargs)
            return True

        if self.filtered:

            if kwargs['key'] == 'custom' or kwargs.get('alt'):
                log.debug("Skipped caching of custom background: %s", kwargs)
                return True
            if profanity_filter.is_profane(kwargs['path']):
                log.debug("Skipped caching of profane content: %s", kwargs)
                return True

        return False