boards/management/commands/generate_keypair.py

__author__ = 'neko259'


from django.core.management import BaseCommand
from django.db import transaction

from boards.models import KeyPair, Post


class Command(BaseCommand):
    help = 'Generates the new keypair. The first one will be primary.'

    @transaction.atomic
    def handle(self, *args, **options):
        first_key = not KeyPair.objects.has_primary()
        key = KeyPair.objects.generate_key(
            primary=first_key)
        print(key)

        if first_key:
            for post in Post.objects.filter(global_id=None):
                post.set_global_id()