Django笔记二十之手动编写migration文件

巅峰抱团暮年刷

本文首发于公众号：Hunter后端
原文链接：Django笔记二十之手动编写migration文件

前面介绍过，migration 文件主要记录的是 Django 系统 model 的变化，然后通过 migrate 命令将变化适配到数据库中。
比如在某个 application 下新增了某张表，或者对某张表更改了字段，可以生成 migration 文件，然后通过 migrate 更改到数据库。
除了系统能够自动生成的，我们还可以手动创建 migration 文件来操作数据库，这个用途主要是用于比如，创建表后，需要写入一些初始化的数据的情况。

• 基础命令
  
• migration文件介绍
  
• 自定义migration文件
  
• RunSQL()
  
• RunPython()
  

1、基础命令

关于 migration 的命令有如下几条：

• makemigrations
  
• migrate
  
• sqlmigrate
  
• showmigrations
  

其中 前面三条命令在第二篇笔记中已经介绍过使用方法，这里介绍一下 showmigrations。
这个作用主要是查看某个 application 下的migration 文件是否已经被更改到数据库中，可以在 Django 系统的根目录用下面的命令测试：

1. python3 manage.py showmigrations blog

复制代码

可以看到下面的输出：

1. blog
2. [X] 0001_initial
3. [X] 0002_auto_20220118_0926
4. [X] 0003_auto_20220121_1016

复制代码

其中，前面的 [X] 表示已经被更改到数据库中，如果我们再对 blog 的 model 进行任意修改，然后执行 makemigrations 的操作，再次执行 showmigrations 的操作，可以看到下面的输出：

1. blog
2. [X] 0001_initial
3. [X] 0002_auto_20220118_0926
4. [X] 0003_auto_20220121_1016 [ ] 0004_alter_book_price

复制代码

可以看到最下面的一条记录 [] 中是没有 X 的，表示这条 migration 文件没有被执行 migrate。
2、migration文件介绍

每一次通过 makemigrations 生成的 migration 文件都存在系统中，一个最基础的 migration 文件像下面这样：

1. from django.db import migrations, models
4. class Migration(migrations.Migration):
7.     dependencies = [('blog', '0001_initial')]
10.     operations = [
11.         migrations.DeleteModel('Tribble'),
12.         migrations.AddField('Author', 'rating', models.IntegerField(default=0)),
13.     ]

复制代码

一个 Migration 的类下，有两个参数，一个是 dependencies，一个是 operations
dependencies 作用是定位上一个执行的 migration 文件的地方，因为每一次 migrate 的执行都是按照顺序的
且他的参数是一个列表，列表的元素是一个元组，里面有两个参数，一个是 application 的名称，一个是上一次运行的 migration 文件，他是可以指定到多个 application 的，意义为在某两个 application 的 migration 文件之后再执行
operations 的作用是 migration 里需要执行的操作，可以是字段的增加、删除、修改、也可以是表的创建和删除
一个 migration 在执行 migrate 前，我们可以手动对其修改，甚至可以完全自己来定义
3、自定义migration文件

前面介绍了 migration 文件的基本结构，其中有一些关于字段和 model 的操作方法，这些操作都可以通过 makemigration 的方式自动生成。
我们自定义的 migration 文件，与上面的保持一致即可，自定义的 migration 文件需要修改的地方是 operations 里的元素。
假设我们有这样一个需求，创建一张基础映射表后，里面是系统运行所必需的数据，需要在创建表后立即写入，那么就用到了我们这个自定义的 migration 文件。
除了对表字段或者表的修改，还有两种方法实现数据的写入，
一种是使用 SQL 语句插入，用到的migration的函数是 RunSQL()
一种是使用 Django 的 ORM 语句，写 python 的函数来插入，函数是 RunPython
假设创建 Blog 表的migration file 是 0001_create_blog.py
现在需要对其插入两条数据，name 和 tagline 分别是 ('name_1', 'tagline_1') 和 ('name_2', 'tagline_2')
下面用 RunSQL() 和 RunPython() 两种方式来分别介绍。
4、RunSQL()

RunSQL() 函数接受一个字符串，或者一个数组作为参数，参数的内容都是 SQL 语句，这也是为什么函数名为 RunSQL()。
字符串的形式为完整的 SQL 语句，比如我们需要插入这两条数据，则是：

1. migrations.RunSQL(
2.         "INSERT INTO blog_blog (name, tagline) values('name_x_4', 'tagline_1'), ('name_x_5', 'tagline_2');"
3. )

复制代码

如果是作为数组传入，形式则是：

1. migrations.RunSQL(
2.     sql=[
3.         (
4.             "INSERT INTO blog_blog (name, tagline) values(%s, %s), (%s, %s);",
5.             ['name_x_6', 'tagline_1', 'name_x_7', 'tagline_2']
6.         )
7.     ]
8. )

复制代码

在数组的传入形式中，我们将需要插入的数据都放到一个数组中传入
reverse_sql
RunSQL() 函数除了 sql 参数，还有一个 reverse_sql 参数，用途是 sql 参数执行的 SQL 语句没有执行成功的情况下的一种操作，一般是用于防止数据污染。
假设说我们的 sql 为插入数据，但是因为某种原因，这条语句没有正确插入，报错了，那么系统就会执行 reverse_sql 中的语句，作为一个可逆的操作。
以下是官方的一个使用示例：

1. migrations.RunSQL(
2.     sql=[("INSERT INTO musician (name) VALUES (%s);", ['Reinhardt'])],
3.     reverse_sql=[("DELETE FROM musician where name=%s;", ['Reinhardt'])],
4. )

复制代码

5、RunPython()

RunSQL() 函数操作的是 SQL 语句，RunPython() 参数则是 Python 函数，可以将我们需要写入的数据都写到函数的步骤里，然后在 RunPython() 中调用
以下是使用示例：

1. def insert_blog_data(apps, schema_editor):
2.     Blog = apps.get_model("blog", "Blog")
3.     db_alias = schema_editor.connection.alias
5.     Blog.objects.using(db_alias).create(name="name_3", tagline="tagline_3")
6.     Blog.objects.using(db_alias).create(name="name_4", tagline="tagline_4")
10. class Migration(migrations.Migration):
11.     dependencies = [
12.         ("blog", "0001_initial"),
13.     ]
16.     operations = [
17.         migrations.RunPython(insert_blog_data)
18.     ]

复制代码

其中，insert_blog_data 是需要执行的函数，在这个函数里，有两个默认参数，apps 和 schema_editor
apps 可以用来获取我们需要的 model，根据函数 apps.get_model()，
这个函数传入两个参数，一个是 application，我们这里是 blog，
一个 model 的名称，我们这里是 Blog
而 schema_editor 则是可以用于获取数据库的 alias
然后，数据的插入的方式就和普通的 model 的操作方法一致了。
RunPython() 函数和 RunSQL 一样，也可以输入两个参数，第二个参数作用也是用于操作失败的回退操作：

1. migrations.RunPython(insert_blog_data, reverse_insert)

复制代码

以上就是介绍 migration 的全部内容了，下一篇笔记将介绍如何在 Django 中使用原生的 SQL 来查询数据。
如果想获取更多后端相关文章，可扫码关注阅读：

来源:https://www.cnblogs.com/hunterxiong/p/17300914.html
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！