mirror of https://github.com/zulip/zulip.git
70 lines
3.1 KiB
Python
70 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Generated by Django 1.10.5 on 2017-03-04 07:33
|
|
from __future__ import unicode_literals
|
|
|
|
from django.db.backends.postgresql_psycopg2.schema import DatabaseSchemaEditor
|
|
from django.db.migrations.state import StateApps
|
|
|
|
from django.conf import settings
|
|
from django.db import migrations, models
|
|
import django.db.models.deletion
|
|
|
|
from django.utils import timezone
|
|
|
|
def backfill_user_activations_and_deactivations(apps, schema_editor):
|
|
# type: (StateApps, DatabaseSchemaEditor) -> None
|
|
migration_time = timezone.now()
|
|
RealmAuditLog = apps.get_model('zerver', 'RealmAuditLog')
|
|
UserProfile = apps.get_model('zerver', 'UserProfile')
|
|
|
|
for user in UserProfile.objects.all():
|
|
RealmAuditLog.objects.create(realm=user.realm, modified_user=user,
|
|
event_type='user_created', event_time=user.date_joined,
|
|
backfilled=False)
|
|
|
|
for user in UserProfile.objects.filter(is_active=False):
|
|
RealmAuditLog.objects.create(realm=user.realm, modified_user=user,
|
|
event_type='user_deactivated', event_time=migration_time,
|
|
backfilled=True)
|
|
|
|
def reverse_code(apps, schema_editor):
|
|
# type: (StateApps, DatabaseSchemaEditor) -> None
|
|
RealmAuditLog = apps.get_model('zerver', 'RealmAuditLog')
|
|
RealmAuditLog.objects.filter(event_type='user_created').delete()
|
|
RealmAuditLog.objects.filter(event_type='user_deactivated').delete()
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('zerver', '0056_userprofile_emoji_alt_code'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.CreateModel(
|
|
name='RealmAuditLog',
|
|
fields=[
|
|
('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('event_type', models.CharField(max_length=40)),
|
|
('backfilled', models.BooleanField(default=False)),
|
|
('event_time', models.DateTimeField()),
|
|
('acting_user', models.ForeignKey(null=True,
|
|
on_delete=django.db.models.deletion.CASCADE,
|
|
related_name='+',
|
|
to=settings.AUTH_USER_MODEL)),
|
|
('modified_stream', models.ForeignKey(null=True,
|
|
on_delete=django.db.models.deletion.CASCADE,
|
|
to='zerver.Stream')),
|
|
('modified_user', models.ForeignKey(null=True,
|
|
on_delete=django.db.models.deletion.CASCADE,
|
|
related_name='+',
|
|
to=settings.AUTH_USER_MODEL)),
|
|
('realm', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='zerver.Realm')),
|
|
],
|
|
),
|
|
|
|
migrations.RunPython(backfill_user_activations_and_deactivations,
|
|
reverse_code=reverse_code),
|
|
|
|
]
|