Перенос данных с юга на новую модель не работает

У меня есть абстрактная модель, которую я конвертирую в конкретную модель. Я успешно использую юг для изменения схемы, но не могу использовать миграцию данных.

Мое начальное состояние:

class UserProfile(models.Model):
    user = models.OneToOneField(User, primary_key=True, \
                                related_name='profile')
    class Meta:
        abstract=True
class SpecificProfile(UserProfile):
    url = models.URLField()

Мое новое состояние:

class UserProfile(models.Model):
    user = models.OneToOneField(User, primary_key=True, \
                                related_name='profile')
class SpecificProfile(UserProfile):
    user_profile = models.OneToOneField(UserProfile, parent_link=True)
    url = models.URLField()

Моя миграция схемы:

class Migration(SchemaMigration):

    def forwards(self, orm):
        # Renaming field 'SpecProfile.user_profile'
        db.rename_column('specificprofile', 'user_id', 'user_profile_id')

        # Adding model 'UserProfile'
        db.create_table('userprofile', (
            ('user', self.gf('django.db.models.fields.related.OneToOneField')(related_name='profile', unique=True, primary_key=True, to=orm['auth.User'])),
        ))
        db.send_create_signal('myapp', ['UserProfile'])

Я отредактировал файл, созданный югом, чтобы переименовать одно поле в SpecificProfile.

Теперь, в процессе переноса данных, я хотел бы создать одну запись UserProfile для каждого SpecificProfile и назначить UserProfile.user_id для SpecificProfile.user_profile_id.

Итак, мой перенос данных вперед:

class Migration(DataMigration):

    def forwards(self, orm):
        for spec in orm.SpecificProfile.objects.all():
            user_profile = orm.UserProfile()
            user_profile.user_id = spec.user_profile_id
            user_profile.save()

Скрипт работает без ошибок, но не создает новую запись в таблице UserProfile. Должен ли я использовать UserProfile() вместо orm.UserProfile()?

Любые идеи?


django data-migration django-south
person duduklein    schedule 01.02.2012    source источник

Ответы (1)


arrow_upward
0
arrow_downward

SpecificProfile.user_profile_id ранее не существовало, поэтому у него нет данных для переноса. То, что вы действительно хотите сделать, это установить user_profile.user в spec.user, а затем установить spec.user_profile в user_profile.

 def forwards(self, orm):
    for spec in orm.SpecificProfile.objects.all():
        user_profile = orm.UserProfile()
        user_profile.user_id = spec.user_id
        user_profile.save()

        # Then,
        spec.user_profile_id = user_profile.id

Однако, как только вы выполнили начальную миграцию, я почти уверен, что SpecificProfile.user больше не существует. Юг удаляет это поле, так как теперь оно находится на UserProfile.

person Chris Pratt    schedule 01.02.2012
comment
SpecificProfile.user_profile_id должен существовать после миграции схемы, поскольку я переименовываю SpecificProfile.user_id в SpecificProfile.user_profile_id. Проверьте мой код миграции схемы выше. Это переименование работает нормально. Я проверил ваш код на всякий случай, но он тоже не сработал, вероятно, по указанной вами причине. - person duduklein; 01.02.2012
comment
Моя проблема в том, что orm.SpecificProfile.objects.all() возвращает пустой список - person duduklein; 02.02.2012