Django 使用 OnetoOneField 创建数据

Question

您好，感谢您的宝贵时间：

我正在尝试通过 management/commands/create_data.py:

向我的模型 table 添加一些数据

问题

除此之外，我想我会在 People 模型的用户字段中收到错误，因为应该是一个 ID。我如何获取 ID？

pets/models.py:


User = get_user_model()



class People(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='person')
    birthday = models.DateField()
    cpf = models.CharField(max_length=11, validators=[RegexValidator(r'^\d{1,10}$')])

    def __str__(self):
        return '%s' % (self.user)

宠物/management/commands/create_data.py:

User = get_user_model()


class GenerateUser():
    username = None
    password = None
    def __init__(self, username, password):
        with open('User.txt') as file:
            for rows in file:
                self.username = slugify(rows)
                self.password = '12345'


class GeneratePeople():
    user = None
    documento = None
    birthday = None
    def __init__(self, documento, birthday):
        with open('People.txt') as file:
            for rows in file:
                t1 = rows.split(', ')
                self.user = slugify(t1[0])
                self.documento = t1[1]
                self.birthday = t1[2]



class Command(BaseCommand):

    def add_arguments(self, parser):
        parser.add_argument(
            'file_name', type=str, help='ok')

    def handle(self, *args, **kwargs):

        username = GenerateUser.username
        password = GenerateUser.password
        user = GeneratePeople.user
        documento = GeneratePeople.documento
        birthday = GeneratePeople.birthday

        one = User(
            username=username,
            password=password
            )

        one.save()

        peop = People(
            user=user,
            documento=documento,
            birthday=birthday
        )

        peop.save()


        self.stdout.write(self.style.SUCCESS('Data imported successfully'))

People.txt:

Johnny Cash,    555555555,  1932-02-26
Sid Vicious,    555555555,  1957-05-10
Axl Rose,   555555555,  1962-02-06
Joey Ramone,    555555555,  1951-05-19
Bruce Dickinson,    555555555,  1958-08-07
Kurt Cobain,    555555555,  1967-02-20
Elvis Presley,  555555555,  2008-19-17

User.txt:

Johnny Cash
Sid Vicious
Axl Rose
Joey Ramone
Bruce Dickinson
Kurt Cobain
Elvis Presley

Answer 1

您好，我还没有足够的声誉来对您的 post 发表评论，但是如果您只想添加数据以填充 django 中的表，我建议您使用文档中描述的方法。这样就轻松多了。

https://docs.djangoproject.com/en/3.0/howto/initial-data/

总结：

在您的应用程序中创建一个 'fixtures' 目录。
创建您要添加的与模型字段匹配的数据文件 json 或 yml。
致电manage.py loaddata datafile_name

希望对您有所帮助。如果没有，因为你可以评论让我知道，我可以删除它。

Django 使用 OnetoOneField 创建数据

Django create data with OnetoOneField

python

django

django-models

django-database

python-3.x