srg/artsurvey/management/commands/00_init_energy_bands.py
2024-04-26 12:43:00 +03:00

50 lines
1.1 KiB
Python

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