move_preprints manage command for janeway

README.md

usage: manage.py move_preprints [-h] [--version] [-v {0,1,2,3}] [--settings SETTINGS] [--pythonpath PYTHONPATH] [--traceback] [--no-color] active_user proxy_user

move preprints from a proxy account to a new account

positional arguments:
  active_user           `email` of new active account
  proxy_user            `email` of old proxy account

optional arguments:
  -h, --help            show this help message and exit
  --version             show program's version number and exit
  -v {0,1,2,3}, --verbosity {0,1,2,3}
                        Verbosity level; 0=minimal output, 1=normal output, 2=verbose output, 3=very verbose output
  --settings SETTINGS   The Python path to a settings module, e.g. "myproject.settings.main". If this isn't provided, the DJANGO_SETTINGS_MODULE environment variable will be used.
  --pythonpath PYTHONPATH
                        A directory to add to the Python path, e.g. "/home/djangoprojects/myproject".
  --traceback           Raise on CommandError exceptions
  --no-color            Don't colorize the command output.

move_preprints.py

from django.core.management.base import BaseCommand, CommandError
from django.db import connection

from core.models import Account


class Command(BaseCommand):
    help = "move preprints from a proxy account to a new account"

    def add_arguments(self, parser):
        parser.add_argument(
            "active_user", help="`email` of new active account", type=str
        )
        parser.add_argument("proxy_user", help="`email` of old proxy account", type=str)

    def handle(self, *args, **options):
        active_user = Account.objects.get(email=options["active_user"])
        proxy_user = Account.objects.get(email=options["proxy_user"])

        if proxy_user == active_user:
            raise CommandError(
                "active_user and proxy_user have the same id, nothing to do"
            )

        if active_user.is_active is False:
            raise CommandError(
                "active_user {} must have an active account".format(
                    options["active_user"]
                )
            )

        if proxy_user.is_active is True:
            raise CommandError(
                "proxy_user {} must not have an active account".format(
                    options["proxy_user"]
                )
            )

        query1 = (
            "update janeway.repository_preprint set owner_id={} where owner_id={};".format(
                active_user.id, proxy_user.id
            )
        )
        query2 = "delete from core_account where id={};".format(proxy_user.id)

        self.stdout.write(self.style.SUCCESS(query1))
        self.stdout.write(self.style.SUCCESS(query2))
        with connection.cursor() as cursor:
            cursor.execute(query1)
            cursor.execute(query2)
            cursor.fetchall()