django-mailbox/django_mailbox/south_migrations/0014_migrate_message_attachments.py

import datetime
from south.db import db
from south.v2 import SchemaMigration
from django.db import models


class Migration(SchemaMigration):
    no_dry_run = True

    def forwards(self, orm):
        for message in orm["django_mailbox.Message"].objects.all():
            for attachment in message.attachments.all():
                attachment.message = message
                attachment.save()

    def backwards(self, orm):
        for attachment in orm["django_mailbox.MessageAttachment"].objects.all():
            if attachment.message:
                attachment.message.attachments.add(attachment)

    models = {
        "django_mailbox.mailbox": {
            "Meta": {"object_name": "Mailbox"},
            "active": ("django.db.models.fields.BooleanField", [], {"default": "True"}),
            "from_email": (
                "django.db.models.fields.CharField",
                [],
                {
                    "default": "None",
                    "max_length": "255",
                    "null": "True",
                    "blank": "True",
                },
            ),
            "id": ("django.db.models.fields.AutoField", [], {"primary_key": "True"}),
            "name": ("django.db.models.fields.CharField", [], {"max_length": "255"}),
            "uri": (
                "django.db.models.fields.CharField",
                [],
                {
                    "default": "None",
                    "max_length": "255",
                    "null": "True",
                    "blank": "True",
                },
            ),
        },
        "django_mailbox.message": {
            "Meta": {"object_name": "Message"},
            "attachments": (
                "django.db.models.fields.related.ManyToManyField",
                [],
                {
                    "symmetrical": "False",
                    "related_name": "'messages_old'",
                    "blank": "True",
                    "to": "orm['django_mailbox.MessageAttachment']",
                },
            ),
            "body": ("django.db.models.fields.TextField", [], {}),
            "from_header": (
                "django.db.models.fields.CharField",
                [],
                {"max_length": "255"},
            ),
            "id": ("django.db.models.fields.AutoField", [], {"primary_key": "True"}),
            "in_reply_to": (
                "django.db.models.fields.related.ForeignKey",
                [],
                {
                    "blank": "True",
                    "related_name": "'replies'",
                    "null": "True",
                    "to": "orm['django_mailbox.Message']",
                },
            ),
            "mailbox": (
                "django.db.models.fields.related.ForeignKey",
                [],
                {"related_name": "'messages'", "to": "orm['django_mailbox.Mailbox']"},
            ),
            "message_id": (
                "django.db.models.fields.CharField",
                [],
                {"max_length": "255"},
            ),
            "outgoing": (
                "django.db.models.fields.BooleanField",
                [],
                {"default": "False"},
            ),
            "processed": (
                "django.db.models.fields.DateTimeField",
                [],
                {"auto_now_add": "True", "blank": "True"},
            ),
            "read": (
                "django.db.models.fields.DateTimeField",
                [],
                {"default": "None", "null": "True", "blank": "True"},
            ),
            "subject": ("django.db.models.fields.CharField", [], {"max_length": "255"}),
            "to_header": ("django.db.models.fields.TextField", [], {}),
        },
        "django_mailbox.messageattachment": {
            "Meta": {"object_name": "MessageAttachment"},
            "document": (
                "django.db.models.fields.files.FileField",
                [],
                {"max_length": "100"},
            ),
            "id": ("django.db.models.fields.AutoField", [], {"primary_key": "True"}),
            "message": (
                "django.db.models.fields.related.ForeignKey",
                [],
                {
                    "blank": "True",
                    "related_name": "'attachments_new'",
                    "null": "True",
                    "to": "orm['django_mailbox.Message']",
                },
            ),
        },
    }

    complete_apps = ["django_mailbox"]