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 HeasarcTable, TableColumn, HeasarcXrayMaster, HeasarcObjectClass


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):
        
        class_count=9999
        try:
            object_class = HeasarcObjectClass.objects.get(class_id=class_count)
        except:
            print("HeasarcObjectClass not found for class_id=", class_count)

        print(object_class.class_name)

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