for pro proposition banner, use location count if v3 enabled

Author: dogboat

dojo/tasks.py

@@ -14,6 +14,7 @@
 from dojo.celery import app
 from dojo.decorators import dojo_async_task
 from dojo.finding.helper import fix_loop_duplicates
+from dojo.location.models import Location
 from dojo.management.commands.jira_status_reconciliation import jira_status_reconciliation
 from dojo.models import Alerts, Announcement, Endpoint, Engagement, Finding, Product, System_Settings, User
 from dojo.notifications.helper import create_notification
@@ -223,7 +224,11 @@ def evaluate_pro_proposition(*args, **kwargs):
     ):
         return
     # Count the objects the determine if the banner should be updated
-    object_count = Finding.objects.count() + Endpoint.objects.count()
+    if settings.V3_FEATURE_LOCATIONS:
+        object_count = Finding.objects.count() + Location.objects.count()
+    else:
+        # TODO: Delete this after the move to Locations
+        object_count = Finding.objects.count() + Endpoint.objects.count()
     # Unless the count is greater than 100k, exit early
     if object_count < 100000:
         return