Django数据初始化方法

Posted 2020-11-27 wangyingblock

开发中经常会遇到一些数据在程序的运行时就要在数据库中存在，所以就需要做数据的初始化。这里简单分享两种Django的数据初始化方法。

背景：项目app：role有两个mode: RoleType和UserRole，modol中已经写好了初始化方法：init_builtin_roles()和init_builtin_user_roles()

一、使用migrations文件

• 1、创建migrations文件

 python manage.py makemigrations --empty role

其中role是app名称

会生成如下图所示文件：

• 2、在生成的文件中编写初始化数据方法

def init_data(apps, schema_editor):
    try:
        RoleType.init_builtin_roles()
        UserRole.init_builtin_user_roles()
    except Exception as e:
        logger.error(u‘数据初始化error：{}‘.format(e))

然后在operations中调用初始化方法：

operations = [
        migrations.RunPython(init_data)
    ]

最后生成的文件如下图所示：

• 3、执行python manage.py migrate 即可完成数据初始化

二、使用AppConfig与信号post_migrate

• 1、在app:role文件夹中创建apps.py文件, 在apps.py中编写初始化数据方法：

def app_ready_handler(sender, **kwargs):
    from models import RoleType, UserRole
    print ‘init builtin roles and user roles data‘
    try:
        RoleType.init_builtin_roles()
        UserRole.init_builtin_user_roles()
    except Exception as e:
        print ‘init builtin roles and user roles data exception: %s‘ % e

• 2、继承AppConfig创建RoleConfig,在ready()方法中使用信号post_migrate来调用初始化方法：

信号post_migrate : 执行migrate命令后,自动触发

• 3、配置default_app_config：默认情况下Django会自动走AppConfig,如果有改动需要重新配置。

在role下面的_init_.py文件中配置default_app_config

• 4、最后执行python manage.py migrate即可。