[REF] website_sale: tokenization - improve desgin

Update design for registered cards

@courtesy of @fhe-odoo

1 files changed, 3 insertions, 2 deletions

diff --git a/addons/website_payment/controllers/main.py b/addons/website_payment/controllers/main.py
index fc3441f0b5f..ed7c0ade11d 100644
--- a/addons/website_payment/controllers/main.py
+++ b/addons/website_payment/controllers/main.py
@@ -7,7 +7,7 @@ from odoo.http import request
 
 class WebsitePayment(http.Controller):
     @http.route(['/my/payment_method'], type='http', auth="user", website=True)
-    def payment_method(self):
+    def payment_method(self, **kwargs):
         acquirers = list(request.env['payment.acquirer'].search([('website_published', '=', True), ('registration_view_template_id', '!=', False)]))
         partner = request.env.user.partner_id
         payment_tokens = partner.payment_token_ids
@@ -16,8 +16,9 @@ class WebsitePayment(http.Controller):
             'pms': payment_tokens,
             'acquirers': acquirers
         }
+        return_url = request.params.get('redirect', '/my/payment_method')
         for acquirer in acquirers:
-            acquirer.form = acquirer.sudo()._registration_render(request.env.user.partner_id.id, {'error': {}, 'error_message': [], 'return_url': '/my/payment_method', 'json': False, 'bootstrap_formatting': True})
+            acquirer.form = acquirer.sudo()._registration_render(request.env.user.partner_id.id, {'error': {}, 'error_message': [], 'return_url': return_url, 'json': False, 'bootstrap_formatting': True})
         return request.render("website_payment.pay_methods", values)
 
     @http.route(['/website_payment/delete/'], methods=['POST'], type='http', auth="user", website=True)