Django создание данных с OnetoOneField - PullRequest
0 голосов
/ 03 апреля 2020

Здравствуйте, спасибо, что уделили время:

Я пытаюсь добавить некоторые данные в таблицу моделей через management / commands / create_data.py:

Вопрос

, кроме того что я думаю, что я получу ошибку в поле пользователя на модели People, потому что это должен быть идентификатор. Как мне получить идентификатор?

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)

pets / 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

1 Ответ

0 голосов
/ 03 апреля 2020

Привет, у меня пока недостаточно репутации, чтобы написать комментарий к вашему сообщению, но если вы просто хотите добавить данные для заполнения таблиц в django, я бы предложил использовать метод, описанный здесь в документации. Таким образом, это гораздо проще и без лишних хлопот.

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

В итоге:

  • создайте каталог 'fixtures' внутри вашего app.
  • создайте файл данных, который вы хотите добавить, который соответствует полям модели как json или yml.
  • call manage.py loaddata datafile_name

Я надеюсь, что помогает. Если нет, так как вы можете комментировать, дайте мне знать, и я могу удалить это.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...