management: Move commands to all use ZulipBaseCommand.

analytics/management/commands/check_analytics_state.py

@@ -3,12 +3,12 @@ import time
 from datetime import timedelta
 from typing import Any, Dict
 
-from django.core.management.base import BaseCommand
 from django.utils.timezone import now as timezone_now
 from typing_extensions import override
 
 from analytics.lib.counts import ALL_COUNT_STATS, CountStat
 from analytics.models import installation_epoch
+from zerver.lib.management import ZulipBaseCommand
 from zerver.lib.timestamp import TimeZoneNotUTCError, floor_to_day, floor_to_hour, verify_UTC
 from zerver.models import Realm
 
@@ -20,7 +20,7 @@ states = {
 }
 
 
-class Command(BaseCommand):
+class Command(ZulipBaseCommand):
     help = """Checks FillState table.
 
     Run as a cron job that runs every hour."""

analytics/management/commands/clear_analytics_tables.py

@@ -1,13 +1,14 @@
 from argparse import ArgumentParser
 from typing import Any
 
-from django.core.management.base import BaseCommand, CommandError
+from django.core.management.base import CommandError
 from typing_extensions import override
 
 from analytics.lib.counts import do_drop_all_analytics_tables
+from zerver.lib.management import ZulipBaseCommand
 
 
-class Command(BaseCommand):
+class Command(ZulipBaseCommand):
     help = """Clear analytics tables."""
 
     @override

analytics/management/commands/clear_single_stat.py

@@ -1,13 +1,14 @@
 from argparse import ArgumentParser
 from typing import Any
 
-from django.core.management.base import BaseCommand, CommandError
+from django.core.management.base import CommandError
 from typing_extensions import override
 
 from analytics.lib.counts import ALL_COUNT_STATS, do_drop_single_stat
+from zerver.lib.management import ZulipBaseCommand
 
 
-class Command(BaseCommand):
+class Command(ZulipBaseCommand):
     help = """Clear analytics tables."""
 
     @override

analytics/management/commands/populate_analytics_db.py

@@ -3,7 +3,6 @@ from datetime import timedelta
 from typing import Any, Dict, List, Mapping, Type, Union
 
 from django.core.files.uploadedfile import UploadedFile
-from django.core.management.base import BaseCommand
 from django.utils.timezone import now as timezone_now
 from typing_extensions import TypeAlias, override
 
@@ -21,6 +20,7 @@ from analytics.models import (
 from zerver.actions.create_realm import do_create_realm
 from zerver.actions.users import do_change_user_role
 from zerver.lib.create_user import create_user
+from zerver.lib.management import ZulipBaseCommand
 from zerver.lib.storage import static_path
 from zerver.lib.stream_color import STREAM_ASSIGNMENT_COLORS
 from zerver.lib.timestamp import floor_to_day
@@ -38,7 +38,7 @@ from zerver.models import (
 from zerver.models.groups import SystemGroups
 
 
-class Command(BaseCommand):
+class Command(ZulipBaseCommand):
     help = """Populates analytics tables with randomly generated data."""
 
     DAYS_OF_DATA = 100

analytics/management/commands/update_analytics_counts.py

@@ -5,19 +5,18 @@ from datetime import timezone
 from typing import Any, Dict
 
 from django.conf import settings
-from django.core.management.base import BaseCommand
 from django.utils.dateparse import parse_datetime
 from django.utils.timezone import now as timezone_now
 from typing_extensions import override
 
 from analytics.lib.counts import ALL_COUNT_STATS, logger, process_count_stat
-from zerver.lib.management import abort_unless_locked
+from zerver.lib.management import ZulipBaseCommand, abort_unless_locked
 from zerver.lib.remote_server import send_server_data_to_push_bouncer
 from zerver.lib.timestamp import floor_to_hour
 from zerver.models import Realm
 
 
-class Command(BaseCommand):
+class Command(ZulipBaseCommand):
     help = """Fills Analytics tables.
 
     Run as a cron job that runs every hour."""