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