70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
import random
|
|
from autofixture import AutoFixture
|
|
from autofixture.generators import ImageGenerator, IntegerGenerator, ChoicesGenerator
|
|
from django.core.management.base import BaseCommand
|
|
from django.core.management import call_command
|
|
from mhackspace.base.models import BannerImage
|
|
from mhackspace.feeds.models import Article, Feed
|
|
from mhackspace.users.models import User, Rfid
|
|
from mhackspace.blog.models import Category, Post
|
|
from mhackspace.rfid.models import Device, DeviceAuth
|
|
|
|
|
|
class ImageFixture(AutoFixture):
|
|
class Values:
|
|
scaled_image = ImageGenerator(width=800, height=300, sizes=((1280, 300),))
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Build test data for development environment'
|
|
|
|
def handle(self, *args, **options):
|
|
feeds = AutoFixture(Article)
|
|
feeds.create(10)
|
|
feed = AutoFixture(Feed)
|
|
feed.create(10)
|
|
|
|
post = AutoFixture(Post)
|
|
post.create(10)
|
|
|
|
categorys = AutoFixture(Category)
|
|
categorys.create(10)
|
|
|
|
# load known data
|
|
call_command('loaddata', 'mhackspace/users/fixtures/groups.json', verbose=0)
|
|
|
|
# AutoFixture.autodiscover()
|
|
|
|
# random data
|
|
users = AutoFixture(User, field_values={
|
|
'title': ChoicesGenerator(('Mr', 'Mrs', 'Emperor', 'Captain'))
|
|
})
|
|
users.create(10)
|
|
|
|
Rfid.objects.all().delete()
|
|
Device.objects.all().delete()
|
|
DeviceAuth.objects.all().delete()
|
|
rfid = AutoFixture(Rfid, field_values={'code': IntegerGenerator(1, 100000)})
|
|
rfid.create(20)
|
|
|
|
device = AutoFixture(Device, field_values={
|
|
'name': ChoicesGenerator(('Door', 'Printer', 'Laser Cutter', ''))
|
|
})
|
|
device.create(5)
|
|
|
|
deviceauth = AutoFixture(DeviceAuth)
|
|
deviceauth.create(5)
|
|
|
|
feed = AutoFixture(Feed)
|
|
feed.create(10)
|
|
|
|
feeds = AutoFixture(Article)
|
|
feeds.create(10)
|
|
|
|
banners = ImageFixture(BannerImage)
|
|
banners.create(10)
|
|
self.stdout.write(
|
|
self.style.SUCCESS(
|
|
'Finished creating test data'))
|
|
|