Misago/misago/management/commands/syncfixtures.py

from optparse import make_option
import os.path
import pkgutil
from django.core.management.base import BaseCommand
from misago.models import Fixture
from misago.utils.fixtures import load_fixture, update_fixture
import misago.fixtures

class Command(BaseCommand):
    """
    Loads Misago fixtures
    """
    help = 'Load Misago fixtures'
    option_list = BaseCommand.option_list + (
        make_option('--quiet',
            action='store_true',
            dest='quiet',
            default=False,
            help='Dont display output from this message'),
        )
    
    def handle(self, *args, **options):
        if not options['quiet']:
            self.stdout.write('\nLoading data from fixtures...')
            
        fixture_data = {}
        for fixture in Fixture.objects.all():
            fixture_data[fixture.name] = fixture

        loaded = 0
        updated = 0
        
        fixtures_path = os.path.dirname(misago.fixtures.__file__)
        for _, name, _ in pkgutil.iter_modules([fixtures_path]):
            if name in fixture_data:
                if update_fixture('misago.fixtures.' + name):
                    updated += 1
                    if not options['quiet']:
                        self.stdout.write('Updating "%s" fixture...' % name)
            else:
                if load_fixture('misago.fixtures.' + name):
                    loaded += 1
                    Fixture.objects.create(name=name)
                    if not options['quiet']:
                        self.stdout.write('Loading "%s" fixture...' % name)

        if not options['quiet']:
            self.stdout.write('\nLoaded %s fixtures and updated %s fixtures.\n' % (loaded, updated))