srg/heasarc/management/commands/heasarc_reference_paper.py
2024-04-25 17:50:51 +03:00

42 lines
1.4 KiB
Python

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