From d5fc0d190874a115209db7b35d5b23a81d4ba680 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Louis=20V=C3=A9zina?=
 <5130500+morpheus65535@users.noreply.github.com>
Date: Mon, 3 Aug 2020 11:32:31 -0400
Subject: [PATCH] Fix for #1051.

---
 views/_main.html | 31 +++++++++++++++++++------------
 1 file changed, 19 insertions(+), 12 deletions(-)

diff --git a/views/_main.html b/views/_main.html
index 0f24d1d49..24008b2c7 100644
--- a/views/_main.html
+++ b/views/_main.html
@@ -254,16 +254,23 @@
                             </ul>
                         </li>
 
-                        <li><a href="#"><i class="fas fa-exclamation-triangle"></i><span
-                                class="hide-menu"> Wanted</span></a>
+                        <li><a href="#"><i class="fas fa-exclamation-triangle"></i>
+                            <span class="hide-menu"> Wanted </span>
+                            {% if settings.general.getboolean('use_sonarr') %}
+                            <span class="wanted_series_badge"></span>
+                            {% endif %}
+                            {% if settings.general.getboolean('use_radarr') %}
+                            <span class="wanted_movies_badge"></span>
+                            {% endif %}
+                        </a>
                             <ul aria-expanded="false" class="collapse">
                                 {% if settings.general.getboolean('use_sonarr') %}
                                     <li>
-                                        <a href="{{ url_for('wantedseries') }}"> Series <span id="wanted_series_badge"></span></a>
+                                        <a href="{{ url_for('wantedseries') }}"> Series <span class="wanted_series_badge"></span></a>
                                     </li>
                                 {% endif %}
                                 {% if settings.general.getboolean('use_radarr') %}
-                                    <li><a href="{{ url_for('wantedmovies') }}"> Movies <span id="wanted_movies_badge"></span></a></li>
+                                    <li><a href="{{ url_for('wantedmovies') }}"> Movies <span class="wanted_movies_badge"></span></a></li>
                                 {% endif %}
                             </ul>
                         </li>
@@ -283,11 +290,11 @@
                         </li>
 
                         <li><a href="#"><i
-                                class="fas fa-laptop"></i><span class="hide-menu"> System</span></a>
+                                class="fas fa-laptop"></i><span class="hide-menu"> System </span><span class="throttled_providers_count"></span></a>
                             <ul aria-expanded="false" class="collapse">
                                 <li><a href="{{ url_for('systemtasks') }}"> Tasks</a></li>
                                 <li><a href="{{ url_for('systemlogs') }}"> Logs</a></li>
-                                <li><a href="{{ url_for('systemproviders') }}"> Providers <span id="throttled_providers_count"></span></a></li>
+                                <li><a href="{{ url_for('systemproviders') }}"> Providers <span class="throttled_providers_count"></span></a></li>
                                 <li><a href="{{ url_for('systemstatus') }}"> Status</a></li>
                                 <li><a href="{{ url_for('systemreleases') }}"> Releases</a></li>
                             </ul>
@@ -426,19 +433,19 @@
                     async: true,
                     success: function (data) {
                         if (data['throttled_providers']) {
-                            $('#throttled_providers_count').html('<span class="badge badge-secondary">' + data['throttled_providers'] + '</span>');
+                            $('.throttled_providers_count').html('<span class="badge badge-secondary">' + data['throttled_providers'] + '</span>');
                         } else {
-                            $('#throttled_providers_count').html('');
+                            $('.throttled_providers_count').html('');
                         }
                         if (data['missing_episodes']) {
-                            $('#wanted_series_badge').html('<span class="badge badge-secondary">' + data['missing_episodes'] + '</span>');
+                            $('.wanted_series_badge').html('<span class="badge badge-secondary">' + data['missing_episodes'] + '</span>');
                         } else {
-                            $('#wanted_series_badge').html('');
+                            $('.wanted_series_badge').html('');
                         }
                         if (data['missing_movies']) {
-                            $('#wanted_movies_badge').html('<span class="badge badge-secondary">' + data['missing_movies'] + '</span>');
+                            $('.wanted_movies_badge').html('<span class="badge badge-secondary">' + data['missing_movies'] + '</span>');
                         } else {
-                            $('#wanted_movies_badge').html('');
+                            $('.wanted_movies_badge').html('');
                         }
 
                     },