# 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),
    ]