Merge pull request #922 from NASA-IMPACT/921-add-included-urls-to-collection-model

921 add included urls to collection model

Author: code-geek

sde_collections/models/collection.py

@@ -8,6 +8,8 @@
 from django.dispatch import receiver
 from model_utils import FieldTracker
 from slugify import slugify
+from django.db import models
+from django.db.models import Q
 
 from config_generation.db_to_xml import XmlEditor
 
@@ -17,6 +19,7 @@
     format_slack_message,
     send_slack_message,
 )
+from .candidate_url import CandidateURL
 from .collection_choice_fields import (
     ConnectorChoices,
     CurationStatusChoices,
@@ -106,6 +109,10 @@ def add_to_public_query(self):
         scraper_content = scraper_editor.update_config_xml()
         gh.create_or_update_file(query_path, scraper_content)
 
+    @property
+    def included_urls_count(self):
+        return CandidateURL.objects.filter(collection=self, excluded=False).count()
+
     @property
     def _scraper_config_path(self) -> str:
         return f"sources/scrapers/{self.config_folder}/default.xml"