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

from datetime import date
from heasarc.models import INPUT_DATA_DIR
import datetime
from django.utils import timezone
import astropy
from astropy.io import ascii
import pandas as pd
import pymysql
from sqlalchemy import create_engine

from heasarc.tdat import tDat
from heasarc.models import HeasarcBase, HeasarcTable, TableColumn, HeasarcIntegral9, HeasarcObjectClass
from heasarc.models import NSIDE_SOURCES, ORDER 
from monthplan.models import  NSIDE_PLATES
from astropy_healpix import HEALPix
from astropy.coordinates import SkyCoord  # High-level coordinates
from astropy.coordinates import ICRS, Galactic, FK4, FK5  # Low-level frames
from astropy.coordinates import Angle, Latitude, Longitude  # Angles
import astropy.units as u

class Command(BaseCommand):
    help = 'Initiates data dase'

    def handle(self, *args, **options):
        
        hp_plate = HEALPix(nside=NSIDE_PLATES, order=ORDER, frame=FK5())

        srcs = HeasarcBase.objects.all()
        for src in srcs:
            crd = SkyCoord(src.ra, src.dec, frame=FK5(), unit="deg")
            src.healpix_plate = hp_plate.skycoord_to_healpix(crd)
            src.save()

        self.stdout.write(self.style.SUCCESS('Done'))