From 26d5730ad2275f70e92b38c572c2a5d43f7ec208 Mon Sep 17 00:00:00 2001
From: Trenton H <797416+stumpylog@users.noreply.github.com>
Date: Mon, 24 Jul 2023 10:19:02 -0700
Subject: [PATCH] Handles a special case where the classification model exists,
 but no items are auto matching any longer

---
 src/documents/tasks.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/documents/tasks.py b/src/documents/tasks.py
index f1b65c45f..2dbc9d6eb 100644
--- a/src/documents/tasks.py
+++ b/src/documents/tasks.py
@@ -65,6 +65,12 @@ def train_classifier():
         and not Correspondent.objects.filter(matching_algorithm=Tag.MATCH_AUTO).exists()
         and not StoragePath.objects.filter(matching_algorithm=Tag.MATCH_AUTO).exists()
     ):
+        logger.info("No automatic matching items, not training")
+        # Special case, items were once auto and trained, so remove the model
+        # and prevent its use again
+        if settings.MODEL_FILE.exists():
+            logger.info(f"Removing {settings.MODEL_FILE} so it won't be used")
+            settings.MODEL_FILE.unlink()
         return
 
     classifier = load_classifier()