From aa0de32cd054266e9d0360c2447841c71bf0d431 Mon Sep 17 00:00:00 2001
From: Santiago Lamora <santiago@ribaguifi.com>
Date: Thu, 23 Nov 2023 11:09:45 +0100
Subject: [PATCH] Refactor my-profile & fix set language

---
 .../musician/templates/musician/base.html     |  2 +-
 .../musician/templates/musician/profile.html  |  2 +-
 orchestra/contrib/musician/views.py           | 22 +++++++++++--------
 3 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/orchestra/contrib/musician/templates/musician/base.html b/orchestra/contrib/musician/templates/musician/base.html
index bfe32585..537b08db 100644
--- a/orchestra/contrib/musician/templates/musician/base.html
+++ b/orchestra/contrib/musician/templates/musician/base.html
@@ -94,7 +94,7 @@
             </a>
             <div class="dropdown-menu">
               {% for code, language in languages %}
-                  <a class="dropdown-item" href="{% url 'musician:profile-set-lang' code %}">{{ language }}</a>
+                  <a class="dropdown-item" href="{% url 'musician:profile-set-lang' code %}?next={{ request.path }}">{{ language }}</a>
               {% endfor %}
             </div>
           </div>
diff --git a/orchestra/contrib/musician/templates/musician/profile.html b/orchestra/contrib/musician/templates/musician/profile.html
index f6bdbced..8c76c514 100644
--- a/orchestra/contrib/musician/templates/musician/profile.html
+++ b/orchestra/contrib/musician/templates/musician/profile.html
@@ -18,7 +18,7 @@
       <div class="col-md-9">
         <p class="card-text">{{ profile.username }}</p>
         <p class="card-text">{{ profile.type }}</p>
-        <p class="card-text">{% trans "Preferred language:" %} {{ profile.language|language_name_local }}</p>
+        <p class="card-text">{% trans "Preferred language:" %} {{ preferred_language_code|language_name_local }}</p>
       </div>
       {% comment %}
       <!-- disabled until set_password is implemented -->
diff --git a/orchestra/contrib/musician/views.py b/orchestra/contrib/musician/views.py
index b48f1c6a..05beac5f 100644
--- a/orchestra/contrib/musician/views.py
+++ b/orchestra/contrib/musician/views.py
@@ -1,4 +1,3 @@
-import datetime
 import logging
 import smtplib
 
@@ -123,13 +122,10 @@ class ProfileView(CustomContextMixin, UserTokenRequiredMixin, TemplateView):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        try:
-            pay_source = self.orchestra.retrieve_service_list(
-                PaymentSource.api_name)[0]
-        except IndexError:
-            pay_source = {}
+        user = self.request.user
         context.update({
-            'payment': PaymentSource.new_from_json(pay_source)
+            'payment': user.paymentsources.first(),
+            'preferred_language_code': user.language.lower(),
         })
 
         return context
@@ -139,11 +135,19 @@ def profile_set_language(request, code):
     # set user language as active language
 
     if any(x[0] == code for x in settings.LANGUAGES):
-        # http://127.0.0.1:8080/profile/setLang/es
         user_language = code
         translation.activate(user_language)
 
-        response = HttpResponseRedirect('/dashboard')
+        redirect_to = request.GET.get('next', '')
+        url_is_safe = is_safe_url(
+            url=redirect_to,
+            allowed_hosts={request.get_host()},
+            require_https=request.is_secure(),
+        )
+        if not url_is_safe:
+            redirect_to = reverse_lazy(settings.LOGIN_REDIRECT_URL)
+
+        response = HttpResponseRedirect(redirect_to)
         response.set_cookie(settings.LANGUAGE_COOKIE_NAME, user_language)
 
         return response