Add command for generating cite command for given publications

This commit is contained in:
Maximilian Blochberger committed 2021-03-17 08:48:33 +01:00
1 parent 159e00763e
commit b7f4c941c1
1 file changed
+17
+17
View File
@@ -0,0 +1,17 @@
from typing import List
from django.core.management.base import BaseCommand, CommandParser
from sok.models import Publication
class Command(BaseCommand):
def add_arguments(self, parser: CommandParser):
parser.add_argument('pk', nargs='+', type=int)
def handle(self, *args, **options):
pks: List[int] = options['pk']
publications = [Publication.objects.get(pk=pk) for pk in pks]
cite_keys = [publication.cite_key for publication in publications]
self.stdout.write(r"\cite{" + ",".join(cite_keys) + "}", ending='')