zulip/zerver/migrations/0236_remove_illegal_charact...

31 lines
985 B
Python

# Generated by Django 1.11.20 on 2019-06-28 21:45
from unicodedata import category
from django.db import migrations
from django.db.backends.postgresql.schema import DatabaseSchemaEditor
from django.db.migrations.state import StateApps
NAME_INVALID_CHARS = ['*', '`', "\\", '>', '"', '@']
def remove_name_illegal_chars(apps: StateApps, schema_editor: DatabaseSchemaEditor) -> None:
UserProfile = apps.get_model("zerver", "UserProfile")
for user in UserProfile.objects.all():
stripped = []
for char in user.full_name:
if (char not in NAME_INVALID_CHARS) and (category(char)[0] != "C"):
stripped.append(char)
user.full_name = "".join(stripped)
user.save(update_fields=["full_name"])
class Migration(migrations.Migration):
dependencies = [
('zerver', '0235_userprofile_desktop_icon_count_display'),
]
operations = [
migrations.RunPython(remove_name_illegal_chars, elidable=True)
]