OpenSlides/openslides/mediafiles/migrations/0005_directories_and_permissions_2.py
2019-07-19 13:45:39 +02:00

73 lines
2.4 KiB
Python

# Generated by Django 2.2.2 on 2019-06-28 06:09
import os.path
from typing import List
from django.contrib.auth.models import Permission
from django.contrib.contenttypes.models import ContentType
from django.db import migrations
def change_logo_and_font_configs(apps, schema_editor):
"""
All old paths with "/media/file/..." will be replaced with "/media/...".
"""
ConfigStore = apps.get_model("core", "ConfigStore")
logo_font_keys: List[str] = []
try:
logo_font_keys.extend(ConfigStore.objects.get(key="logos_available").value)
logo_font_keys.extend(ConfigStore.objects.get(key="fonts_available").value)
except ConfigStore.DoesNotExist:
pass
for key in logo_font_keys:
config = ConfigStore.objects.get(key=key)
logo = config.value
path = logo.get("path")
if path and path.startswith("/media/file/"):
logo["path"] = path.replace("/media/file/", "/media/", 1)
config.value = logo
config.save()
def copy_filename(apps, schema_editor):
Mediafile = apps.get_model("mediafiles", "Mediafile")
for mediafile in Mediafile.objects.all():
filename = os.path.basename(mediafile.mediafile.name)
mediafile.original_filename = filename
mediafile.save(skip_autoupdate=True)
def set_groups_and_delete_old_permissions(apps, schema_editor):
Mediafile = apps.get_model("mediafiles", "Mediafile")
mediafile_content_type = ContentType.objects.get(model="mediafile")
try:
can_see_hidden = Permission.objects.get(
codename="can_see_hidden", content_type=mediafile_content_type
)
group_ids = [group.id for group in can_see_hidden.group_set.all()]
for mediafile in Mediafile.objects.filter(hidden=True):
mediafile.access_groups.set(group_ids)
mediafile.save(skip_autoupdate=True)
# Delete permissions
can_see_hidden.delete()
Permission.objects.filter(
codename="can_upload", content_type=mediafile_content_type
).delete()
except Permission.DoesNotExist:
pass
class Migration(migrations.Migration):
dependencies = [("mediafiles", "0004_directories_and_permissions_1")]
operations = [
migrations.RunPython(change_logo_and_font_configs),
migrations.RunPython(copy_filename),
migrations.RunPython(set_groups_and_delete_old_permissions),
]