170 lines
5.6 KiB
Python
170 lines
5.6 KiB
Python
from django.apps import apps
|
|
from django.contrib import admin
|
|
from django.shortcuts import HttpResponse
|
|
from django.contrib.admin.sites import AlreadyRegistered
|
|
from .models import Announcement, Edition, School, Student
|
|
from django.db import transaction
|
|
|
|
from io import BytesIO
|
|
import xlsxwriter
|
|
|
|
|
|
class EditionModelAdmin(admin.ModelAdmin):
|
|
list_display = ('__str__', 'active')
|
|
ordering = ('-active', '-year')
|
|
|
|
change_form_template = 'admin/edition_buttons.html'
|
|
|
|
def response_change(self, request, obj):
|
|
if '_update-student-scores' in request.POST:
|
|
file = obj.scores.open(mode='r')
|
|
content = filter(lambda x: x != '', file.read().split('\n'))
|
|
file.close()
|
|
|
|
rows = [x.split(',') for x in content]
|
|
|
|
with transaction.atomic():
|
|
for index, row in enumerate(rows):
|
|
if index == 0:
|
|
continue
|
|
|
|
Student.objects.filter(identifier=row[0]).update(
|
|
score_first=row[1], score_second=row[2])
|
|
|
|
if '_update-disable-submissions' in request.POST and obj.active and obj.submissions:
|
|
if obj.active is True:
|
|
schools = School.objects.all()
|
|
|
|
with transaction.atomic():
|
|
for school in schools:
|
|
school.updateTeam()
|
|
|
|
obj.submissions = False
|
|
obj.save()
|
|
|
|
if '_generate-student-list' in request.POST:
|
|
year = obj.year
|
|
students = Student.objects.filter(identifier__startswith=year)
|
|
schools_set = {int(student.identifier.split('-')
|
|
[1]) for student in students}
|
|
|
|
output = BytesIO()
|
|
workbook = xlsxwriter.Workbook(output)
|
|
|
|
worksheet = workbook.add_worksheet("Lista uczniów")
|
|
|
|
worksheet.write(0, 0, 'Identyfikator')
|
|
worksheet.write(0, 1, 'Wynik - eliminacje')
|
|
worksheet.write(0, 2, 'Wynik - finał')
|
|
worksheet.write(0, 3, 'Imię')
|
|
worksheet.write(0, 4, 'Nazwisko')
|
|
worksheet.write(0, 5, 'Klasa')
|
|
worksheet.write(0, 6, 'Szkoła - nazwa')
|
|
worksheet.write(0, 7, 'Szkoła - miejscowość')
|
|
worksheet.write(0, 8, 'Szkoła - adres')
|
|
|
|
row = 1
|
|
for school_id in schools_set:
|
|
for student in sorted(Student.objects.filter(identifier__startswith=f'{year}-{school_id}-'), key=lambda x: x.identifier.split('-')[2]):
|
|
# Identyfikator
|
|
worksheet.write(row, 0, student.identifier)
|
|
# Wynik - eliminacje
|
|
worksheet.write(row, 1, student.score_first)
|
|
# Wynik - finał
|
|
worksheet.write(row, 2, student.score_second)
|
|
worksheet.write(row, 3, student.name) # Imię
|
|
worksheet.write(row, 4, student.surname) # Nazwisko
|
|
worksheet.write(row, 5, student.grade) # Klasa
|
|
# Szkoła - nazwa
|
|
worksheet.write(row, 6, student.school_name)
|
|
# Szkoła - miejscowość
|
|
worksheet.write(row, 7, student.school_town)
|
|
# Szkoła - adres
|
|
worksheet.write(row, 8, student.school_address)
|
|
|
|
row += 1
|
|
|
|
workbook.close()
|
|
xlsx_data = output.getvalue()
|
|
|
|
response = HttpResponse(content_type='application/vnd.ms-excel')
|
|
response['Content-Disposition'] = 'attachment; filename=Uczniowie.xlsx'
|
|
response.write(xlsx_data)
|
|
|
|
return response
|
|
|
|
return super().response_change(request, obj)
|
|
|
|
|
|
class StudentYearFilter(admin.SimpleListFilter):
|
|
title = 'aktualna edycja'
|
|
parameter_name = 'aktualna edycja'
|
|
|
|
def lookups(self, request, model_admin):
|
|
return (
|
|
('Yes', 'Jedynie aktualna edycja'),
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
value = self.value()
|
|
year = Edition.current().year
|
|
|
|
if value == 'Yes':
|
|
return queryset.filter(identifier__startswith=str(year))
|
|
|
|
return queryset
|
|
|
|
|
|
class StudentModelAdmin(admin.ModelAdmin):
|
|
list_display = ('identifier', '__str__', 'score_first',
|
|
'score_second', 'school_name', 'school_town', 'grade')
|
|
search_fields = ('identifier', 'name', 'surname', 'grade')
|
|
ordering = ('-identifier',)
|
|
list_filter = (StudentYearFilter,)
|
|
|
|
|
|
class SchoolYearFilter(admin.SimpleListFilter):
|
|
title = 'aktualna edycja'
|
|
parameter_name = 'aktualna edycja'
|
|
|
|
def lookups(self, request, model_admin):
|
|
return (
|
|
('Yes', 'Jedynie aktualna edycja'),
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
value = self.value()
|
|
year = Edition.current().year
|
|
|
|
if value == 'Yes':
|
|
return queryset.filter(identifier__startswith=str(year))
|
|
|
|
return queryset
|
|
|
|
|
|
@admin.register(School)
|
|
class SchoolAdmin(admin.ModelAdmin):
|
|
list_display = ('view_name', 'in_current', 'email', 'phone')
|
|
|
|
@admin.display(description='Nazwa', empty_value='-')
|
|
def view_name(self, obj):
|
|
return obj.name or '-'
|
|
|
|
|
|
admin.site.register(Student, StudentModelAdmin)
|
|
admin.site.register(Edition, EditionModelAdmin)
|
|
|
|
|
|
app_models = apps.get_app_config('app').get_models()
|
|
for model in app_models:
|
|
try:
|
|
admin.site.register(model)
|
|
except AlreadyRegistered:
|
|
pass
|
|
|
|
admin.site.site_header = 'Puchar Dyrektora LO V'
|
|
admin.site.site_title = 'Puchar LO V'
|
|
admin.site.index_title = 'Panel administracyjny'
|
|
admin.site.site_url = '/'
|
|
admin.site.enable_nav_sidebar = False
|