Fix source_completeness field display and filtering behaviour

django/cantusdb_project/main_app/forms.py

@@ -73,7 +73,6 @@ def label_from_instance(self, obj):
     widget = CheckboxSelectMultiple()
 
 
-
 class CantusDBLatinField(forms.CharField):
     """
     A custom CharField for chant text fields. Validates that the text
@@ -107,6 +106,7 @@ def validate(self, value):
             except ValueError as exc:
                 raise forms.ValidationError("Invalid characters in text.") from exc
 
+
 class StyledChoiceField(forms.ChoiceField):
     """
     A custom ChoiceField that uses the custom SelectWidget defined in widgets.py
@@ -116,7 +116,6 @@ class StyledChoiceField(forms.ChoiceField):
     widget = SelectWidget()
 
 
-
 class ChantCreateForm(forms.ModelForm):
     class Meta:
         model = Chant
@@ -314,7 +313,7 @@ class Meta:
         required=False,
     )
 
-    TRUE_FALSE_CHOICES_INVEN = ((True, "Complete"), (False, "Incomplete"))
+    TRUE_FALSE_CHOICES_INVEN = ((True, "Full inventory"), (False, "Partial inventory"))
 
     complete_inventory = StyledChoiceField(
         choices=TRUE_FALSE_CHOICES_INVEN, required=False
@@ -510,8 +509,8 @@ class Meta:
     )
 
     CHOICES_COMPLETE_INV = (
-        (True, "complete inventory"),
-        (False, "partial inventory"),
+        (True, "Full inventory"),
+        (False, "Partial inventory"),
     )
     complete_inventory = StyledChoiceField(choices=CHOICES_COMPLETE_INV, required=False)
 
@@ -763,15 +762,13 @@ class Meta:
     #     help_text="RISM-style siglum + Shelf-mark (e.g. GB-Ob 202).",
     # )
 
-
     shelfmark = forms.CharField(
         required=True,
         widget=TextInputWidget,
     )
 
     name = forms.CharField(required=False, widget=TextInputWidget)
 
-
     holding_institution = forms.ModelChoiceField(
         queryset=Institution.objects.all().order_by("city", "name"),
         required=False,
@@ -841,7 +838,7 @@ class Meta:
         widget=FilteredSelectMultiple(verbose_name="other editors", is_stacked=False),
     )
 
-    TRUE_FALSE_CHOICES_INVEN = ((True, "Complete"), (False, "Incomplete"))
+    TRUE_FALSE_CHOICES_INVEN = ((True, "Full inventory"), (False, "Partial inventory"))
 
     complete_inventory = forms.ChoiceField(
         choices=TRUE_FALSE_CHOICES_INVEN, required=False

django/cantusdb_project/main_app/management/commands/populate_source_completeness.py

@@ -12,7 +12,7 @@ class Command(BaseCommand):
     def handle(self, *args, **options):
         sources = Source.objects.all()
         for source in sources:
-            if source.full_source:
+            if source.full_source or source.full_source is None:
                 source.source_completeness = (
                     source.SourceCompletenessChoices.FULL_SOURCE
                 )

django/cantusdb_project/main_app/models/source.py

@@ -78,14 +78,14 @@ class Source(BaseModel):
     )
 
     class SourceCompletenessChoices(models.IntegerChoices):
-        FULL_SOURCE = 1, "Full source"
-        FRAGMENT = 2, "Fragment/Fragmented"
+        FULL_SOURCE = 1, "Complete source"
+        FRAGMENT = 2, "Fragment"
         RECONSTRUCTION = 3, "Reconstruction"
 
     source_completeness = models.IntegerField(
         choices=SourceCompletenessChoices.choices,
         default=SourceCompletenessChoices.FULL_SOURCE,
-        verbose_name="Full Source/Fragment",
+        verbose_name="Complete Source/Fragment",
     )
 
     full_source = models.BooleanField(blank=True, null=True)

django/cantusdb_project/main_app/templates/source_create.html

@@ -162,7 +162,7 @@ <h3>Create Source</h3>
                 <div class="form-row">
                     <div class="form-group m-1 col-lg-2">
                         <label for="{{ form.complete_inventory.id_for_label }}">
-                            <small>Complete / partial inventory:</small>
+                            <small>Full / partial inventory:</small>
                         </label>
                         {{ form.complete_inventory }}
                     </div>

django/cantusdb_project/main_app/templates/source_detail.html

@@ -113,11 +113,11 @@ <h3>{{ source.heading }}</h3>
     {% endif %}
 
     {% if source.complete_inventory is not None %}
-        <dt>Complete/Partial Inventory</dt>
-        <dd>{{ source.complete_inventory|yesno:"Complete Inventory,Partial Inventory" }}</dd>
+        <dt>Full/Partial Inventory</dt>
+        <dd>{{ source.complete_inventory|yesno:"Full Inventory,Partial Inventory" }}</dd>
     {% endif %}
 
-    <dt>Full Source/Fragment</dt>
+    <dt>Complete Source/Fragment</dt>
     <dd>{{ source.get_source_completeness_display }}</dd>
 
     {% if user.is_authenticated %}

django/cantusdb_project/main_app/templates/source_edit.html

@@ -170,7 +170,7 @@ <h3>
         <div class="form-row mb-3">
             <div class="form-group m-1 col-lg-5">
                 <label for="{{ form.complete_inventory.id_for_label }}">
-                    Complete / partial inventory:
+                    Full / partial inventory:
                 </label>
                 {{ form.complete_inventory }}
             </div>

django/cantusdb_project/main_app/templates/source_list.html

@@ -70,7 +70,7 @@ <h3>Browse Sources</h3>
                 </select>
             </div>
             <div class="form-group m-1 col-lg">
-                <label for="fullSourceFilter"><small>Full source/Fragment</small></label>
+                <label for="fullSourceFilter"><small>Complete Source/Fragment</small></label>
                 <select id="fullSourceFilter"  name="fullSource" class="form-control custom-select custom-select-sm">
                     <option value="">- Any -</option>
                     <option value="true">Complete source</option>

django/cantusdb_project/main_app/tests/make_fakes.py

@@ -405,7 +405,7 @@ def make_fake_source(
     summary: Optional[str] = None,
     provenance: Optional[Provenance] = None,
     century: Optional[Century] = None,
-    full_source: Optional[bool] = True,
+    source_completeness: int = Source.SourceCompletenessChoices.FULL_SOURCE,
     indexing_notes: Optional[str] = None,
 ) -> Source:
     """Generates a fake Source object."""
@@ -447,7 +447,7 @@ def make_fake_source(
         summary=summary,
         provenance=provenance,
         # century: ManyToManyField, must be set below
-        full_source=full_source,
+        source_completeness=source_completeness,
         indexing_notes=indexing_notes,
         provenance_notes=faker.sentence(),
         date=faker.sentence(nb_words=3),

django/cantusdb_project/main_app/tests/test_populate_source_completeness.py

@@ -9,15 +9,24 @@ class TestPopulateSourceCompleteness(TestCase):
     def test_populate_source_completeness(self):
         # make a few "Full Source" sources
         for _ in range(5):
-            make_fake_source(full_source=True)
+            source = make_fake_source()
+            source.full_source = True
+            source.save()
         # make a few "Fragment" sources
         for _ in range(3):
-            make_fake_source(full_source=False)
+            source = make_fake_source()
+            source.full_source = False
+            source.save()
+        # make a few "Full Source" sources with full_source=None
+        for _ in range(2):
+            source = make_fake_source()
+            source.full_source = None
+            source.save()
         # run the command
         call_command("populate_source_completeness")
         sources = Source.objects.all()
         for source in sources:
-            if source.full_source:
+            if source.full_source or source.full_source is None:
                 self.assertEqual(
                     source.source_completeness,
                     source.SourceCompletenessChoices.FULL_SOURCE,

django/cantusdb_project/main_app/tests/test_views/test_source.py

@@ -846,44 +846,42 @@ def test_filter_by_century(self):
 
     def test_filter_by_full_source(self):
         full_source = make_fake_source(
-            full_source=True,
+            source_completeness=Source.SourceCompletenessChoices.FULL_SOURCE,
             published=True,
             shelfmark="full source",
         )
         fragment = make_fake_source(
-            full_source=False,
+            source_completeness=Source.SourceCompletenessChoices.FRAGMENT,
             published=True,
             shelfmark="fragment",
         )
-        unknown = make_fake_source(
-            full_source=None,
+        reconstruction = make_fake_source(
+            source_completeness=Source.SourceCompletenessChoices.RECONSTRUCTION,
             published=True,
             shelfmark="full_source field is empty",
         )
 
         # display full sources
         response = self.client.get(reverse("source-list"), {"fullSource": "true"})
         sources = response.context["sources"]
-        # full_source and unknown_source should be in the list, fragment should not
         self.assertIn(full_source, sources)
         self.assertNotIn(fragment, sources)
-        self.assertIn(unknown, sources)
+        self.assertNotIn(reconstruction, sources)
 
         # display fragments
         response = self.client.get(reverse("source-list"), {"fullSource": "false"})
         sources = response.context["sources"]
-        # fragment should be in the list, full_source and unknown_source should not
         self.assertNotIn(full_source, sources)
         self.assertIn(fragment, sources)
-        self.assertNotIn(unknown, sources)
+        self.assertNotIn(reconstruction, sources)
 
         # display all sources
         response = self.client.get(reverse("source-list"))
         sources = response.context["sources"]
         # all three should be in the list
         self.assertIn(full_source, sources)
         self.assertIn(fragment, sources)
-        self.assertIn(unknown, sources)
+        self.assertIn(reconstruction, sources)
 
     def test_search_by_title(self):
         """The "general search" field searches in `title`, `shelfmark`, `description`, and `summary`"""

django/cantusdb_project/main_app/views/source.py

@@ -256,10 +256,13 @@ def get_queryset(self) -> QuerySet[Source]:
             q_obj_filter &= Q(segment__id=int(segment_id))
         if (full_source_str := self.request.GET.get("fullSource")) in ["true", "false"]:
             if full_source_str == "true":
-                full_source_q = Q(full_source=True) | Q(full_source=None)
-                q_obj_filter &= full_source_q
+                q_obj_filter &= Q(
+                    source_completeness=Source.SourceCompletenessChoices.FULL_SOURCE
+                )
             else:
-                q_obj_filter &= Q(full_source=False)
+                q_obj_filter &= Q(
+                    source_completeness=Source.SourceCompletenessChoices.FRAGMENT
+                )
 
         if general_str := self.request.GET.get("general"):
             # Strip spaces at the beginning and end. Then make list of terms split on spaces