开发中经常会遇到一些数据在程序的运行时就要在数据库中存在,所以就需要做数据的初始化。这里简单分享两种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
即可。
原文地址:https://www.cnblogs.com/wangyingblock/p/12668463.html
时间: 2024-10-20 05:58:58