feat(locations): New Location.user_count field (#471)

open_prices/locations/admin.py

@@ -12,6 +12,7 @@ class LocationAdmin(admin.ModelAdmin):
         "osm_tag_key",
         "osm_tag_value",
         "price_count",
+        "user_count",
         "product_count",
         "proof_count",
         "created",

open_prices/locations/factories.py

@@ -19,5 +19,6 @@ class Meta:
     osm_name = factory.Faker("name")
     osm_address_country = factory.Faker("country")
     # price_count = factory.LazyAttribute(lambda x: random.randrange(0, 100))
+    # user_count = factory.LazyAttribute(lambda x: random.randrange(0, 100))
     # product_count = factory.LazyAttribute(lambda x: random.randrange(0, 100))
     # proof_count = factory.LazyAttribute(lambda x: random.randrange(0, 100))

open_prices/locations/migrations/0004_location_user_count.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.1 on 2024-09-27 15:42
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("locations", "0003_location_product_count"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="location",
+            name="user_count",
+            field=models.PositiveIntegerField(blank=True, default=0, null=True),
+        ),
+    ]

open_prices/locations/models.py

@@ -45,6 +45,7 @@ class Location(models.Model):
     )
 
     price_count = models.PositiveIntegerField(default=0, blank=True, null=True)
+    user_count = models.PositiveIntegerField(default=0, blank=True, null=True)
     product_count = models.PositiveIntegerField(default=0, blank=True, null=True)
     proof_count = models.PositiveIntegerField(default=0, blank=True, null=True)
 
@@ -103,6 +104,17 @@ def update_price_count(self):
         self.price_count = self.prices.count()
         self.save(update_fields=["price_count"])
 
+    def update_user_count(self):
+        from open_prices.prices.models import Price
+
+        self.user_count = (
+            Price.objects.filter(location=self)
+            .values_list("owner", flat=True)
+            .distinct()
+            .count()
+        )
+        self.save(update_fields=["user_count"])
+
     def update_product_count(self):
         from open_prices.prices.models import Price
 

open_prices/locations/tests.py

@@ -94,6 +94,7 @@ class LocationPropertyTest(TestCase):
     @classmethod
     def setUpTestData(cls):
         cls.user = UserFactory()
+        cls.user_2 = UserFactory()
         cls.location = LocationFactory(**LOCATION_NODE_652825274)
         cls.proof = ProofFactory(
             location_osm_id=cls.location.osm_id,
@@ -113,6 +114,7 @@ def setUpTestData(cls):
             location_osm_id=cls.location.osm_id,
             location_osm_type=cls.location.osm_type,
             price=2.0,
+            owner=cls.user_2.user_id,
         )
 
     def test_update_price_count(self):
@@ -125,6 +127,13 @@ def test_update_price_count(self):
         self.location.update_price_count()
         self.assertEqual(self.location.price_count, 0)
 
+    def test_update_user_count(self):
+        self.location.refresh_from_db()
+        self.assertEqual(self.location.user_count, 0)
+        # update_user_count() should fix user_count
+        self.location.update_user_count()
+        self.assertEqual(self.location.user_count, 2)
+
     def test_update_product_count(self):
         self.location.refresh_from_db()
         self.assertEqual(self.location.product_count, 0)