adding the division pattern model

Author: bishwaspraveen

sde_collections/models/pattern.py

@@ -10,7 +10,7 @@
     parse_title,
     resolve_title,
 )
-from .collection_choice_fields import DocumentTypes
+from .collection_choice_fields import Divisions, DocumentTypes
 
 
 class BaseMatchPattern(models.Model):
@@ -250,6 +250,29 @@ class Meta:
         unique_together = ("collection", "match_pattern")
 
 
+class DivisionPattern(BaseMatchPattern):
+    division = models.IntegerField(choices=Divisions.choices)
+
+    def apply(self) -> None:
+        matched_urls = self.matched_urls()
+        matched_urls.update(division=self.division)
+        candidate_url_ids = list(matched_urls.values_list("id", flat=True))
+        self.candidate_urls.through.objects.bulk_create(
+            objs=[
+                DivisionPattern.candidate_urls.through(candidateurl_id=candidate_url_id, divisionpattern_id=self.id)
+                for candidate_url_id in candidate_url_ids
+            ]
+        )
+
+    def unapply(self) -> None:
+        self.candidate_urls.update(division=None)
+
+    class Meta:
+        verbose_name = "Division Pattern"
+        verbose_name_plural = "Division Patterns"
+        unique_together = ("collection", "match_pattern")
+
+
 # @receiver(post_save, sender=TitlePattern)
 # def post_save_handler(sender, instance, created, **kwargs):
 #     if created: