1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
# -*- coding: utf-8 -*-
# Author: AxiaCore S.A.S. https://axiacore.com
from django.conf import settings
from django.core.cache import DEFAULT_CACHE_ALIAS, caches
from django.core.cache.backends.base import InvalidCacheBackendError
from django.core.management.base import BaseCommand, CommandError
from django_extensions.management.utils import signalcommand
class Command(BaseCommand):
"""A simple management command which clears the site-wide cache."""
help = "Fully clear site-wide cache."
def add_arguments(self, parser):
parser.add_argument("--cache", action="append", help="Name of cache to clear")
parser.add_argument(
"--all",
"-a",
action="store_true",
default=False,
dest="all_caches",
help="Clear all configured caches",
)
@signalcommand
def handle(self, cache, all_caches, *args, **kwargs):
if not cache and not all_caches:
cache = [DEFAULT_CACHE_ALIAS]
elif cache and all_caches:
raise CommandError("Using both --all and --cache is not supported")
elif all_caches:
cache = getattr(settings, "CACHES", {DEFAULT_CACHE_ALIAS: {}}).keys()
for key in cache:
try:
caches[key].clear()
except InvalidCacheBackendError:
self.stderr.write('Cache "%s" is invalid!\n' % key)
else:
self.stdout.write('Cache "%s" has been cleared!\n' % key)
|
