zulip/analytics/management/commands/clear_analytics_tables.py

26 lines
765 B
Python
Raw Normal View History

2016-10-06 00:42:56 +02:00
import sys
from argparse import ArgumentParser
from django.core.management.base import BaseCommand
from analytics.lib.counts import do_drop_all_analytics_tables
2016-10-06 00:42:56 +02:00
from typing import Any
2016-10-06 00:42:56 +02:00
class Command(BaseCommand):
help = """Clear analytics tables."""
2016-10-06 00:42:56 +02:00
def add_arguments(self, parser):
# type: (ArgumentParser) -> None
parser.add_argument('--force',
action='store_true',
help="Clear analytics tables.")
2016-10-06 00:42:56 +02:00
def handle(self, *args, **options):
# type: (*Any, **Any) -> None
if options['force']:
do_drop_all_analytics_tables()
2016-10-06 00:42:56 +02:00
else:
print("Would delete all data from analytics tables (!); use --force to do so.")
sys.exit(1)