39 lines
1 KiB
Python
39 lines
1 KiB
Python
|
# Generated by Django 3.2.18 on 2023-08-28 06:02
|
||
|
|
|
||
|
|
from django.db import migrations, models
|
||
|
|
|
||
|
|
|
||
|
|
def migrate(apps, schema_editor):
|
||
|
|
UserRoom = apps.get_model("chat_box", "UserRoom")
|
||
|
|
Message = apps.get_model("chat_box", "Message")
|
||
|
|
|
||
|
|
for ur in UserRoom.objects.all():
|
||
|
|
if not ur.room:
|
||
|
|
continue
|
||
|
|
messages = ur.room.message_set
|
||
|
|
last_msg = messages.first()
|
||
|
|
try:
|
||
|
|
if last_msg and last_msg.author != ur.user:
|
||
|
|
ur.unread_count = messages.filter(time__gte=ur.last_seen).count()
|
||
|
|
else:
|
||
|
|
ur.unread_count = 0
|
||
|
|
ur.save()
|
||
|
|
except:
|
||
|
|
continue
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
("chat_box", "0013_alter_message_time"),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.AddField(
|
||
|
|
model_name="userroom",
|
||
|
|
name="unread_count",
|
||
|
|
field=models.IntegerField(db_index=True, default=0),
|
||
|
|
),
|
||
|
|
migrations.RunPython(migrate, migrations.RunPython.noop, atomic=True),
|
||
|
|
]
|