mirror of
https://github.com/zulip/zulip.git
synced 2025-11-17 20:41:46 +00:00
f2c9ee8000
This function now matches the copy in zerver/lib/actions.py. This is the same migration as b250e42f61c525029bd2b3bbb8f4ea93ece62072; orjson enforces that we don't use integers as keys in JSON dictionaries.
69 lines
3.0 KiB
Python
69 lines
3.0 KiB
Python
# Generated by Django 2.2.12 on 2020-05-16 18:34
|
|
from typing import Any, Dict
|
|
|
|
import orjson
|
|
from django.db import migrations
|
|
from django.db.backends.postgresql.schema import DatabaseSchemaEditor
|
|
from django.db.migrations.state import StateApps
|
|
from django.db.models import Count
|
|
from django.utils.timezone import now as timezone_now
|
|
|
|
|
|
def set_realm_admins_as_realm_owners(apps: StateApps, schema_editor: DatabaseSchemaEditor) -> None:
|
|
UserProfile = apps.get_model('zerver', 'UserProfile')
|
|
RealmAuditLog = apps.get_model('zerver', 'RealmAuditLog')
|
|
|
|
UserProfile.ROLE_REALM_OWNER = 100
|
|
UserProfile.ROLE_REALM_ADMINISTRATOR = 200
|
|
UserProfile.ROLE_MEMBER = 400
|
|
UserProfile.ROLE_GUEST = 600
|
|
|
|
RealmAuditLog.USER_ROLE_CHANGED = 105
|
|
RealmAuditLog.OLD_VALUE = '1'
|
|
RealmAuditLog.NEW_VALUE = '2'
|
|
RealmAuditLog.ROLE_COUNT = '10'
|
|
RealmAuditLog.ROLE_COUNT_HUMANS = '11'
|
|
RealmAuditLog.ROLE_COUNT_BOTS = '12'
|
|
|
|
def realm_user_count_by_role(realm: Any) -> Dict[str, Any]:
|
|
human_counts = {str(UserProfile.ROLE_REALM_ADMINISTRATOR): 0,
|
|
str(UserProfile.ROLE_REALM_OWNER): 0,
|
|
str(UserProfile.ROLE_MEMBER): 0,
|
|
str(UserProfile.ROLE_GUEST): 0}
|
|
for value_dict in list(UserProfile.objects.filter(
|
|
realm=realm, is_bot=False, is_active=True).values('role').annotate(Count('role'))):
|
|
human_counts[str(value_dict['role'])] = value_dict['role__count']
|
|
bot_count = UserProfile.objects.filter(realm=realm, is_bot=True, is_active=True).count()
|
|
return {
|
|
RealmAuditLog.ROLE_COUNT_HUMANS: human_counts,
|
|
RealmAuditLog.ROLE_COUNT_BOTS: bot_count,
|
|
}
|
|
|
|
objects_to_create = []
|
|
|
|
for user in UserProfile.objects.filter(is_active=True, role=UserProfile.ROLE_REALM_ADMINISTRATOR):
|
|
user.role = UserProfile.ROLE_REALM_OWNER
|
|
user.save(update_fields=['role'])
|
|
audit_log_entry = RealmAuditLog(realm=user.realm, modified_user=user,
|
|
event_type=RealmAuditLog.USER_ROLE_CHANGED,
|
|
event_time=timezone_now(),
|
|
extra_data=orjson.dumps({
|
|
RealmAuditLog.OLD_VALUE: UserProfile.ROLE_REALM_ADMINISTRATOR,
|
|
RealmAuditLog.NEW_VALUE: UserProfile.ROLE_REALM_OWNER,
|
|
RealmAuditLog.ROLE_COUNT: realm_user_count_by_role(user.realm),
|
|
}).decode())
|
|
objects_to_create.append(audit_log_entry)
|
|
RealmAuditLog.objects.bulk_create(objects_to_create)
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('zerver', '0283_apple_auth'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(set_realm_admins_as_realm_owners,
|
|
reverse_code=migrations.RunPython.noop,
|
|
elidable=True),
|
|
]
|