2016-10-07 40 views
0

我的目標是在數據庫中有一個dummyuser。我已經寫了以下功能在Django中創建dummyuser(在啓動時?)

def get_test_user(): 
    user, created = get_user_model.objects.get_or_create(username=TESTUSER_USERNAME) 
    user.set_password(TESTUSER_PASSWORD) 
    user.save 

但是在哪裏以及如何稱呼它來實現我的目標。我知道測試django會創建自己的數據庫,並且有創建測試用戶的功能。但是這實際上也應該在生產數據庫中運行。

回答

0

您可以創建data migration,例如:

from django.contrib.auth.hashers import make_password 
from django.db import migrations 


def create_user(apps, schema_editor): 
    User = apps.get_registered_model('auth', 'User') 
    user = User(
     username='user', 
     email='[email protected]', 
     password=make_password('pass'), 
     is_superuser=False, 
     is_staff=False 
    ) 
    user.save() 


class Migration(migrations.Migration): 
    dependencies = [ 
     ('auth', '0001_initial') 
    ] 

    operations = [ 
     migrations.RunPython(create_user), 
    ]