开发中经常会遇到一些数据在程序的运行时就要在数据库中存在,所以就需要做数据的初始化。这里简单分享两种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)
]
最后生成的文件如下图所示: