Repositório utilizado para estudar migração de dados utilizando o Django
Aqui fiz um app, onde inicialmente eu tinha uma model Book, e nessa mesma model ficavam as seguintes informações sobre livros:
- Id
- Título
- Páginas
- Autor
Nessa situação fictícia, eu preciso separar os autores em uma model separada, e utilizarei as migrations do Django para isso.
Ao final do processo, a model Book deve conter os campos:
- Id
- Título
- Páginas
- Autor (ForeignKey)
E a model Author deve conter:
- Id
- Nome
Nessa simulação, a model Book foi inicialmente declarada assim:
class Book(models.Model): id = models.UUIDField(primary_key=True, default=uuid.uuid4) title = models.CharField(max_length=255) pages = models.IntegerField() author = models.CharField(max_length=255)O primeiro passo, foi criar a model Author:
class Author(models.Model): id = models.UUIDField(primary_key=True, default=uuid.uuid4) name = models.CharField(max_length=255)- Aqui eu gerei uma migration
Segundo passo, adicionar na model Book, o campo do tipo ForeignKey que fará o vínculo do livro com o autor. Importante manter inicialmente este campo com default=None e null=True,:
class Book(models.Model): id = models.UUIDField(primary_key=True, default=uuid.uuid4) title = models.CharField(max_length=255) pages = models.IntegerField() author = models.CharField(max_length=255) author_foreign_key = models.ForeignKey( Author, on_delete=models.CASCADE, default=None, null=True, blank=True, )- Aqui eu gerei uma migration
Agora o momento principal: Eu crio uma migration vazia, com o comando:
python manage.py makemigrations core --empty Isso nos da uma migration vazia com o seguinte templade:
# Generated by Django 4.0.1 on 2022-01-06 00:09 from django.db import migrations class Migration(migrations.Migration): dependencies = [ ('core', '0003_book_author_foreign_key'), ] operations = [ ]Renomeei essa migration para 0004_separate_authors_from_books.py
É aqui que vou escrever a lógica da migração dos dados
Talvez você precise alterar o "dono" do arquivo para poder editá-lo, caso seja necessário, utilize o comando:
sudo chown $USER 0004_separate_authors_from_books.py A migration com a lógica completa de separação dos dados ficou assim:
from django.db import migrations class Migration(migrations.Migration): def create_authors_from_books(apps, schema_editor): model_book = apps.get_model("core", "Book") model_author = apps.get_model("core", "Author") books = model_book.objects.all() for book in books: author, _ = model_author.objects.get_or_create( name=book.author ) book.author_foreign_key = author book.save() def reverse_create_authors_from_books(apps, schema_editor): model_book = apps.get_model("core", "Book") model_author = apps.get_model("core", "Author") books = model_book.objects.all() for book in books: book.author_foreign_key = None book.save() model_author.objects.all().delete() dependencies = [ ('core', '0003_book_author_foreign_key'), ] operations = [ migrations.RunPython( create_authors_from_books, reverse_create_authors_from_books ) ]TODO: Explicar melhor os métodos criados aqui.
- Aqui eu executei a migration criada acima
Com os dados migrados, eu removo o campo author da model Book
- Aqui eu gerei uma migration
Por último, eu renomeio o field author_foreign_key para um nome mais descente, no caso, author mesmo.
- Aqui eu gerei uma migration