Django數據初始化方法

開發中經常會遇到一些數據在程序的運行時就要在數據庫中存在，所以就需要做數據的初始化。這里簡單分享兩種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即可。