Merge pull request #87 from pirogramming/feature/kje

Author: river040118

moodico/main/admin.py

@@ -1,3 +1,7 @@
 from django.contrib import admin
+from .models import RankedProduct, VotingSession
 
 # Register your models here.
+
+admin.site.register(RankedProduct)
+admin.site.register(VotingSession)

moodico/main/management/__init__.py

@@ -0,0 +1,60 @@
+from django.core.management.base import BaseCommand
+from django.utils import timezone
+from django.db import transaction
+from moodico.main.models import RankedProduct, VotingSession
+from moodico.products.utils.common_utils import get_top_liked_products
+
+# 매일 자정, 현재의 랭킹 최상위 두 개 제품으로 투표 세션을 생성하는 함수
+class Command(BaseCommand):
+
+    def handle(self, *args, **options):
+        self.stdout.write(f"[{timezone.now()}] 투표 세션 생성 시작")
+
+        active_sessions = VotingSession.objects.filter(is_active=True)
+        if active_sessions.exists():
+            for session in active_sessions:
+                session.close_session()
+            self.stdout.write(self.style.WARNING(f"{len(active_sessions)}개의 기존 활성 세션을 종료했습니다."))
+        
+        try:
+            ranked1_data, ranked2_data = get_top_liked_products(2)
+        except (ValueError, IndexError):
+            self.stdout.write(self.style.ERROR("상위 2개 제품을 가져오는 데 실패했습니다. 작업을 중단합니다."))
+            return
+
+        # 가져온 데이터로 Product 객체 생성 및 업데이트
+        product1, created1 = RankedProduct.objects.update_or_create(
+            product_id=ranked1_data["product_id"],
+            defaults={
+                'name': ranked1_data["product_name"],
+                'brand': ranked1_data["product_brand"],
+                'price': ranked1_data["product_price"],
+                'image_url': ranked1_data["product_image"],
+                'like_count': ranked1_data["like_count"]
+            }
+        )
+        if created1:
+            self.stdout.write(f"새로운 제품 '{product1.name}'이 랭킹 1위로 투표 대상 객체가 생성")
+        
+        product2, created2 = RankedProduct.objects.update_or_create(
+            product_id=ranked2_data["product_id"],
+            defaults={
+                'name': ranked2_data["product_name"],
+                'brand': ranked2_data["product_brand"],
+                'price': ranked2_data["product_price"],
+                'image_url': ranked2_data["product_image"],
+                'like_count': ranked2_data["like_count"]
+            }
+        )
+        if created2:
+            self.stdout.write(f"새로운 제품 '{product2.name}'이 랭킹 2위로 투표 대상 객체가 생성")
+
+        # 투표 세션 생성
+        new_session = VotingSession.objects.create(
+            product1=product1,
+            product2=product2
+        )
+
+        self.stdout.write(self.style.SUCCESS(
+            f"새로운 투표 세션(ID: {new_session.id})을 생성했습니다: '{product1.name}' vs '{product2.name}'"
+        ))

moodico/main/management/commands/__init__.py

@@ -0,0 +1,42 @@
+# Generated by Django 5.2.4 on 2025-08-17 12:07
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='RankedProduct',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('product_id', models.CharField(db_index=True, max_length=100, unique=True)),
+                ('name', models.CharField(max_length=200)),
+                ('brand', models.CharField(max_length=100)),
+                ('price', models.PositiveIntegerField(default=0)),
+                ('image_url', models.CharField(max_length=500)),
+                ('like_count', models.PositiveIntegerField(default=0)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='VotingSession',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('product1_votes', models.PositiveIntegerField(default=0)),
+                ('product2_votes', models.PositiveIntegerField(default=0)),
+                ('is_active', models.BooleanField(default=True)),
+                ('start_time', models.DateTimeField(auto_now_add=True)),
+                ('end_time', models.DateTimeField(blank=True, null=True)),
+                ('product1', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='product1_sessions', to='main.rankedproduct')),
+                ('product2', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='product2_sessions', to='main.rankedproduct')),
+                ('winner', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='won_sessions', to='main.rankedproduct')),
+            ],
+        ),
+    ]

moodico/main/management/commands/create_daily_vote_session.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.2.4 on 2025-08-17 15:28
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('main', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='rankedproduct',
+            name='price',
+            field=models.CharField(max_length=100, verbose_name='가격'),
+        ),
+    ]

moodico/main/migrations/0001_initial.py

@@ -0,0 +1,29 @@
+# Generated by Django 5.2.4 on 2025-08-17 18:18
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('main', '0002_alter_rankedproduct_price'),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Vote',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('session', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='main.votingsession')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+                ('voted_product', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='main.rankedproduct')),
+            ],
+            options={
+                'unique_together': {('session', 'user')},
+            },
+        ),
+    ]

moodico/main/migrations/0002_alter_rankedproduct_price.py

@@ -1,3 +1,59 @@
 from django.db import models
+from django.utils import timezone
+from django.conf import settings
 
 # Create your models here.
+class RankedProduct(models.Model):
+    product_id = models.CharField(max_length=100, unique=True, db_index=True) # 제품 id
+    name = models.CharField(max_length=200)
+    brand = models.CharField(max_length=100)
+    price = models.CharField(max_length=100, verbose_name="가격")
+    image_url = models.CharField(max_length=500)
+    like_count = models.PositiveIntegerField(default=0)
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f"[{self.brand}] {self.name}"
+
+class VotingSession(models.Model):
+    """
+    하나의 투표 세션 - 24시간 간격으로 생성
+    """
+    #session_id = 
+    product1 = models.ForeignKey(RankedProduct, on_delete=models.CASCADE, related_name='product1_sessions')
+    product2 = models.ForeignKey(RankedProduct, on_delete=models.CASCADE, related_name='product2_sessions')
+    product1_votes = models.PositiveIntegerField(default=0)
+    product2_votes = models.PositiveIntegerField(default=0)
+    is_active = models.BooleanField(default=True)
+
+    winner = models.ForeignKey(RankedProduct, on_delete=models.SET_NULL, null=True, blank=True, related_name='won_sessions')
+    
+    start_time = models.DateTimeField(auto_now_add=True)
+    end_time = models.DateTimeField(null=True, blank=True)
+
+    def __str__(self):
+        status = "Active" if self.is_active else "Finished"
+        return f"Session {self.id}: {self.product1.name} vs {self.product2.name} ({status})"
+    
+    # session is_active false 변경 함수
+    def close_session(self):
+        self.end_time = timezone.now()
+        self.is_active = False
+
+        if self.product1_votes > self.product2_votes:
+            self.winner = self.product1
+        elif self.product2_votes > self.product1_votes:
+            self.winner = self.product2
+        
+        self.save()
+
+# 세션 - 투표 - 사용자를 잇는 투표 모델
+class Vote(models.Model):
+    session = models.ForeignKey(VotingSession, on_delete=models.CASCADE)
+    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
+    voted_product = models.ForeignKey(RankedProduct, on_delete=models.CASCADE)
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    class Meta:
+        # 한 사용자는 한 세션에 하나의 투표만 가능
+        unique_together = ('session', 'user')

moodico/main/migrations/0003_vote.py

@@ -4,4 +4,5 @@
 urlpatterns = [
     path('', views.main, name='main'), #메인 페이지
     path('personalcolor/', views.personalcolor, name='personalcolor'), #퍼스널 컬러 페이지
+    path('voting_api', views.voting_api, name='voting_api'), # 투표 기능 api
 ]