Как я могу улучшить этот «регистр» вид в Django?

Question

У меня есть сайт на Django, который позволяет пользователям регистрироваться (но требует, чтобы администратор утвердил учетную запись, прежде чем они смогут просматривать определенные части сайта). Я основываюсь на django.contrib.auth. Я требую, чтобы пользователи регистрировались по адресу электронной почты с определенного доменного имени, поэтому я переопределил методы UserCreationForm save() и clean_email().

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

def register(request):
    if request.method == 'POST':
        form = UserCreationForm(request.POST)

        if form.is_valid():
            message = None

            form.save()

            username = form.cleaned_data['username']
            password = form.cleaned_data['password1']

            user = authenticate(username=username, password=password)

            first_name = form.cleaned_data['first_name']
            last_name = form.cleaned_data['last_name']
            email = user.email

            # If valid new user account
            if (user is not None) and (user.is_active):
                login(request, user)
                message = "<strong>Congratulations!</strong> You have been registered."

                # Send emails
                try:
                    # Admin email
                    pk = None
                    try: pk = User.objects.filter(username=username)[0].pk
                    except: pass

                    admin_email_template = loader.get_template('accounts/email_notify_admin_of_registration.txt')
                    admin_email_context = Context({
                        'first_name': first_name,
                        'last_name': last_name,
                        'username': username,
                        'email': email,
                        'pk': pk,
                    })
                    admin_email_body = admin_email_template.render(admin_email_context)
                    mail_admins("New User Registration", admin_email_body)

                    # User email
                    user_email_template = loader.get_template('accounts/email_registration_success.txt')
                    user_email_context = Context({
                        'first_name': form.cleaned_data['first_name'],
                        'username': username,
                        'password': password,
                    })
                    user_email_body = user_email_template.render(user_email_context)
                    user.email_user("Successfully Registered at example.com", user_email_body)
                except:
                    message = "There was an error sending you the confirmation email. You should still be able to login normally."
            else:
                message = "There was an error automatically logging you in. Try <a href=\"/login/\">logging in</a> manually."

            # Display success page
            return render_to_response('accounts/register_success.html', {
                    'username': username,
                    'message': message,
                },
                context_instance=RequestContext(request)
            )
    else: # If not POST
        form = UserCreationForm()

    return render_to_response('accounts/register.html', {
            'form': form,
        },
        context_instance=RequestContext(request)
    )

Answer 1

Вам даже не нужен этот код, но я думаю, что стиль:

pk = None
try: pk = User.objects.filter(username=username)[0].pk
except: pass

более естественно написано как:

try:
    user = User.objects.get(username=username)
except User.DoesNotExist:
    user = None

, а затем в вашем админ-шаблоне используйте {{ user.id }} вместо {{ pk }}.

Но, как я уже сказал, вам не нужен этот код вообще, потому что у вас уже есть пользовательский объект от вашего вызова до authenticate().

Как правило, в Python считается плохой практикой иметь пустой обработчик исключений в блоке try / exception. Другими словами, всегда фиксируйте определенное исключение, такое как User.DoesNotExist для этого случая.

Это также плохая практика - иметь много строк внутри try части блока try / исключением. Лучше написать код следующим образом:

try:
    ... a line of code that can generate exceptions to be handled ...
except SomeException:
    ... handle this particular exception ...
else:
    ... the rest of the code to execute if there were no exceptions ...

И последняя рекомендация - не отправлять электронное письмо прямо на ваш взгляд, поскольку оно не будет масштабироваться, если на вашем сайте начнутся большие запросы на регистрацию. Лучше добавить в приложение django-mailer , чтобы выгрузить работу в очередь, обрабатываемую другим процессом.

Answer 2

Лично я стараюсь сначала поставить короткую ветвь оператора if-else. Особенно, если он вернется. Это позволяет избежать получения больших веток, где трудно увидеть, что и где заканчивается. Многие другие делают то же самое, что и вы, и оставляют комментарии в выражении else. Но Python не всегда имеет оператор конца блока - например, если форма не подходит для вас.

Так, например:

def register(request):
    if request.method != 'POST':
       form = UserCreationForm()
       return render_to_response('accounts/register.html', 
                                 { 'form': form, },
                                 context_instance=RequestContext(request)
                                 )

    form = UserCreationForm(request.POST)
    if not form.is_valid():
       return render_to_response('accounts/register.html', 
                                 { 'form': form, },
                                 context_instance=RequestContext(request)
                                 )
    ...

Answer 3

Первый ответ: Выглядит чертовски лучше, чем 95% вопросов «улучшить мой код».

Есть что-то, чем вы особенно недовольны?