mirror of
https://gitlab.crans.org/bde/nk20
synced 2024-11-30 04:13:01 +00:00
a6b479db19
- /apps/activity/api/serializers.py - /apps/activity/api/urls.py - /apps/activity/api/views.py - /apps/activity/tests/test_activities.py - /apps/activity/__init__.py - /apps/activity/admin.py - /apps/activity/apps.py - /apps/activity/forms.py - /apps/activity/tables.py - /apps/activity/urls.py - /apps/activity/views.py - /apps/api/__init__.py - /apps/api/apps.py - /apps/api/serializers.py - /apps/api/tests.py - /apps/api/urls.py - /apps/api/views.py - /apps/api/viewsets.py - /apps/logs/signals.py - /apps/logs/apps.py - /apps/logs/__init__.py - /apps/logs/api/serializers.py - /apps/logs/api/urls.py - /apps/logs/api/views.py - /apps/member/api/serializers.py - /apps/member/api/urls.py - /apps/member/api/views.py - /apps/member/templatetags/memberinfo.py - /apps/member/__init__.py - /apps/member/admin.py - /apps/member/apps.py - /apps/member/auth.py - /apps/member/forms.py - /apps/member/hashers.py - /apps/member/signals.py - /apps/member/tables.py - /apps/member/urls.py - /apps/member/views.py - /apps/note/api/serializers.py - /apps/note/api/urls.py - /apps/note/api/views.py - /apps/note/models/__init__.py - /apps/note/static/note/js/consos.js - /apps/note/templates/note/mails/negative_balance.txt - /apps/note/templatetags/getenv.py - /apps/note/templatetags/pretty_money.py - /apps/note/tests/test_transactions.py - /apps/note/__init__.py - /apps/note/admin.py - /apps/note/apps.py - /apps/note/forms.py - /apps/note/signals.py - /apps/note/tables.py - /apps/note/urls.py - /apps/note/views.py - /apps/permission/api/serializers.py - /apps/permission/api/urls.py - /apps/permission/api/views.py - /apps/permission/templatetags/perms.py - /apps/permission/tests/test_oauth2.py - /apps/permission/tests/test_permission_denied.py - /apps/permission/tests/test_permission_queries.py - /apps/permission/tests/test_rights_page.py - /apps/permission/__init__.py - /apps/permission/admin.py - /apps/permission/backends.py - /apps/permission/apps.py - /apps/permission/decorators.py - /apps/permission/permissions.py - /apps/permission/scopes.py - /apps/permission/signals.py - /apps/permission/tables.py - /apps/permission/urls.py - /apps/permission/views.py - /apps/registration/tests/test_registration.py - /apps/registration/__init__.py - /apps/registration/apps.py - /apps/registration/forms.py - /apps/registration/tables.py - /apps/registration/tokens.py - /apps/registration/urls.py - /apps/registration/views.py - /apps/treasury/api/serializers.py - /apps/treasury/api/urls.py - /apps/treasury/api/views.py - /apps/treasury/templatetags/escape_tex.py - /apps/treasury/tests/test_treasury.py - /apps/treasury/__init__.py - /apps/treasury/admin.py - /apps/treasury/apps.py - /apps/treasury/forms.py - /apps/treasury/signals.py - /apps/treasury/tables.py - /apps/treasury/urls.py - /apps/treasury/views.py - /apps/wei/api/serializers.py - /apps/wei/api/urls.py - /apps/wei/api/views.py - /apps/wei/forms/surveys/__init__.py - /apps/wei/forms/surveys/base.py - /apps/wei/forms/surveys/wei2021.py - /apps/wei/forms/surveys/wei2022.py - /apps/wei/forms/surveys/wei2023.py - /apps/wei/forms/__init__.py - /apps/wei/forms/registration.py - /apps/wei/management/commands/export_wei_registrations.py - /apps/wei/management/commands/import_scores.py - /apps/wei/management/commands/wei_algorithm.py - /apps/wei/templates/wei/weilist_sample.tex - /apps/wei/tests/test_wei_algorithm_2021.py - /apps/wei/tests/test_wei_algorithm_2022.py - /apps/wei/tests/test_wei_algorithm_2023.py - /apps/wei/tests/test_wei_registration.py - /apps/wei/__init__.py - /apps/wei/admin.py - /apps/wei/apps.py - /apps/wei/tables.py - /apps/wei/urls.py - /apps/wei/views.py - /note_kfet/settings/__init__.py - /note_kfet/settings/base.py - /note_kfet/settings/development.py - /note_kfet/settings/secrets_example.py - /note_kfet/static/js/base.js - /note_kfet/admin.py - /note_kfet/inputs.py - /note_kfet/middlewares.py - /note_kfet/urls.py - /note_kfet/views.py - /note_kfet/wsgi.py - /entrypoint.sh
59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
# Copyright (C) 2018-2024 by BDE ENS Paris-Saclay
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
from argparse import ArgumentParser, FileType
|
|
|
|
from django.core.management import BaseCommand
|
|
from django.db import transaction
|
|
|
|
from ...forms import CurrentSurvey
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Attribute to each first year member a bus for the WEI"
|
|
|
|
def add_arguments(self, parser: ArgumentParser):
|
|
parser.add_argument('--doit', '-d', action='store_true', help='Finally run the algorithm in non-dry mode.')
|
|
parser.add_argument('--output', '-o', nargs='?', type=FileType('w'), default=self.stdout,
|
|
help='Output file for the algorithm result. Default is standard output.')
|
|
|
|
@transaction.atomic
|
|
def handle(self, *args, **options):
|
|
"""
|
|
Run the WEI algorithm to attribute a bus to each first year member.
|
|
"""
|
|
sid = transaction.savepoint()
|
|
|
|
algorithm = CurrentSurvey.get_algorithm_class()()
|
|
|
|
try:
|
|
from tqdm import tqdm
|
|
del tqdm
|
|
display_tqdm = True
|
|
except ImportError:
|
|
display_tqdm = False
|
|
|
|
algorithm.run_algorithm(display_tqdm=display_tqdm)
|
|
|
|
output = options['output']
|
|
registrations = algorithm.get_registrations()
|
|
per_bus = {bus: [r for r in registrations if 'selected_bus_pk' in r.information
|
|
and r.information['selected_bus_pk'] == bus.pk]
|
|
for bus in algorithm.get_buses()}
|
|
for bus, members in per_bus.items():
|
|
output.write(bus.name + "\n")
|
|
output.write("=" * len(bus.name) + "\n")
|
|
_order = -1
|
|
for r in members:
|
|
survey = CurrentSurvey(r)
|
|
for _order, (b, _score) in enumerate(survey.ordered_buses()):
|
|
if b == bus:
|
|
break
|
|
output.write(f"{r.user.username} ({_order + 1})\n")
|
|
output.write("\n")
|
|
|
|
if not options['doit']:
|
|
self.stderr.write(self.style.WARNING("Running in dry mode. "
|
|
"Use --doit option to really execute the algorithm."))
|
|
transaction.savepoint_rollback(sid)
|
|
return
|