more apps

artsurvey/management/commands/00_artsurvey_init.py

@@ -0,0 +1,105 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.db.models import Q
+
+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 ArtSurveySource, ArtSurvey, EnergyBand
+from srglib.utils import find_counterparts
+from srgcat.models import SrgDataDump
+
+def init_survey(init_version,survey_name,dump_name,slug):
+    name=dump_name+'.'+slug
+    try:
+        survey = ArtSurvey.objects.get(Q(version=init_version) & Q(name=survey_name))
+    except ArtSurvey.DoesNotExist:
+        print("create ArtSurvey")
+        survey = ArtSurvey(version=init_version, name=survey_name)
+        survey.save()
+        
+    print("use ArtSurvey {}".format(survey))
+
+    try:
+        band = EnergyBand.objects.get(slug=slug)
+        print("use EnergyBand {}".format(band))
+    except EnergyBand.DoesNotExist:
+        print("EnergyBand not found")
+        return
+
+    
+
+    """ copy catalog from this dump """
+    dump = SrgDataDump.objects.get(name=name)
+    print(dump)
+    
+    srcs = dump.skymapsource_set.all()
+    for src in srcs:
+        if not (src.artsurvey_source):
+            print("{} {}".format(slug,src))
+            artsource = ArtSurveySource.objects.create_source(src,survey,band)
+            artsource.save()
+
+
+class Command(BaseCommand):
+    help = 'Initiates data dase'
+
+    def handle(self, *args, **options):
+        
+        """
+        bands = ['E0','E1','E2','E3','E4']
+        for slug in bands:
+            init_survey(12.5,'S12',slug)
+        """
+
+        #init_survey(123.0,'S123','E0')
+        #init_survey(1234.0,'1-4','S1-4','E0')
+        #init_survey(3.0,'3','S3','E0')
+        
+
+        #version=0.1
+        #init_survey(version,'L20','L20','E0')
+
+        
+        #init_survey(0.13,'L20rb','L20rb','E0')
+        #init_survey(0.15,'L20rb_ml','L20rb_ml','E0')
+        #init_survey(12345.12,'1-5', '22y','E0')
+        #init_survey(12345.12,'1-5', '22y','E5')
+        #init_survey(12345.13,'1-5', '22y','E5')
+        init_survey(12345.14,'1-5', '22y','E5')
+
+        #init_survey(version,'GC2019','GC','E0')
+        #init_survey(34.0,'34','S34','E0')
+        #init_survey(5.0,'5','S5','E0')
+        #init_survey(12345.0,'1-5','S1-5','E0')
+
+        #init_survey(0.12,'L20','L20','E0')
+        #init_survey(0.12,'L20p1','L20p1','E0')
+        #init_survey(0.12,'L20p2','L20p2','E0')
+        #init_survey(0.12,'L20p3','L20p3','E0')
+        #init_survey(0.12,'L20p4','L20p4','E0')
+
+        
+        version=0.11
+        #fields=[50,45,40,35,30,'025','020','015','010','005','000','355','350','345',]
+        #fields=['070','075','080','085','090']
+        #fields=['270','275','280','285','290','295','300','305','310','215','320','325']
+        #fields=['315']
+        
+        fields=['330','335','340']
+        """
+        for field in fields:
+            init_survey(version,"GP{}".format(field),"GP{}".format(field),'E0')
+            init_survey(version,"GP{}".format(field),"GP{}".format(field),'E1')
+            init_survey(version,"GP{}".format(field),"GP{}".format(field),'E2')
+            init_survey(version,"GP{}".format(field),"GP{}".format(field),'E3')
+            init_survey(version,"GP{}".format(field),"GP{}".format(field),'E4')
+        """
+
+        self.stdout.write(self.style.SUCCESS('Done'))