from django.core.management.base import BaseCommand
from api.models import Conversation

class Command(BaseCommand):
    help = 'Create N test conversations'

    def add_arguments(self, parser):
        parser.add_argument('--number', type=int, help='Number of conversations to create')

    def handle(self, *args, **kwargs):
        number = kwargs.get('number') or 1
        
        for i in range(number):
            conv = Conversation.objects.create(title="test-conversation")
            self.stdout.write(self.style.SUCCESS(f'Successfully created test conversation (id : {conv.id})'))