from django.core.management.base import BaseCommand, CommandError

from astrobasis.utils import astrobasis_assign_healpix
from astrobasis.models import TwoMASS

class Command(BaseCommand):
    help = 'Initiates healpix for 2MASS'

    def handle(self, *args, **options):
        astrobasis_assign_healpix(TwoMASS)
        self.stdout.write(self.style.SUCCESS('Done'))