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.models import HeasarcBase
from artsurvey.models import EnergyBand, ArtSurveySource, ArtSurvey
from srglib.utils import find_counterparts
from srgcat.models import SrgDataDump

"""
0   4.0  12.0
1   4.0   7.0
2   7.0  12.0
3  12.0  20.0
4  20.0  30.0
"""

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

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

        bands = EnergyBand.objects.all()
        bands.delete()

        e0=EnergyBand(e1=4.0, e2=12.0, slug='E0')
        e0.save()

        e1=EnergyBand(e1=4.0, e2=7.0, slug='E1')
        e1.save()

        e2=EnergyBand(e1=7.0, e2=12.0, slug='E2')
        e2.save()

        e3=EnergyBand(e1=12.0, e2=20.0, slug='E3')
        e3.save()

        e4=EnergyBand(e1=20.0, e2=30.0, slug='E4')
        e4.save()


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