Здравствуйте, спасибо, что уделили время:
Я пытаюсь добавить некоторые данные в таблицу моделей через 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