56 lines
1.8 KiB
Python
56 lines
1.8 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.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'))
|