zulip/zerver/migrations/0256_userprofile_stream_set_recipient_column_values.py

from django.db import migrations


class Migration(migrations.Migration):

    dependencies = [
        ('zerver', '0255_userprofile_stream_add_recipient_column'),
    ]

    operations = [
        migrations.RunSQL(
            """
            UPDATE zerver_userprofile
            SET recipient_id = zerver_recipient.id
            FROM zerver_recipient
            WHERE zerver_recipient.type_id = zerver_userprofile.id AND zerver_recipient.type = 1;
            """,
            reverse_sql='UPDATE zerver_userprofile SET recipient_id = NULL'),
        migrations.RunSQL(
            """
            UPDATE zerver_stream
            SET recipient_id = zerver_recipient.id
            FROM zerver_recipient
            WHERE zerver_recipient.type_id = zerver_stream.id AND zerver_recipient.type = 2;
            """,
            reverse_sql='UPDATE zerver_stream SET recipient_id = NULL'),
    ]