Add course profile link to overview

Closes #78

Author: alexeygrigorev

courses/templates/courses/course_list.html

@@ -115,11 +115,21 @@ <h3 class="mt-2 text-xl font-semibold md:text-2xl">
                             {% if course.home_registration_open %}
                                 <div class="flex flex-wrap gap-3 md:justify-end">
                                     <a href="{{ course.registration_url }}"
-                                       class="inline-flex items-center font-medium app-link">
+                                       class="primer-button primer-button-secondary whitespace-nowrap">
+                                        <i class="fas fa-user-plus"></i>
                                         Register
                                     </a>
                                 </div>
                             {% endif %}
+                            {% if course.home_is_enrolled %}
+                                <div class="flex flex-wrap gap-3 md:justify-end">
+                                    <a href="{% url 'enrollment' course.slug %}"
+                                       class="primer-button primer-button-secondary whitespace-nowrap">
+                                        <i class="fas fa-user-edit"></i>
+                                        Edit profile
+                                    </a>
+                                </div>
+                            {% endif %}
                         </div>
                     </article>
                 {% endfor %}
@@ -151,6 +161,11 @@ <h4 class="text-lg font-semibold app-link md:text-xl">{{ course.title }}</h4>
                                         <div class="flex flex-wrap gap-x-3 gap-y-1 text-sm app-muted md:justify-end">
                                             <span>{{ course.homework_count }} homework</span>
                                             <span>{{ course.project_count }} projects</span>
+                                            {% if course.home_is_enrolled %}
+                                                <span>
+                                                    <a href="{% url 'enrollment' course.slug %}" class="app-link">Edit profile</a>
+                                                </span>
+                                            {% endif %}
                                         </div>
                                         <span class="hidden shrink-0 app-link md:block" aria-hidden="true">&rarr;</span>
                                     </a>

courses/tests/test_course.py

@@ -983,6 +983,36 @@ def test_course_list_shows_registration_before_course_start(self):
             response,
             "https://courses.datatalks.club/test-course/register",
         )
+
+    def test_course_list_shows_edit_profile_for_enrolled_courses(self):
+        other_course = Course.objects.create(
+            title="Other Course",
+            slug="other-course",
+            visible=True,
+        )
+
+        self.client.login(**credentials)
+
+        url = reverse("course_list")
+        response = self.client.get(url)
+
+        self.assertEqual(response.status_code, 200)
+
+        courses_by_slug = {
+            course.slug: course
+            for course in response.context["active_courses"]
+        }
+
+        self.assertTrue(courses_by_slug[self.course.slug].home_is_enrolled)
+        self.assertFalse(courses_by_slug[other_course.slug].home_is_enrolled)
+        self.assertContains(
+            response,
+            reverse("enrollment", kwargs={"course_slug": self.course.slug}),
+        )
+        self.assertNotContains(
+            response,
+            reverse("enrollment", kwargs={"course_slug": other_course.slug}),
+        )
 
     def test_hidden_course_accessible_via_direct_link(self):
         """Test that non-visible courses are still accessible via direct link"""

courses/views/course.py

@@ -157,11 +157,17 @@ def current_assignment_info(course: Course, now) -> tuple[str, dict | None]:
     return "Last assignment", assignments[-1]
 
 
-def add_course_homepage_info(course: Course, now) -> None:
+def add_course_homepage_info(
+    course: Course,
+    now,
+    enrolled_course_ids: set[int] | None = None,
+) -> None:
     today = timezone.localdate(now)
+    enrolled_course_ids = enrolled_course_ids or set()
 
     course.home_outcome = get_course_outcome(course)
     course.home_year = course_year(course)
+    course.home_is_enrolled = course.id in enrolled_course_ids
     course.home_duration_label = course_duration_label(course)
     course.home_registration_open = (
         bool(course.registration_url)
@@ -176,6 +182,16 @@ def add_course_homepage_info(course: Course, now) -> None:
 
 def course_list(request):
     now = timezone.now()
+    enrolled_course_ids = set()
+
+    if request.user.is_authenticated:
+        enrolled_course_ids = set(
+            Enrollment.objects.filter(student=request.user).values_list(
+                "course_id",
+                flat=True,
+            )
+        )
+
     courses = (
         Course.objects.filter(visible=True)
         .annotate(
@@ -192,7 +208,7 @@ def course_list(request):
     archive_courses_by_year = defaultdict(list)
 
     for course in courses:
-        add_course_homepage_info(course, now)
+        add_course_homepage_info(course, now, enrolled_course_ids)
 
         if course.finished:
             finished_courses.append(course)