Move PriceTagStatus to proof constants

open_prices/api/proofs/tests.py

@@ -6,7 +6,6 @@
 from django.urls import reverse
 from PIL import Image
 
-from open_prices.common.constants import PriceTagStatus
 from open_prices.locations import constants as location_constants
 from open_prices.locations.factories import LocationFactory
 from open_prices.prices.factories import PriceFactory
@@ -406,11 +405,11 @@ def setUpTestData(cls):
         cls.price_tag_1 = PriceTagFactory(
             proof=cls.proof,
             price=cls.price,
-            status=PriceTagStatus.linked_to_price.value,
+            status=proof_constants.PriceTagStatus.linked_to_price.value,
         )
         cls.price_tag_2 = PriceTagFactory(proof=cls.proof)
         cls.price_tag_3 = PriceTagFactory(
-            proof=cls.proof_2, status=PriceTagStatus.deleted
+            proof=cls.proof_2, status=proof_constants.PriceTagStatus.deleted
         )
 
     def test_price_tag_list(self):
@@ -556,7 +555,10 @@ def test_price_tag_create_with_price(self):
         self.assertEqual(response.data["created_by"], self.user_session.user.user_id)
         self.assertEqual(response.data["updated_by"], self.user_session.user.user_id)
         self.assertEqual(response.data["price_id"], self.price.id)
-        self.assertEqual(response.data["status"], PriceTagStatus.linked_to_price.value)
+        self.assertEqual(
+            response.data["status"],
+            proof_constants.PriceTagStatus.linked_to_price.value,
+        )
 
 
 class PriceTagUpdateApiTest(TestCase):
@@ -616,7 +618,10 @@ def test_price_tag_set_price_id(self):
         # Price ID was set to the new value
         self.assertEqual(response.data["price_id"], self.price.id)
         # Status was automatically set to linked_to_price
-        self.assertEqual(response.data["status"], PriceTagStatus.linked_to_price.value)
+        self.assertEqual(
+            response.data["status"],
+            proof_constants.PriceTagStatus.linked_to_price.value,
+        )
 
     def test_price_tag_set_invalid_price_id(self):
         self.assertEqual(self.price_tag.price_id, None)
@@ -638,11 +643,13 @@ def test_price_tag_set_status(self):
             self.url,
             content_type="application/json",
             # Price associated with another proof
-            data={"status": PriceTagStatus.not_readable.value},
+            data={"status": proof_constants.PriceTagStatus.not_readable.value},
             headers={"Authorization": f"Bearer {self.user_session.token}"},
         )
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(response.data["status"], PriceTagStatus.not_readable.value)
+        self.assertEqual(
+            response.data["status"], proof_constants.PriceTagStatus.not_readable.value
+        )
 
     def test_price_tag_invalid_status(self):
         self.assertEqual(self.price_tag.status, None)
@@ -721,5 +728,6 @@ def test_price_tag_delete(self):
         self.assertEqual(response.data, None)
         self.assertEqual(PriceTag.objects.filter(id=self.price_tag.id).count(), 1)
         self.assertEqual(
-            PriceTag.objects.get(id=self.price_tag.id).status, PriceTagStatus.deleted
+            PriceTag.objects.get(id=self.price_tag.id).status,
+            proof_constants.PriceTagStatus.deleted,
         )

open_prices/api/proofs/views.py

@@ -24,7 +24,7 @@
 )
 from open_prices.api.utils import get_source_from_request
 from open_prices.common.authentication import CustomAuthentication
-from open_prices.common.constants import PriceTagStatus
+from open_prices.proofs import constants as proof_constants
 from open_prices.proofs.ml import extract_from_price_tag
 from open_prices.proofs.models import PriceTag, Proof
 from open_prices.proofs.utils import store_file
@@ -175,7 +175,7 @@ def destroy(self, request: Request, *args, **kwargs) -> Response:
                 {"detail": "Cannot delete price tag with associated prices."},
                 status=status.HTTP_403_FORBIDDEN,
             )
-        price_tag.status = PriceTagStatus.deleted
+        price_tag.status = proof_constants.PriceTagStatus.deleted
         price_tag.save()
         return Response(status=status.HTTP_204_NO_CONTENT)
 

open_prices/common/constants.py

@@ -1,16 +1,4 @@
-import enum
-
 from babel.numbers import list_currencies
 
 CURRENCY_LIST = sorted(list_currencies())
 CURRENCY_CHOICES = [(key, key) for key in CURRENCY_LIST]
-
-
-class PriceTagStatus(enum.IntEnum):
-    deleted = 0
-    linked_to_price = 1
-    not_readable = 2
-    not_price_tag = 3
-
-
-PRICE_TAG_STATUS_CHOICES = [(item.value, item.name) for item in PriceTagStatus]

open_prices/proofs/constants.py

@@ -1,3 +1,5 @@
+import enum
+
 TYPE_PRICE_TAG = "PRICE_TAG"
 TYPE_RECEIPT = "RECEIPT"
 TYPE_GDPR_REQUEST = "GDPR_REQUEST"
@@ -31,3 +33,13 @@
 PRICE_TAG_PREDICTION_TYPE_CHOICES = [
     (PRICE_TAG_EXTRACTION_TYPE, PRICE_TAG_EXTRACTION_TYPE)
 ]
+
+
+class PriceTagStatus(enum.IntEnum):
+    deleted = 0
+    linked_to_price = 1
+    not_readable = 2
+    not_price_tag = 3
+
+
+PRICE_TAG_STATUS_CHOICES = [(item.value, item.name) for item in PriceTagStatus]

open_prices/proofs/models.py

@@ -409,7 +409,7 @@ def status_unknown(self):
         return self.filter(status=None)
 
     def status_linked_to_price(self):
-        return self.filter(status=constants.PriceTagStatus.linked_to_price.value)
+        return self.filter(status=proof_constants.PriceTagStatus.linked_to_price.value)
 
 
 class PriceTag(models.Model):
@@ -437,7 +437,7 @@ class PriceTag(models.Model):
         help_text="Coordinates of the bounding box, in the format [y_min, x_min, y_max, x_max]",
     )
     status = models.IntegerField(
-        choices=constants.PRICE_TAG_STATUS_CHOICES,
+        choices=proof_constants.PRICE_TAG_STATUS_CHOICES,
         null=True,
         blank=True,
         help_text="The annotation status",
@@ -533,8 +533,8 @@ def clean(self, *args, **kwargs):
                 )
 
             if self.status is None:
-                self.status = constants.PriceTagStatus.linked_to_price.value
-            elif self.status != constants.PriceTagStatus.linked_to_price.value:
+                self.status = proof_constants.PriceTagStatus.linked_to_price.value
+            elif self.status != proof_constants.PriceTagStatus.linked_to_price.value:
                 utils.add_validation_error(
                     validation_errors,
                     "status",

open_prices/stats/models.py

@@ -2,7 +2,7 @@
 from django.utils import timezone
 from solo.models import SingletonModel
 
-from open_prices.common.constants import PriceTagStatus
+from open_prices.proofs import constants as proof_constants
 
 
 class TotalStats(SingletonModel):
@@ -119,7 +119,7 @@ def update_price_tag_stats(self):
             status=None
         ).count()
         self.price_tag_status_linked_to_price_count = PriceTag.objects.filter(
-            status=PriceTagStatus.linked_to_price.value
+            status=proof_constants.PriceTagStatus.linked_to_price.value
         ).count()
         self.save(update_fields=self.PRICE_TAG_COUNT_FIELDS + ["updated"])
 

open_prices/stats/tests.py

@@ -1,7 +1,6 @@
 from django.db import IntegrityError
 from django.test import TestCase
 
-from open_prices.common.constants import PriceTagStatus
 from open_prices.locations.factories import LocationFactory
 from open_prices.prices import constants as price_constants
 from open_prices.prices.factories import PriceFactory
@@ -77,7 +76,7 @@ def setUpTestData(cls):
         PriceTagFactory(
             proof=cls.proof_price_tag,
             price=cls.price,
-            status=PriceTagStatus.linked_to_price.value,
+            status=proof_constants.PriceTagStatus.linked_to_price.value,
         )
 
     def test_update_price_stats(self):