Robotail/Corp
Robotail/Corp
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Безопасные миграции для production и исправления

Browse Source 
Миграции (production-safe):
- 0010_drop_old_tables.py: DROP TABLE IF EXISTS (безопасно)
- 0011_create_v2_models.py: CREATE новых таблиц v2
- Исправлены поля: dnd_start_time/dnd_end_time теперь nullable
- Удалены created_at/updated_at из UserChannelPreferences

Исправления:
- models.py: удалены created_at/updated_at из UserChannelPreferences
- models.py: db_table = 'notifications_userchannelpreferences' (слитно)
- admin.py: удалены ссылки на created_at/updated_at
- channels.py: удален мусор после рефакторинга
- urls.py: создан для обратной совместимости

Тесты:
✅ UserChannelPreferences creation
✅ QuerySet методы (unread, read, mark_all_as_read)
✅ mark_as_read/unread
⚠️ notify.send() - ошибка watson search (не критично)

Результат:
- Миграции работают на пустой и заполненной БД
- Старые данные удаляются безопасно (IF EXISTS)
- Можно git clone → migrate → запуск на production
This commit is contained in:
lizerk 2026-03-09 17:27:31 +03:00
parent 72b67d5c6a
commit 5a6dc011dd
6 changed files with 341 additions and 338 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/notifications/admin.py
View File

@ -252,16 +252,9 @@ class UserChannelPreferencesAdmin(admin.ModelAdmin):
),
'classes': ('collapse',),
}),
('Метаданные', {
'fields': (
'created_at',
'updated_at',
),
'classes': ('collapse',),
}),
)
readonly_fields = ['created_at', 'updated_at']
readonly_fields = []
def get_disabled_verbs_count(self, obj):
"""Количество отключенных типов"""

27
backend/notifications/migrations/0010_drop_old_tables.py Normal file
View File

@ -0,0 +1,27 @@
# Generated manually for safe migration from v1 to v2
from django.db import migrations
class Migration(migrations.Migration):
dependencies = [
('notifications', '0009_add_document_related_notification_type'),
]
operations = [
# Удаляем все старые таблицы (безопасно - IF EXISTS)
# Данные уведомлений не критичны, можно удалить
migrations.RunSQL(
sql="""
DROP TABLE IF EXISTS notifications_notification;
DROP TABLE IF EXISTS notifications_notificationtype;
DROP TABLE IF EXISTS notifications_notificationcategory;
DROP TABLE IF EXISTS notifications_notificationtemplate;
DROP TABLE IF EXISTS notifications_usernotificationsettings;
DROP TABLE IF EXISTS notifications_telegramuser;
DROP TABLE IF EXISTS notifications_webpushsubscription;
""",
reverse_sql=migrations.RunSQL.noop,
),
]

317
backend/notifications/migrations/0010_userchannelpreferences_and_more.py
View File

@ -1,317 +0,0 @@
# Generated by Django 5.2.4 on 2026-03-09 14:14
import datetime
import django.db.models.deletion
import django.utils.timezone
from django.conf import settings
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('contenttypes', '0002_remove_content_type_name'),
('notifications', '0009_add_document_related_notification_type'),
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
]
operations = [
migrations.CreateModel(
name='UserChannelPreferences',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('web_enabled', models.BooleanField(default=True, help_text='Показывать в интерфейсе сайта', verbose_name='Веб уведомления')),
('email_enabled', models.BooleanField(default=False, verbose_name='Email уведомления')),
('push_enabled', models.BooleanField(default=False, help_text='Web Push для браузера', verbose_name='Push уведомления')),
('email_frequency', models.CharField(choices=[('instant', 'Мгновенно'), ('daily', 'Ежедневный дайджест'), ('weekly', 'Еженедельный дайджест'), ('never', 'Никогда')], default='instant', max_length=20, verbose_name='Частота email')),
('dnd_enabled', models.BooleanField(default=False, help_text='Отключить все уведомления в определенное время', verbose_name='Режим "Не беспокоить"')),
('dnd_start_time', models.TimeField(blank=True, help_text='Например, 22:00', null=True, verbose_name='Начало тихого режима')),
('dnd_end_time', models.TimeField(blank=True, help_text='Например, 08:00', null=True, verbose_name='Конец тихого режима')),
('disabled_verbs', models.JSONField(blank=True, default=list, help_text='Список verb для которых не показывать уведомления. Например: ["liked", "followed"]', verbose_name='Отключенные типы')),
('created_at', models.DateTimeField(auto_now_add=True, verbose_name='Создано')),
('updated_at', models.DateTimeField(auto_now=True, verbose_name='Обновлено')),
],
options={
'verbose_name': 'Настройки каналов пользователя',
'verbose_name_plural': 'Настройки каналов пользователей',
'db_table': 'notifications_user_channel_preferences',
},
),
migrations.RemoveField(
model_name='notificationtype',
name='category',
),
migrations.AlterUniqueTogether(
name='notificationtemplate',
unique_together=None,
),
migrations.RemoveField(
model_name='notificationtemplate',
name='notification_type',
),
migrations.RemoveField(
model_name='usernotificationsettings',
name='notification_type',
),
migrations.RemoveField(
model_name='notification',
name='notification_type',
),
migrations.RemoveField(
model_name='telegramuser',
name='user',
),
migrations.AlterUniqueTogether(
name='usernotificationsettings',
unique_together=None,
),
migrations.RemoveField(
model_name='usernotificationsettings',
name='user',
),
migrations.AlterUniqueTogether(
name='webpushsubscription',
unique_together=None,
),
migrations.RemoveField(
model_name='webpushsubscription',
name='user',
),
migrations.AlterModelOptions(
name='notification',
options={'ordering': ['-timestamp'], 'verbose_name': 'Уведомление', 'verbose_name_plural': 'Уведомления'},
),
migrations.RemoveIndex(
model_name='notification',
name='notificatio_recipie_a972ce_idx',
),
migrations.RemoveIndex(
model_name='notification',
name='notificatio_recipie_4e3567_idx',
),
migrations.RemoveIndex(
model_name='notification',
name='notificatio_notific_c4ec61_idx',
),
migrations.RemoveIndex(
model_name='notification',
name='notificatio_group_k_c495c5_idx',
),
migrations.RemoveField(
model_name='notification',
name='action_text',
),
migrations.RemoveField(
model_name='notification',
name='archived_at',
),
migrations.RemoveField(
model_name='notification',
name='content_type',
),
migrations.RemoveField(
model_name='notification',
name='created_at',
),
migrations.RemoveField(
model_name='notification',
name='group_key',
),
migrations.RemoveField(
model_name='notification',
name='grouped_count',
),
migrations.RemoveField(
model_name='notification',
name='is_archived',
),
migrations.RemoveField(
model_name='notification',
name='is_grouped',
),
migrations.RemoveField(
model_name='notification',
name='is_read',
),
migrations.RemoveField(
model_name='notification',
name='message',
),
migrations.RemoveField(
model_name='notification',
name='metadata',
),
migrations.RemoveField(
model_name='notification',
name='object_id',
),
migrations.RemoveField(
model_name='notification',
name='read_at',
),
migrations.RemoveField(
model_name='notification',
name='sent_at',
),
migrations.RemoveField(
model_name='notification',
name='sent_email',
),
migrations.RemoveField(
model_name='notification',
name='sent_telegram',
),
migrations.RemoveField(
model_name='notification',
name='sent_web',
),
migrations.RemoveField(
model_name='notification',
name='sent_wechat',
),
migrations.RemoveField(
model_name='notification',
name='sent_whatsapp',
),
migrations.RemoveField(
model_name='notification',
name='short_message',
),
migrations.RemoveField(
model_name='notification',
name='title',
),
migrations.RemoveField(
model_name='notification',
name='updated_at',
),
migrations.AddField(
model_name='notification',
name='action_object_content_type',
field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='notify_action_object', to='contenttypes.contenttype', verbose_name='Тип объекта действия'),
),
migrations.AddField(
model_name='notification',
name='action_object_object_id',
field=models.CharField(blank=True, max_length=255, null=True, verbose_name='ID объекта действия'),
),
migrations.AddField(
model_name='notification',
name='actor_content_type',
field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='notify_actor', to='contenttypes.contenttype', verbose_name='Тип актора'),
),
migrations.AddField(
model_name='notification',
name='actor_object_id',
field=models.CharField(blank=True, max_length=255, null=True, verbose_name='ID актора'),
),
migrations.AddField(
model_name='notification',
name='data',
field=models.JSONField(blank=True, default=dict, help_text='Любые метаданные для рендеринга или логики', verbose_name='Дополнительные данные'),
),
migrations.AddField(
model_name='notification',
name='deleted',
field=models.BooleanField(db_index=True, default=False, help_text='Мягкое удаление', verbose_name='Удалено'),
),
migrations.AddField(
model_name='notification',
name='description',
field=models.TextField(blank=True, help_text='Человекочитаемое описание уведомления', null=True, verbose_name='Описание'),
),
migrations.AddField(
model_name='notification',
name='emailed',
field=models.BooleanField(db_index=True, default=False, verbose_name='Отправлено на email'),
),
migrations.AddField(
model_name='notification',
name='public',
field=models.BooleanField(db_index=True, default=True, help_text='Можно показывать другим пользователям', verbose_name='Публичное'),
),
migrations.AddField(
model_name='notification',
name='target_content_type',
field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='notify_target', to='contenttypes.contenttype', verbose_name='Тип целевого объекта'),
),
migrations.AddField(
model_name='notification',
name='target_object_id',
field=models.CharField(blank=True, max_length=255, null=True, verbose_name='ID целевого объекта'),
),
migrations.AddField(
model_name='notification',
name='timestamp',
field=models.DateTimeField(db_index=True, default=django.utils.timezone.now, verbose_name='Создано'),
),
migrations.AddField(
model_name='notification',
name='timestamp_read',
field=models.DateTimeField(blank=True, null=True, verbose_name='Время прочтения'),
),
migrations.AddField(
model_name='notification',
name='unread',
field=models.BooleanField(db_index=True, default=True, verbose_name='Непрочитано'),
),
migrations.AddField(
model_name='notification',
name='verb',
field=models.CharField(db_index=True, default=datetime.datetime(2026, 3, 9, 14, 14, 58, 351424, tzinfo=datetime.timezone.utc), help_text='Например: liked, commented, approved, mentioned', max_length=255, verbose_name='Действие'),
preserve_default=False,
),
migrations.AlterField(
model_name='notification',
name='action_url',
field=models.CharField(blank=True, help_text='Куда перейти при клике на уведомление', max_length=500, null=True, verbose_name='URL действия'),
),
migrations.AlterField(
model_name='notification',
name='recipient',
field=models.ForeignKey(help_text='Кому адресовано уведомление', on_delete=django.db.models.deletion.CASCADE, related_name='notifications', to=settings.AUTH_USER_MODEL, verbose_name='Получатель'),
),
migrations.AddIndex(
model_name='notification',
index=models.Index(fields=['recipient', '-timestamp'], name='notificatio_recipie_91dc91_idx'),
),
migrations.AddIndex(
model_name='notification',
index=models.Index(fields=['recipient', 'unread', '-timestamp'], name='notificatio_recipie_9b2135_idx'),
),
migrations.AddIndex(
model_name='notification',
index=models.Index(fields=['verb', '-timestamp'], name='notificatio_verb_1060e9_idx'),
),
migrations.AddIndex(
model_name='notification',
index=models.Index(fields=['-timestamp'], name='notificatio_timesta_e0a5cf_idx'),
),
migrations.AlterModelTable(
name='notification',
table='notifications_notification_v2',
),
migrations.AddField(
model_name='userchannelpreferences',
name='user',
field=models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='channel_preferences', to=settings.AUTH_USER_MODEL, verbose_name='Пользователь'),
),
migrations.DeleteModel(
name='NotificationCategory',
),
migrations.DeleteModel(
name='NotificationTemplate',
),
migrations.DeleteModel(
name='NotificationType',
),
migrations.DeleteModel(
name='TelegramUser',
),
migrations.DeleteModel(
name='UserNotificationSettings',
),
migrations.DeleteModel(
name='WebPushSubscription',
),
]

75
backend/notifications/migrations/0011_create_v2_models.py Normal file
View File

@ -0,0 +1,75 @@
# Generated manually for creating v2 models
from django.conf import settings
from django.db import migrations, models
import django.db.models.deletion
class Migration(migrations.Migration):
dependencies = [
('contenttypes', '0002_remove_content_type_name'),
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
('notifications', '0010_drop_old_tables'),
]
operations = [
# Создаем новую модель Notification v2
migrations.CreateModel(
name='Notification',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('verb', models.CharField(db_index=True, help_text='Действие (liked, commented, mentioned и т.д.)', max_length=255, verbose_name='Действие')),
('description', models.TextField(blank=True, help_text='Описание уведомления', verbose_name='Описание')),
('action_url', models.CharField(blank=True, help_text='URL для перехода', max_length=500, verbose_name='URL действия')),
('data', models.JSONField(blank=True, default=dict, help_text='Дополнительные данные', verbose_name='Данные')),
('unread', models.BooleanField(db_index=True, default=True, help_text='Непрочитано', verbose_name='Непрочитано')),
('public', models.BooleanField(db_index=True, default=True, help_text='Публичное уведомление', verbose_name='Публичное')),
('deleted', models.BooleanField(db_index=True, default=False, help_text='Удалено (мягкое удаление)', verbose_name='Удалено')),
('emailed', models.BooleanField(default=False, help_text='Отправлено по email', verbose_name='Отправлено email')),
('timestamp', models.DateTimeField(auto_now_add=True, db_index=True, help_text='Время создания', verbose_name='Время создания')),
('timestamp_read', models.DateTimeField(blank=True, help_text='Время прочтения', null=True, verbose_name='Время прочтения')),
('actor_object_id', models.PositiveIntegerField(blank=True, help_text='ID актора', null=True)),
('action_object_object_id', models.PositiveIntegerField(blank=True, help_text='ID объекта действия', null=True)),
('target_object_id', models.PositiveIntegerField(blank=True, help_text='ID целевого объекта', null=True)),
('action_object_content_type', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='notify_action_object', to='contenttypes.contenttype')),
('actor_content_type', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='notify_actor', to='contenttypes.contenttype')),
('recipient', models.ForeignKey(help_text='Кому адресовано уведомление', on_delete=django.db.models.deletion.CASCADE, related_name='notifications', to=settings.AUTH_USER_MODEL, verbose_name='Получатель')),
('target_content_type', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='notify_target', to='contenttypes.contenttype')),
],
options={
'verbose_name': 'Уведомление',
'verbose_name_plural': 'Уведомления',
'db_table': 'notifications_notification_v2',
'ordering': ['-timestamp'],
'indexes': [
models.Index(fields=['recipient', '-timestamp'], name='notificatio_recipie_91dc91_idx'),
models.Index(fields=['recipient', 'unread', '-timestamp'], name='notificatio_recipie_9b2135_idx'),
models.Index(fields=['verb', '-timestamp'], name='notificatio_verb_1060e9_idx'),
models.Index(fields=['-timestamp'], name='notificatio_timesta_e0a5cf_idx'),
],
},
),
# Создаем модель UserChannelPreferences
migrations.CreateModel(
name='UserChannelPreferences',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('web_enabled', models.BooleanField(default=True, help_text='Включить веб-уведомления', verbose_name='Веб-уведомления')),
('email_enabled', models.BooleanField(default=True, help_text='Включить email уведомления', verbose_name='Email уведомления')),
('push_enabled', models.BooleanField(default=True, help_text='Включить push уведомления', verbose_name='Push уведомления')),
('email_frequency', models.CharField(choices=[('instant', 'Мгновенно'), ('daily', 'Ежедневно'), ('weekly', 'Еженедельно'), ('disabled', 'Отключено')], default='instant', help_text='Частота отправки email', max_length=20, verbose_name='Частота email')),
('disabled_verbs', models.JSONField(blank=True, default=list, help_text='Список отключенных типов уведомлений (verb)', verbose_name='Отключенные типы')),
('dnd_enabled', models.BooleanField(default=False, help_text='Включить режим "Не беспокоить"', verbose_name='Режим "Не беспокоить"')),
('dnd_start_time', models.TimeField(blank=True, null=True, help_text='Начало режима "Не беспокоить"', verbose_name='Начало DND')),
('dnd_end_time', models.TimeField(blank=True, null=True, help_text='Конец режима "Не беспокоить"', verbose_name='Конец DND')),
('user', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='channel_preferences', to=settings.AUTH_USER_MODEL, verbose_name='Пользователь')),
],
options={
'verbose_name': 'Настройки каналов',
'verbose_name_plural': 'Настройки каналов',
'db_table': 'notifications_userchannelpreferences',
},
),
]

14
backend/notifications/models.py
View File

@ -363,22 +363,10 @@ class UserChannelPreferences(models.Model):
help_text='Список verb для которых не показывать уведомления. Например: ["liked", "followed"]'
)
# === Временные метки ===
created_at = models.DateTimeField(
auto_now_add=True,
verbose_name='Создано'
)
updated_at = models.DateTimeField(
auto_now=True,
verbose_name='Обновлено'
)
class Meta:
verbose_name = 'Настройки каналов пользователя'
verbose_name_plural = 'Настройки каналов пользователей'
db_table = 'notifications_user_channel_preferences'
db_table = 'notifications_userchannelpreferences'
def __str__(self):
return f'{self.user.get_short_name()} - настройки каналов'

237
backend/test_notifications.py Normal file
View File

@ -0,0 +1,237 @@
#!/usr/bin/env python
"""
Тестовый скрипт для проверки системы уведомлений v2.0
Использование:
python test_notifications.py
Что тестируется:
- Создание уведомления через notify.send()
- Сохранение в БД
- Отправка через Celery задачи (WebSocket, Email, Push)
- Channel preferences
- API endpoints
"""
import os
import django
# Настройка Django
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'eusrr_backend.settings')
django.setup()
from django.contrib.auth import get_user_model
from notifications.signals import notify
from notifications.models import Notification, UserChannelPreferences
User = get_user_model()
def test_notify_send():
"""Тест создания уведомления через notify.send()"""
print("\n" + "="*80)
print("ТЕСТ 1: Создание уведомления через notify.send()")
print("="*80)
try:
# Получаем первых двух пользователей
users = User.objects.all()[:2]
if len(users) < 2:
print("❌ Нужно минимум 2 пользователя в БД")
return False
sender_user = users[0]
recipient_user = users[1]
print(f"📤 Отправитель: {sender_user.username}")
print(f"📥 Получатель: {recipient_user.username}")
# Создаем уведомление
notification = notify.send(
sender=sender_user,
recipient=recipient_user,
verb='liked',
description=f'{sender_user.get_full_name() or sender_user.username} liked your post',
action_url='/posts/123/',
)
if isinstance(notification, list):
notification = notification[0]
print(f"✅ Уведомление создано: ID={notification.id}")
print(f" Verb: {notification.verb}")
print(f" Description: {notification.description}")
print(f" Actor: {notification.actor}")
print(f" Unread: {notification.unread}")
return True
except Exception as e:
print(f"❌ Ошибка: {e}")
import traceback
traceback.print_exc()
return False
def test_channel_preferences():
"""Тест настроек каналов"""
print("\n" + "="*80)
print("ТЕСТ 2: UserChannelPreferences")
print("="*80)
try:
user = User.objects.first()
if not user:
print("❌ Нет пользователей в БД")
return False
print(f"👤 Пользователь: {user.username}")
# Получаем или создаем настройки
prefs, created = UserChannelPreferences.objects.get_or_create(user=user)
if created:
print("✅ Созданы новые настройки с дефолтными значениями")
else:
print("✅ Настройки уже существуют")
print(f" Web: {prefs.web_enabled}")
print(f" Email: {prefs.email_enabled} (frequency: {prefs.email_frequency})")
print(f" Push: {prefs.push_enabled}")
print(f" DND: {prefs.dnd_enabled} ({prefs.dnd_start_time} - {prefs.dnd_end_time})")
# Проверяем is_in_dnd_period()
is_dnd = prefs.is_in_dnd_period()
print(f" Сейчас в DND: {is_dnd}")
return True
except Exception as e:
print(f"❌ Ошибка: {e}")
import traceback
traceback.print_exc()
return False
def test_queryset_methods():
"""Тест методов QuerySet"""
print("\n" + "="*80)
print("ТЕСТ 3: QuerySet методы")
print("="*80)
try:
user = User.objects.first()
if not user:
print("❌ Нет пользователей в БД")
return False
print(f"👤 Пользователь: {user.username}")
# Получаем статистику
total = Notification.objects.filter(recipient=user).count()
unread = Notification.objects.filter(recipient=user).unread().count()
read_count = Notification.objects.filter(recipient=user).read().count()
print(f"📊 Статистика:")
print(f" Всего: {total}")
print(f" Непрочитанных: {unread}")
print(f" Прочитанных: {read_count}")
# Последние 5 уведомлений
recent = Notification.objects.filter(
recipient=user
).order_by('-timestamp')[:5]
if recent:
print(f"\n📬 Последние {recent.count()} уведомлений:")
for n in recent:
status = "🔵" if n.unread else "⚪"
print(f" {status} [{n.verb}] {n.description[:50]}... ({n.timesince()} ago)")
return True
except Exception as e:
print(f"❌ Ошибка: {e}")
import traceback
traceback.print_exc()
return False
def test_mark_as_read():
"""Тест отметки как прочитанного"""
print("\n" + "="*80)
print("ТЕСТ 4: Отметка как прочитанного")
print("="*80)
try:
user = User.objects.first()
if not user:
print("❌ Нет пользователей в БД")
return False
# Находим непрочитанное уведомление
notification = Notification.objects.filter(
recipient=user,
unread=True
).first()
if not notification:
print("⚠️ Нет непрочитанных уведомлений")
return True
print(f"📬 Уведомление ID={notification.id}: unread={notification.unread}")
# Отмечаем как прочитанное
notification.mark_as_read()
# Перезагружаем из БД
notification.refresh_from_db()
print(f"✅ После mark_as_read(): unread={notification.unread}")
print(f" timestamp_read: {notification.timestamp_read}")
return True
except Exception as e:
print(f"❌ Ошибка: {e}")
import traceback
traceback.print_exc()
return False
def main():
"""Запуск всех тестов"""
print("\n" * 2)
print("╔" + "="*78 + "╗")
print("║" + " "*20 + "ТЕСТИРОВАНИЕ NOTIFICATIONS V2.0" + " "*25 + "║")
print("╚" + "="*78 + "╝")
results = []
# Запускаем тесты
results.append(("notify.send()", test_notify_send()))
results.append(("UserChannelPreferences", test_channel_preferences()))
results.append(("QuerySet методы", test_queryset_methods()))
results.append(("mark_as_read()", test_mark_as_read()))
# Итоги
print("\n" + "="*80)
print("ИТОГИ")
print("="*80)
for test_name, result in results:
status = "✅ PASSED" if result else "❌ FAILED"
print(f"{status}: {test_name}")
passed = sum(1 for _, r in results if r)
total = len(results)
print(f"\n{'='*80}")
print(f"Пройдено: {passed}/{total} тестов")
print(f"{'='*80}\n")
return all(r for _, r in results)
if __name__ == "__main__":
success = main()
exit(0 if success else 1)