Source code for oioioi.participants.management.commands.import_participants

import os

import urllib.request
from django.contrib.auth.models import User
from django.core.management.base import BaseCommand, CommandError
from django.db import DatabaseError, transaction
from django.utils.translation import gettext as _

from oioioi.contests.models import Contest
from oioioi.participants.admin import ParticipantAdmin
from oioioi.participants.models import Participant


class Command(BaseCommand):
    help = _(
        "Updates the list of participants of <contest_id> from the given "
        "text file (one login per line).\n"
        "Lines starting with '#' are ignored."
    )

    requires_model_validation = True

    def add_arguments(self, parser):
        parser.add_argument('contest_id', type=str, help='Contest to import to')
        parser.add_argument('filename_or_url', type=str, help='Source CSV file')

    def handle(self, *args, **options):
        try:
            contest = Contest.objects.get(id=options['contest_id'])
        except Contest.DoesNotExist:
            raise CommandError(_("Contest %s does not exist") % options['contest_id'])

        rcontroller = contest.controller.registration_controller()
        if not issubclass(
            getattr(rcontroller, 'participant_admin', type(None)), ParticipantAdmin
        ):
            raise CommandError(_("Wrong type of contest"))

        arg = options['filename_or_url']

        if arg.startswith('http://') or arg.startswith('https://'):
            self.stdout.write(_("Fetching %s...\n") % (arg,))
            stream = urllib.request.urlopen(arg)
        else:
            if not os.path.exists(arg):
                raise CommandError(_("File not found: %s") % arg)
            stream = open(arg, 'r')

        with transaction.atomic():
            ok = True
            all_count = 0
            for line in stream:
                line = line.strip()
                if not line:
                    continue
                if line.startswith('#'):
                    continue
                login = line
                if not isinstance(login, str):
                    login = login.decode('utf8')
                all_count += 1

                try:
                    user = User.objects.get(username=login)
                    _participant, created = Participant.objects.get_or_create(
                        contest=contest, user=user
                    )
                except User.DoesNotExist:
                    self.stdout.write(
                        _("Error for user=%(user)s: user does not exist\n")
                        % {'user': login}
                    )
                    ok = False
                except DatabaseError as e:
                    # This assumes that we'll get the message in this
                    # encoding. It is not perfect, but much better than
                    # ascii.
                    message = e.message.decode('utf-8')
                    self.stdout.write(
                        _("DB Error for user=%(user)s: %(message)s\n")
                        % {'user': login, 'message': message}
                    )
                    ok = False
            if ok:
                self.stdout.write(_("Processed %d entries") % (all_count))
            else:
                raise CommandError(
                    _("There were some errors. Database not changed.\n")
                )