Какой лучший метод добавления новых строк в связанных таблицах через OneToOneField?

Question

[maksam07]

Добрый вечер. Предположим, хочу расширить модель пользователя (так легче будет объяснить) вариантом OneToOneField
Возьму код из хабра:

class Profile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    bio = models.TextField(max_length=500, blank=True)
    location = models.CharField(max_length=30, blank=True)
    birth_date = models.DateField(null=True, blank=True)

Если мы создадим пользователя, например, в админке и обратимся к user.profile.bio, то будет ошибка, что строки в БД нет и таких данных тоже. На хабре рассказывают о сигналах в джанго, которые при добавлении/обновления юзера создают/обновляют строку в таблице Profile. Вариант не плохой, но что будет, если до создания модели Profile было добавлено уже несколько тысяч пользователей? У них снова же не будет этих строк в таблице Profile и строки будут создаваться только для новых пользователей.
Вопрос. Как лучше пользоваться расширением моделей посредством OneToOneField, чтобы не натыкаться на исключения, когда нет связанных строк в БД? Или может лучше как-то по-умолчанию добавлять строки в бд со значениями по умолчанию?

Answer 1

[Сергей Горностаев]

что будет, если до создания модели Profile было добавлено уже несколько тысяч пользователей

Хорошо, если будет миграция, добавляющая профили.

Как лучше пользоваться расширением моделей посредством OneToOneField, чтобы не натыкаться на исключения, когда нет связанных строк в БД?

Добавить обработку исключения.

Comments

[maksam07]

если будет миграция, добавляющая профили.

в миграции можно сделать добавление количества строк равным количеству уже созданных пользователей?

[Сергей Горностаев]

Максим Компаниец, миграция - это про python-скрипт, в котором используется джанговское API для работы с БД. Вы можете написать в миграции любой код, выполняющий любые действия с данными.

Answer 2

[Владимир]

Тут все зависит от разработчика. Можно добавить значения по умолчанию, можно добавить try-catch блоки, можно добавить логику на проверку существования профиля при логине юзера и предлагать (или даже требовать) его заполнить если нет. Конкретный выбор зависит от ситуации и от того, что больше нравится именно вам.

Comments

[maksam07]

Если рассматривать вариант значений по-умолчанию, то какое решение может быть?

[Владимир]

Максим Компаниец, поскольку для всех полей, кроме user, стоит blank=True, то и создаем для каждого существующего юзера запись с пустыми полями. Исключение уже не будет выбрасываться, можно будет нормально работать с базой.

[maksam07]

Владимир, собственно и вопрос в том, как создавать эти новые записи по-умолчанию автоматически?

[Владимир]

Максим Компаниец, если имеется в виду записи с профилем для уже существующих юзеров, то я бы наверное тупо зашел в базу и написал бы INSERT INTO Profiles (user) SELECT id FROM Users и делу конец. Ну и прописал бы в коде логику добавления профиля при добавлении юзера в дальнейшем. Но можно, наверное, и через миграции это сделать, кому как проще.

Если вопрос именно в том как добавлять профиль при создании юзера, то вот как сделал я в базе данных фехтовальщиков, где у юзера может быть также "профиль бойца":

1. Заводим нового юзера и делаем переадресацию на создание профиля бойца:

def users_add(request):
    if request.user.is_authenticated:
        return HttpResponseRedirect('/')
    if request.method == 'POST':
        # create a form instance and populate it with data from the request:
        form = UserCreationForm(request.POST)
        # check whether it's valid:
        if form.is_valid():
            # process the data in form.cleaned_data as required
            user = form.save()
            password = form.cleaned_data['password']
            user.set_password(password)
            user.save()
            login(request, user)
            # redirect to a new URL:
            return HttpResponseRedirect('/fighters/add/')
        # if a GET (or any other method) we'll create a blank form
    else:
        form = UserCreationForm()
    return render(request, 'users/add_user.html', {'form': form})

2. Заводим профиль бойца и редактируем его:

@login_required()
def fighter_add(request):
    user = request.user
    if hasattr(user, 'fighter'):
        return HttpResponseRedirect('/')
    else:
        user.groups.add(Group.objects.get(name='Fighters'))
        user.save()
        fighter = Fighters(user=user, first_name=user.first_name, last_name=user.last_name, email=user.email)
        fighter.save()
        # Редирект на редактирование свежесозданной записи
        return HttpResponseRedirect("/fighters/edit/")