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

from datetime import date
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 HeasarcTable, TableColumn, HeasarcObjectClass, NSIDE_SOURCES, ORDER, HeasarcXrayMaster, HeasarcXMMSSC, HeasarcBase
from srgcat.models import ArtCat, ArtSource, SkyMaps, SkyMapSource, SrgDataDump
from astropy.table import Table
from astropy_healpix import HEALPix, neighbours
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

from srglib.utils import find_heasarc_counterparts
from srglib.utils import load_simbad_for_skymap_sources

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

#    def add_arguments(self, parser):
#        parser.add_argument('poll_id', nargs='+', type=int)

    def handle(self, *args, **options):

        """
        cats = ArtCat.objects.all()
        for cat in cats:
            cat.source_set = cat.artsource_set
            find_heasarc_counterparts(cat)
        
        
        skymaps = SkyMaps.objects.all()
        for cat in skymaps:
            ntot = cat.skymapsource_set.count()
            if (ntot):
                cat.source_set = cat.skymapsource_set
                find_heasarc_counterparts(cat)
        """
        try: 
            dump = SrgDataDump.objects.get(name='srg_20200530_231807_000')
        except:
            return
        srcs = dump.skymapsource_set.all()
        load_simbad_for_skymap_sources(srcs,minrad=30,maxdist=120)
        

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