hackspace/mhackspace/base/management/commands/generate_test_data.py

27 lines
711 B
Python

from autofixture import AutoFixture
from django.core.management.base import BaseCommand
from mhackspace.base.models import BannerImages
from mhackspace.feeds.models import Article, Feed
from mhackspace.users.models import User
class Command(BaseCommand):
help = 'Imports the RSS feeds from active blogs'
def handle(self, *args, **options):
users = AutoFixture(User)
users.create(10)
feed = AutoFixture(Feed)
feed.create(10)
feeds = AutoFixture(Article)
feeds.create(10)
banners = AutoFixture(BannerImages)
banners.create(10)
self.stdout.write(
self.style.SUCCESS(
'Finished creating test data'))