zulip/zerver/management/commands/fill_memcached_caches.py

25 lines
729 B
Python
Raw Normal View History

from argparse import ArgumentParser
from typing import Any, Optional
from django.core.management.base import BaseCommand
from typing_extensions import override
from zerver.lib.cache_helpers import cache_fillers, fill_remote_cache
class Command(BaseCommand):
@override
def add_arguments(self, parser: ArgumentParser) -> None:
parser.add_argument(
"--cache", help="Populate one specific cache", choices=cache_fillers.keys()
)
@override
def handle(self, *args: Any, **options: Optional[str]) -> None:
if options["cache"] is not None:
fill_remote_cache(options["cache"])
return
for cache in cache_fillers:
fill_remote_cache(cache)