Как написать middleware для вывода дополнительной строки на каждой странице?

Question

[Ирина Пахомова]

Разбираюсь понемногу с middleware. Но разобраться пока не могу.
Вот такой пример у меня не работает: python.su/forum/topic/11368
Пишу я тоже process_response. В доках написано, что response is the HttpResponse. Проверяю, у меня он почему-то TemplateResponse. Это у меня что-то во вьюшках не так? Ну ладно, сделала вот так:
class MyOwnMiddleWare(object):
def process_response(self, request, response):
content = response.rendered_content
content = content.replace('</body>', ' my piece of text '</body>')
response = HttpResponse(content, content_type="text/html")
return response

И на некоторых страницах это работает. А на некоторых нет :))) Возникает ошибка 'HttpResponseRedirect' object has no attribute 'rendered_content', То есть response уже другой объект уже получается.
Как мне это поправить? Я вообще так делаю? А то вот еще есть process_template_response метод, может его надо?

Comments

[sim3x]

за такое, сильно по попе бьют - так делать не нужно

[Ирина Пахомова]

sim3x: объяснитесь, пожалуйста! За какое? А как нужно? Чем бьют?

[sim3x]

Ирина:
за миддлвару
за инжекцию кода
за регулярки
за переписывание контент тайпа

Нужно определить темлейт от которого будут наследоваться все остальные и в него включить ' my piece of text '

Бьют тяжелым - чтоб запомнилось навсегда

[Ирина Пахомова]

sim3x: ну будьте вы уж добреньким, объясните поподробнее! Написала же, что только разбираюсь!
Базовый темплейт уже написала.
А что такое инжекция кода? При чем тут регулярки?

[Ирина Пахомова]

sim3x: врпрочем, про регулярки я, кажется, поняла.

[sim3x]

Ирина:
от мидлвар стоит отказаться - есть куча способов, которые делают дела лучше и понятнее
Мидлвары нужны для тех кто делает автоновные аппы для переиспользования и установки из pip

регулярки - писать правильно умеют единицы
в текущем варианте ты не учла такие body>

[Ирина Пахомова]

спасибо! ну, в общем, да, я немного покопалась, то, что пыталась сделать - сделала, но не думаю, что буду лезть в мидлвар

Answer 1

[Oscar Django]

А то вот еще есть process_template_response метод, может его надо?

Да, именно так. Оказалось не совсем.

Пишу я тоже process_response. В доках написано, что response is the HttpResponse. Проверяю, у меня он почему-то TemplateResponse.

TemplateResponse наследуется от SimpleTemplateResponse, для которого родительским является HttpResponse. Поэтому "response is the HttpResponse".

Возникает ошибка 'HttpResponseRedirect' object has no attribute 'rendered_content', То есть response уже другой объект уже получается.

rendered_content -- это свойство (property), которое устанавливается в SimpleTemplateResponse, который, как я говорил выше, является родительским классом для TemplateResponse. Но HttpResponseRedirect(HttpResponseRedirectBase) наследуется напрямую от HttpResponse, поэтому у него нет свойства rendered_content.

по поводу отличия process_response и process_template_response

process_template_response выполняется только если у response есть метод render, а process_response выполняется всегда. Логику можно посмотреть process_template_response и process_response
Метод render задается в классе SimpleTemplateResponse от которого, как мы выяснили, наследуется TemplateResponse, но не HttpResponseRedirect.

Comments

[Ирина Пахомова]

спасибо большое за такой подробный ответ! Я даже не ожидала такого! Кое-что для меня прояснилось, но пока еще не все.
Получается, что на случай переадресации мне надо проверить код статуса, как советуют выше?
И еще, никак в толк не возьму, в примере содержание правили напрямую. С response.rendered_content это не проходит. То есть в случае process_template_response мы не напрямую правим, а добавляем переменную в контекст?

[Oscar Django]

> То есть в случае process_template_response мы не напрямую правим, а добавляем переменную в контекст?
Ваша правда -- моя вина. Совсем упустил, что render происходит уже после всех process_template_response. А значит изменить что в уже отрендеренном шаблоне возможно только в process_response.
> Получается, что на случай переадресации мне надо проверить код статуса, как советуют выше?
Полагаю, это действительно самый простой вариант.
Так писать не надо
response = HttpResponse(content, content_type="text/html")
Лучше так
response.content = content
А вместо
content = response.rendered_content
лучше
content = response.content

PS ну и, как и сказал sim3x, не надо таким образом что то добавлять. Поиграться чтобы разобраться как работает можно, но в реальной работе не стоит.

[Ирина Пахомова]

Oscar Django: спасибо большое! Там с просто content еще возникали проблемы с переводом байтов в юникод и обратно, поэтому я просто добавила в process_template_response добавление переменных контекста, и на этом и оставила.

Answer 2

[un1t]

Проверяй response.status_code - если 200 то подменяй, если другой то не подменяй.

Comments

[Ирина Пахомова]

спасибо за совет! возьму на заметку!