You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and dots ('.'), can be up to 35 characters long. Letters must be lowercase.
15 lines
595 B
15 lines
595 B
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})')) |