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 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 heasarc.models import HeasarcBase
from artsurvey.models import ArtSurveySource, ArtSurvey
from artsurvey.utils import setup_metasource
from srglib.utils import load_vizier_nvss
from srglib.utils import load_vizier_first
from srglib.utils import load_vizier_sumss
from astrobasis.models import GAIADR3
from srglib.utils import find_counterparts
from srglib.utils import load_vizier_allwise

def do_all(survey_version):
    minrad=40
    maxdist=120
        
    try:
        survey = ArtSurvey.objects.get(version=survey_version)
        print("use ArtSurvey {}".format(survey))
    except ArtSurvey.DoesNotExist:
        print("ArtSurvey {} not found".format(survey_version))
        return
        
    srcs = survey.artsurveysource_set.all()
    print("Selected {} sources".format(srcs.count()))
    for src in srcs:
        if(src.skymapsource_set.all().count()):
            orig=src.skymapsource_set.all()[0]
            """
            src.nfalse=orig.nfalse
            src.save()
            print("{} <-- {}".format(src.nfalse,orig.nfalse))
            """
            if not orig.ext_id == '':
                #src.ext_id=orig.ext_id
                #src.save()
                print("{} {} <-- {}".format(src,src.ext_id,orig.ext_id))


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):

        """
        setup_metasource('J041324.8+102752','J041326.3+102725')
        setup_metasource('J151003.9+333057','J151014.3+332942')
        setup_metasource('J210336.2+454507','J210331.8+454453')
        setup_metasource('J142557.4+374954','J142604.6+374930')
        setup_metasource('J171226.3-232138','J171223.5-232048')

        setup_metasource('J151350.7-590814','J151355.6-590806')
        setup_metasource('J151350.7-590814','J151355.7-590806')
        setup_metasource('J151350.7-590814','J151355.6-590802')
        setup_metasource('J151350.7-590814','J151354.2-590805')
        setup_metasource('J151350.7-590814','J151354.1-590805')
        setup_metasource('J151350.7-590814','J151354.3-590805')
        setup_metasource('J151350.7-590814','J151355.4-590802')
        setup_metasource('J151350.7-590814','J151354.5-590806')
        # "Empty" MetaSource SRGA J151354.5-590806 has been deleted by hand
        """

        """
        setup_metasource('J151054.2+054422','J151054.9+054357')
        setup_metasource('J151054.2+054422','J151053.6+054345')
        setup_metasource('J151054.2+054422','J151053.6+054344')
        """
        setup_metasource('J210336.2+454507','J210331.8+454453')
        setup_metasource('J151054.2+054422','J151054.9+054357')
        setup_metasource('J151054.2+054422','J151053.6+054344')
        setup_metasource('J151054.2+054422','J151053.6+054345')
        setup_metasource('J142557.4+374954','J142604.6+374930')

        setup_metasource('J121033.1+392421','J121030.2+392422')
        setup_metasource('J123855.1-271830','J123823.9-253148')
        setup_metasource('J174155.6-121207','J174154.0-121231')
        setup_metasource('J184817.8-031020','J184816.1-030956')
        setup_metasource('J092234.6-631738','J092239.7-631738')
        #setup_metasource('','')

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