mirror of https://github.com/zulip/zulip.git
22 lines
720 B
Python
22 lines
720 B
Python
from argparse import ArgumentParser
|
|
from typing import Any
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
from analytics.lib.counts import do_drop_all_analytics_tables
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = """Clear analytics tables."""
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument('--force',
|
|
action='store_true',
|
|
help="Clear analytics tables.")
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
if options['force']:
|
|
do_drop_all_analytics_tables()
|
|
else:
|
|
raise CommandError("Would delete all data from analytics tables (!); use --force to do so.")
|