|
3 | 3 | from collections.abc import Iterable |
4 | 4 | from datetime import date, timedelta |
5 | 5 | from decimal import Decimal |
| 6 | +from typing import override |
6 | 7 |
|
7 | 8 | from django.core.management import CommandParser, call_command |
8 | 9 | from django.core.management.base import BaseCommand |
|
30 | 31 |
|
31 | 32 |
|
32 | 33 | class Command(BaseCommand): |
| 34 | + @override |
33 | 35 | def add_arguments(self, parser: CommandParser) -> None: |
34 | 36 | parser.add_argument( |
35 | 37 | "--dataset", |
@@ -63,8 +65,8 @@ def handle(self, *args, **options) -> None: |
63 | 65 | random.seed(selected_seed) |
64 | 66 | self.stdout.write(f"Seed used: {selected_seed}") |
65 | 67 |
|
66 | | - should_seed_cms = dataset in ("cms", "both") |
67 | | - should_seed_metrics = dataset in ("metrics", "both") |
| 68 | + should_seed_cms = dataset in {"cms", "both"} |
| 69 | + should_seed_metrics = dataset in {"metrics", "both"} |
68 | 70 |
|
69 | 71 | counts: dict[str, int] = { |
70 | 72 | "Theme": 0, |
@@ -205,8 +207,8 @@ def _seed_time_series_rows( |
205 | 207 | for geography in geographies: |
206 | 208 | for day_offset in range(days): |
207 | 209 | current_date = start_date + timedelta(days=day_offset) |
208 | | - base_value = random.uniform(5.0, 250.0) |
209 | | - metric_value = round(base_value + random.uniform(-10.0, 10.0), 2) |
| 210 | + base_value = random.uniform(5.0, 250.0) # noqa: S311 |
| 211 | + metric_value = round(base_value + random.uniform(-10.0, 10.0), 2) # noqa: S311 |
210 | 212 | epidemiological_week = current_date.isocalendar().week |
211 | 213 |
|
212 | 214 | core_rows.append( |
|
0 commit comments