HEASARC

heasarc/management/commands/heasarc_reference_paper.py

@@ -0,0 +1,41 @@
+from django.core.management.base import BaseCommand, CommandError
+
+from nasa.models import Category
+from nasa.models import Ads
+
+from heasarc.models import HeasarcCSC
+from heasarc.models import Heasarc4XMMDR12
+from heasarc.models import HeasarcXMMSSC
+from heasarc.models import HeasarcRASS2RXS
+from heasarc.models import HeasarcXTEASSCAT
+from heasarc.models import Heasarc2SXPS
+from heasarc.models import Heasarc4XMMDR9
+from heasarc.models import Heasarc4XMMDR10
+
+def reference_paper(bibcode,survey):
+    category_open = Category.objects.get(slug="open")
+    #print(category_open)
+        
+    ads = Ads(bibcode=bibcode)
+    ads.save()
+    ads.category.add(category_open)
+
+    srcs = survey.objects.all()
+    print ("Selected {} total from {}".format(srcs.count(), survey))
+    srcs.update(ads=ads)
+
+
+class Command(BaseCommand):
+    help = 'Initiates data dase'
+
+    def handle(self, *args, **options):
+        #reference_paper("2010ApJS..189...37E", HeasarcCSC)
+        #reference_paper("2020A&A...641A.136W",Heasarc4XMMDR12)
+        reference_paper("2020A&A...641A.136W",HeasarcXMMSSC)
+        reference_paper("2016A&A...588A.103B",HeasarcRASS2RXS)
+        reference_paper("2004A&A...418..927R",HeasarcXTEASSCAT)
+        reference_paper("2020ApJS..247...54E",Heasarc2SXPS)
+        reference_paper("2020A&A...641A.137T",Heasarc4XMMDR9)
+        reference_paper("2020A&A...641A.136W",Heasarc4XMMDR10)
+        self.stdout.write(self.style.SUCCESS('Done'))
+