Built base things!

Author: Yosef-AlSabbah

api/migrations/0004_trigram_ext.py

@@ -0,0 +1,14 @@
+# Generated by Django 5.2 on 2025-04-13 10:17
+from django.contrib.postgres.operations import TrigramExtension
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0003_rename_picture_url_profile_picture'),
+    ]
+
+    operations = [
+        TrigramExtension(),
+    ]

api/migrations/0005_order_orderitem_alter_category_options_and_more.py

@@ -0,0 +1,110 @@
+# Generated by Django 5.2 on 2025-04-13 13:39
+
+import django.db.models.deletion
+import taggit.managers
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0004_trigram_ext'),
+        ('taggit', '0006_rename_taggeditem_content_type_object_id_taggit_tagg_content_8fc721_idx'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Order',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('quantity', models.IntegerField()),
+                ('order_date', models.DateTimeField(auto_now_add=True)),
+            ],
+            options={
+                'verbose_name': 'Order',
+                'verbose_name_plural': 'Orders',
+                'ordering': ['-order_date'],
+            },
+        ),
+        migrations.CreateModel(
+            name='OrderItem',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('quantity', models.PositiveSmallIntegerField()),
+            ],
+            options={
+                'verbose_name': 'Order Item',
+                'verbose_name_plural': 'Order Items',
+                'ordering': ['order'],
+            },
+        ),
+        migrations.AlterModelOptions(
+            name='category',
+            options={'ordering': ['name'], 'verbose_name': 'Category', 'verbose_name_plural': 'Categories'},
+        ),
+        migrations.AlterModelOptions(
+            name='product',
+            options={'ordering': ['name'], 'verbose_name': 'Product', 'verbose_name_plural': 'Products'},
+        ),
+        migrations.AlterModelOptions(
+            name='profile',
+            options={'ordering': ['user'], 'verbose_name': 'Profile', 'verbose_name_plural': 'Profiles'},
+        ),
+        migrations.AddField(
+            model_name='product',
+            name='tags',
+            field=taggit.managers.TaggableManager(help_text='A comma-separated list of tags.', through='taggit.TaggedItem', to='taggit.Tag', verbose_name='Tags'),
+        ),
+        migrations.AddIndex(
+            model_name='category',
+            index=models.Index(fields=['name'], name='api_categor_name_53a3ad_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='product',
+            index=models.Index(fields=['slug'], name='api_product_slug_e70af7_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='product',
+            index=models.Index(fields=['name'], name='api_product_name_73c704_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='product',
+            index=models.Index(fields=['category'], name='api_product_categor_5c53c5_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='profile',
+            index=models.Index(fields=['user'], name='api_profile_user_id_bc09a5_idx'),
+        ),
+        migrations.AddField(
+            model_name='order',
+            name='user',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL),
+        ),
+        migrations.AddField(
+            model_name='orderitem',
+            name='order',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='items', to='api.order'),
+        ),
+        migrations.AddField(
+            model_name='orderitem',
+            name='product',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.product'),
+        ),
+        migrations.AddIndex(
+            model_name='order',
+            index=models.Index(fields=['user'], name='api_order_user_id_764746_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='order',
+            index=models.Index(fields=['order_date'], name='api_order_order_d_f254aa_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='orderitem',
+            index=models.Index(fields=['order'], name='api_orderit_order_i_3f9c70_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='orderitem',
+            index=models.Index(fields=['product'], name='api_orderit_product_1a13b6_idx'),
+        ),
+    ]

api/migrations/0006_order_status.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.2 on 2025-04-13 13:46
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0005_order_orderitem_alter_category_options_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='order',
+            name='status',
+            field=models.CharField(choices=[('PE', 'Pending'), ('CO', 'Completed'), ('CA', 'Cancelled')], default='PE', max_length=10),
+        ),
+    ]

api/recommender.py

@@ -0,0 +1,75 @@
+import redis
+from django.conf import settings
+
+from api.models import Product
+
+# connect to redis
+r = redis.Redis(
+    host=settings.REDIS_HOST,
+    port=settings.REDIS_PORT,
+    db=settings.REDIS_DB
+)
+
+
+class Recommender:
+    def get_product_key(self, id):
+        return f'product:{id}:purchased_with'
+
+    def products_bought(self, products):
+        product_ids = [p.id for p in products]
+        for product_id in product_ids:
+            for with_id in product_ids:
+                # get the other products bought with each product
+                if product_id != with_id:
+                    # increment score for product purchased together
+                    r.zincrby(
+                        self.get_product_key(product_id), 1, with_id
+                    )
+
+    def suggest_products_for(self, products, max_results=6):
+        product_ids = [p.id for p in products]
+        if len(products) == 1:
+            # only 1 product
+            suggestions = r.zrange(
+                self.get_product_key(product_ids[0]), 0, -1, desc=True
+            )[:max_results]
+        else:
+            # generate a temporary key
+            flat_ids = ''.join([str(id) for id in product_ids])
+            tmp_key = f'tmp_{flat_ids}'
+            # multiple products, combine scores of all products
+            # store the resulting sorted set in a temporary key
+            keys = [self.get_product_key(id) for id in product_ids]
+            r.zunionstore(tmp_key, keys)
+            # remove ids for the products the recommendation is for
+            r.zrem(tmp_key, *product_ids)
+            # get the product ids by their score, descendant sort
+            suggestions = r.zrange(
+                tmp_key, 0, -1, desc=True
+            )[:max_results]
+            # remove the temporary key
+            r.delete(tmp_key)
+        suggested_products_ids = [int(id) for id in suggestions]
+        # get suggested products and sort by order of appearance
+        suggested_products = list(
+            Product.objects.filter(id__in=suggested_products_ids)
+        )
+        suggested_products.sort(
+            key=lambda x: suggested_products_ids.index(x.id)
+        )
+        return suggested_products
+    #
+    # def suggest_for_user(self, user, max_results=100):
+    #     """
+    #     Suggest products for the logged-in user based on their previous purchases.
+    #     """
+    #     # Get the products purchased by the user
+    #     purchased_products = Product.objects.filter(
+    #         order__user=user
+    #     ).distinct()  # Assuming an `Order` model exists with a `user` field
+    #
+    #     if not purchased_products.exists():
+    #         return []  # Return an empty list if no purchases are found
+    #
+    #     # Use the existing recommendation logic
+    #     return self.suggest_products_for(purchased_products, max_results=max_results)