Добавил абуз вч отметок
This commit is contained in:
@@ -2,17 +2,24 @@
|
||||
Management command для генерации тестовых отметок сигналов.
|
||||
|
||||
Использование:
|
||||
python manage.py generate_test_marks --satellite_id=1 --days=90 --marks_per_day=5
|
||||
python manage.py generate_test_marks --satellite_id=1 --user_id=1 --date_range=10.10.2025-15.10.2025
|
||||
|
||||
Параметры:
|
||||
--satellite_id: ID спутника (обязательный)
|
||||
--days: Количество дней для генерации (по умолчанию 90)
|
||||
--marks_per_day: Количество отметок в день (по умолчанию 3)
|
||||
--user_id: ID пользователя CustomUser (обязательный)
|
||||
--date_range: Диапазон дат в формате ДД.ММ.ГГГГ-ДД.ММ.ГГГГ (обязательный)
|
||||
--clear: Удалить существующие отметки перед генерацией
|
||||
|
||||
Особенности:
|
||||
- Генерирует отметки только в будние дни (пн-пт)
|
||||
- Время отметок: утро с 8:00 до 11:00
|
||||
- Одна отметка в день для всех сигналов спутника
|
||||
- Все отметки в один день имеют одинаковый timestamp (пакетное сохранение)
|
||||
- Все отметки имеют значение True (сигнал присутствует)
|
||||
"""
|
||||
|
||||
import random
|
||||
from datetime import timedelta
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from django.core.management.base import BaseCommand, CommandError
|
||||
from django.utils import timezone
|
||||
@@ -31,16 +38,16 @@ class Command(BaseCommand):
|
||||
help='ID спутника для генерации отметок'
|
||||
)
|
||||
parser.add_argument(
|
||||
'--days',
|
||||
'--user_id',
|
||||
type=int,
|
||||
default=90,
|
||||
help='Количество дней для генерации (по умолчанию 90)'
|
||||
required=True,
|
||||
help='ID пользователя CustomUser - автор всех отметок'
|
||||
)
|
||||
parser.add_argument(
|
||||
'--marks_per_day',
|
||||
type=int,
|
||||
default=3,
|
||||
help='Среднее количество отметок в день на теханализ (по умолчанию 3)'
|
||||
'--date_range',
|
||||
type=str,
|
||||
required=True,
|
||||
help='Диапазон дат в формате ДД.ММ.ГГГГ-ДД.ММ.ГГГГ (например: 10.10.2025-15.10.2025)'
|
||||
)
|
||||
parser.add_argument(
|
||||
'--clear',
|
||||
@@ -50,10 +57,34 @@ class Command(BaseCommand):
|
||||
|
||||
def handle(self, *args, **options):
|
||||
satellite_id = options['satellite_id']
|
||||
days = options['days']
|
||||
marks_per_day = options['marks_per_day']
|
||||
user_id = options['user_id']
|
||||
date_range = options['date_range']
|
||||
clear = options['clear']
|
||||
|
||||
# Проверяем существование пользователя
|
||||
try:
|
||||
custom_user = CustomUser.objects.select_related('user').get(id=user_id)
|
||||
except CustomUser.DoesNotExist:
|
||||
raise CommandError(f'Пользователь CustomUser с ID {user_id} не найден')
|
||||
|
||||
# Парсим диапазон дат
|
||||
try:
|
||||
start_str, end_str = date_range.split('-')
|
||||
start_date = datetime.strptime(start_str.strip(), '%d.%m.%Y')
|
||||
end_date = datetime.strptime(end_str.strip(), '%d.%m.%Y')
|
||||
|
||||
# Делаем timezone-aware
|
||||
start_date = timezone.make_aware(start_date)
|
||||
end_date = timezone.make_aware(end_date)
|
||||
|
||||
if start_date > end_date:
|
||||
raise CommandError('Начальная дата должна быть раньше конечной')
|
||||
|
||||
except ValueError as e:
|
||||
raise CommandError(
|
||||
f'Неверный формат даты. Используйте ДД.ММ.ГГГГ-ДД.ММ.ГГГГ (например: 10.10.2025-15.10.2025). Ошибка: {e}'
|
||||
)
|
||||
|
||||
# Проверяем существование спутника
|
||||
try:
|
||||
satellite = Satellite.objects.get(id=satellite_id)
|
||||
@@ -61,16 +92,17 @@ class Command(BaseCommand):
|
||||
raise CommandError(f'Спутник с ID {satellite_id} не найден')
|
||||
|
||||
# Получаем теханализы для спутника
|
||||
tech_analyzes = TechAnalyze.objects.filter(satellite=satellite)
|
||||
ta_count = tech_analyzes.count()
|
||||
tech_analyzes = list(TechAnalyze.objects.filter(satellite=satellite))
|
||||
ta_count = len(tech_analyzes)
|
||||
|
||||
if ta_count == 0:
|
||||
raise CommandError(f'Нет теханализов для спутника "{satellite.name}"')
|
||||
|
||||
self.stdout.write(f'Спутник: {satellite.name}')
|
||||
self.stdout.write(f'Теханализов: {ta_count}')
|
||||
self.stdout.write(f'Период: {days} дней')
|
||||
self.stdout.write(f'Отметок в день: ~{marks_per_day}')
|
||||
self.stdout.write(f'Пользователь: {custom_user}')
|
||||
self.stdout.write(f'Период: {start_str} - {end_str} (только будние дни)')
|
||||
self.stdout.write(f'Время: 8:00 - 11:00')
|
||||
|
||||
# Удаляем существующие отметки если указан флаг
|
||||
if clear:
|
||||
@@ -81,56 +113,46 @@ class Command(BaseCommand):
|
||||
self.style.WARNING(f'Удалено существующих отметок: {deleted_count}')
|
||||
)
|
||||
|
||||
# Получаем или создаём тестового пользователя
|
||||
test_users = self._get_or_create_test_users()
|
||||
|
||||
# Генерируем отметки
|
||||
now = timezone.now()
|
||||
start_date = now - timedelta(days=days)
|
||||
|
||||
total_marks = 0
|
||||
marks_to_create = []
|
||||
workdays_count = 0
|
||||
|
||||
for ta in tech_analyzes:
|
||||
# Для каждого теханализа генерируем отметки
|
||||
current_date = start_date
|
||||
|
||||
# Начальное состояние сигнала (случайное)
|
||||
signal_present = random.choice([True, False])
|
||||
|
||||
while current_date < now:
|
||||
# Случайное количество отметок в этот день (от 0 до marks_per_day * 2)
|
||||
day_marks = random.randint(0, marks_per_day * 2)
|
||||
current_date = start_date
|
||||
# Включаем конечную дату в диапазон
|
||||
end_date_inclusive = end_date + timedelta(days=1)
|
||||
|
||||
while current_date < end_date_inclusive:
|
||||
# Проверяем, что это будний день (0=пн, 4=пт)
|
||||
if current_date.weekday() < 5:
|
||||
workdays_count += 1
|
||||
|
||||
for _ in range(day_marks):
|
||||
# Случайное время в течение дня
|
||||
random_hours = random.randint(0, 23)
|
||||
random_minutes = random.randint(0, 59)
|
||||
mark_time = current_date.replace(
|
||||
hour=random_hours,
|
||||
minute=random_minutes,
|
||||
second=random.randint(0, 59)
|
||||
)
|
||||
|
||||
# Пропускаем если время в будущем
|
||||
if mark_time > now:
|
||||
continue
|
||||
|
||||
# С вероятностью 70% сигнал остаётся в том же состоянии
|
||||
# С вероятностью 30% меняется
|
||||
if random.random() > 0.7:
|
||||
signal_present = not signal_present
|
||||
|
||||
# Генерируем случайное время в диапазоне 8:00-11:00
|
||||
random_hour = random.randint(8, 10)
|
||||
random_minute = random.randint(0, 59)
|
||||
random_second = random.randint(0, 59)
|
||||
|
||||
mark_time = current_date.replace(
|
||||
hour=random_hour,
|
||||
minute=random_minute,
|
||||
second=random_second,
|
||||
microsecond=0
|
||||
)
|
||||
|
||||
# Создаём отметки для всех теханализов с одинаковым timestamp
|
||||
for ta in tech_analyzes:
|
||||
marks_to_create.append(ObjectMark(
|
||||
tech_analyze=ta,
|
||||
mark=signal_present,
|
||||
created_by=random.choice(test_users),
|
||||
mark=True, # Всегда True
|
||||
timestamp=mark_time,
|
||||
created_by=custom_user,
|
||||
))
|
||||
total_marks += 1
|
||||
|
||||
current_date += timedelta(days=1)
|
||||
|
||||
current_date += timedelta(days=1)
|
||||
|
||||
# Bulk create для производительности
|
||||
self.stdout.write(f'Рабочих дней: {workdays_count}')
|
||||
self.stdout.write(f'Создание {total_marks} отметок...')
|
||||
|
||||
# Создаём партиями по 1000
|
||||
@@ -140,65 +162,8 @@ class Command(BaseCommand):
|
||||
ObjectMark.objects.bulk_create(batch)
|
||||
self.stdout.write(f' Создано: {min(i + batch_size, len(marks_to_create))}/{total_marks}')
|
||||
|
||||
# Обновляем timestamp для созданных отметок (bulk_create не вызывает auto_now_add корректно)
|
||||
self.stdout.write('Обновление временных меток...')
|
||||
|
||||
# Получаем созданные отметки и обновляем их timestamp
|
||||
created_marks = ObjectMark.objects.filter(
|
||||
tech_analyze__satellite=satellite
|
||||
).order_by('id')
|
||||
|
||||
# Распределяем временные метки
|
||||
current_date = start_date
|
||||
mark_index = 0
|
||||
|
||||
for ta in tech_analyzes:
|
||||
ta_marks = list(created_marks.filter(tech_analyze=ta).order_by('id'))
|
||||
if not ta_marks:
|
||||
continue
|
||||
|
||||
# Распределяем отметки по времени
|
||||
time_step = timedelta(days=days) / len(ta_marks) if ta_marks else timedelta(hours=1)
|
||||
|
||||
for i, mark in enumerate(ta_marks):
|
||||
mark_time = start_date + (time_step * i)
|
||||
# Добавляем случайное смещение
|
||||
mark_time += timedelta(
|
||||
hours=random.randint(0, 23),
|
||||
minutes=random.randint(0, 59)
|
||||
)
|
||||
if mark_time > now:
|
||||
mark_time = now - timedelta(minutes=random.randint(1, 60))
|
||||
|
||||
ObjectMark.objects.filter(id=mark.id).update(timestamp=mark_time)
|
||||
|
||||
self.stdout.write(
|
||||
self.style.SUCCESS(
|
||||
f'Успешно создано {total_marks} отметок для {ta_count} теханализов'
|
||||
f'Успешно создано {total_marks} отметок для {ta_count} теханализов за {workdays_count} рабочих дней'
|
||||
)
|
||||
)
|
||||
|
||||
def _get_or_create_test_users(self):
|
||||
"""Получает или создаёт тестовых пользователей для отметок."""
|
||||
from django.contrib.auth.models import User
|
||||
|
||||
test_usernames = ['operator1', 'operator2', 'operator3', 'analyst1', 'analyst2']
|
||||
custom_users = []
|
||||
|
||||
for username in test_usernames:
|
||||
user, created = User.objects.get_or_create(
|
||||
username=username,
|
||||
defaults={
|
||||
'first_name': username.capitalize(),
|
||||
'last_name': 'Тестовый',
|
||||
'is_active': True,
|
||||
}
|
||||
)
|
||||
|
||||
custom_user, _ = CustomUser.objects.get_or_create(
|
||||
user=user,
|
||||
defaults={'role': 'user'}
|
||||
)
|
||||
custom_users.append(custom_user)
|
||||
|
||||
return custom_users
|
||||
|
||||
Reference in New Issue
Block a user