Add teams

apps/wei/forms.py

@@ -4,9 +4,9 @@
 from django import forms
 from django.contrib.auth.models import User
 from django.utils.translation import ugettext_lazy as _
-from note_kfet.inputs import AmountInput, DatePickerInput, Autocomplete
+from note_kfet.inputs import AmountInput, DatePickerInput, Autocomplete, ColorWidget
 
-from .models import WEIClub, WEIRegistration, Bus
+from .models import WEIClub, WEIRegistration, Bus, BusTeam
 
 
 class WEIForm(forms.ModelForm):
@@ -62,7 +62,7 @@ class BusForm(forms.ModelForm):
 
 class BusTeamForm(forms.ModelForm):
     class Meta:
-        model = Bus
+        model = BusTeam
         fields = '__all__'
         widgets = {
             "bus": Autocomplete(
@@ -72,4 +72,5 @@ class BusTeamForm(forms.ModelForm):
                     'placeholder': 'Bus ...',
                 },
             ),
+            "color": ColorWidget(),
         }

apps/wei/models.py

@@ -50,6 +50,12 @@ class Bus(models.Model):
         verbose_name=_("name"),
     )
 
+    description = models.TextField(
+        blank=True,
+        default="",
+        verbose_name=_("description"),
+    )
+
     def __str__(self):
         return self.name
 
@@ -77,6 +83,12 @@ class BusTeam(models.Model):
         help_text=_("The color of the T-Shirt, stored with its number equivalent"),
     )
 
+    description = models.TextField(
+        blank=True,
+        default="",
+        verbose_name=_("description"),
+    )
+
     def __str__(self):
         return self.name + " (" + str(self.bus) + ")"
 

apps/wei/tables.py

@@ -107,18 +107,22 @@ class BusTeamTable(tables.Table):
     color = tables.Column(
         attrs={
             "td": {
-                "style": lambda record: "background-color: #" + "".format(record.color) + ";"
+                "style": lambda record: "background-color: #{:06X}; color: #{:06X};"
+                                        .format(record.color, 0xFFFFFF - record.color, )
             }
         }
     )
 
+    def render_color(self, value):
+        return "#{:06X}".format(value)
+
     class Meta:
         attrs = {
             'class': 'table table-condensed table-striped table-hover'
         }
         model = BusTeam
         template_name = 'django_tables2/bootstrap4.html'
-        fields = ('name', 'color', 'team',)
+        fields = ('name', 'color',)
         row_attrs = {
             'class': 'table-row',
             'id': lambda record: "row-" + str(record.pk),

apps/wei/urls.py

@@ -3,7 +3,8 @@
 
 from django.urls import path
 
-from .views import WEIListView, WEICreateView, WEIDetailView, WEIUpdateView, BusCreateView,\
+from .views import WEIListView, WEICreateView, WEIDetailView, WEIUpdateView,\
+    BusCreateView, BusManageView, BusUpdateView, BusTeamCreateView,\
     WEIRegisterView, WEIUpdateRegistrationView
 
 
@@ -14,6 +15,9 @@ urlpatterns = [
     path('detail/<int:pk>/', WEIDetailView.as_view(), name="wei_detail"),
     path('update/<int:pk>/', WEIUpdateView.as_view(), name="wei_update"),
     path('add-bus/<int:pk>/', BusCreateView.as_view(), name="add_bus"),
+    path('manage-bus/<int:pk>/', BusManageView.as_view(), name="manage_bus"),
+    path('update-bus/<int:pk>/', BusUpdateView.as_view(), name="update_bus"),
+    path('add-bus-team/<int:pk>/', BusTeamCreateView.as_view(), name="add_team"),
     path('register/<int:wei_pk>/', WEIRegisterView.as_view(), name="wei_register"),
     path('edit-registration/<int:pk>/', WEIUpdateRegistrationView.as_view(), name="wei_update_registration"),
 ]

apps/wei/views.py

@@ -16,9 +16,9 @@ from note.tables import HistoryTable
 from permission.backends import PermissionBackend
 from permission.views import ProtectQuerysetMixin
 
-from .models import WEIClub, WEIRegistration, WEIMembership, Bus
-from .forms import WEIForm, WEIRegistrationForm, BusForm
-from .tables import WEITable, WEIRegistrationTable, BusTable
+from .models import WEIClub, WEIRegistration, WEIMembership, Bus, BusTeam
+from .forms import WEIForm, WEIRegistrationForm, BusForm, BusTeamForm
+from .tables import WEITable, WEIRegistrationTable, BusTable, BusTeamTable
 
 
 class WEIListView(ProtectQuerysetMixin, LoginRequiredMixin, SingleTableView):
@@ -141,7 +141,71 @@ class BusCreateView(ProtectQuerysetMixin, LoginRequiredMixin, CreateView):
 
     def get_success_url(self):
         self.object.refresh_from_db()
-        return reverse_lazy("wei:wei_detail", kwargs={"pk": self.object.wei.pk})
+        return reverse_lazy("wei:manage_bus", kwargs={"pk": self.object.pk})
+
+
+class BusUpdateView(ProtectQuerysetMixin, LoginRequiredMixin, UpdateView):
+    """
+    Update Bus
+    """
+    model = Bus
+    form_class = BusForm
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context["club"] = self.object.wei
+        return context
+
+    def get_form(self, form_class=None):
+        form = super().get_form(form_class)
+        form.fields["wei"].disabled = True
+        return form
+
+    def get_success_url(self):
+        self.object.refresh_from_db()
+        return reverse_lazy("wei:manage_bus", kwargs={"pk": self.object.pk})
+
+
+class BusManageView(ProtectQuerysetMixin, LoginRequiredMixin, DetailView):
+    """
+    Manage Bus
+    """
+    model = Bus
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context["club"] = self.object.wei
+
+        bus = self.object
+        teams = BusTeam.objects.filter(PermissionBackend.filter_queryset(self.request.user, BusTeam, "view"))\
+            .filter(bus=bus)
+        teams_table = BusTeamTable(data=teams, prefix="teams-")
+        context["teams"] = teams_table
+
+        return context
+
+
+class BusTeamCreateView(ProtectQuerysetMixin, LoginRequiredMixin, CreateView):
+    """
+    Create BusTeam
+    """
+    model = BusTeam
+    form_class = BusTeamForm
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        bus = Bus.objects.get(pk=self.kwargs["pk"])
+        context["club"] = bus.wei
+        return context
+
+    def get_form(self, form_class=None):
+        form = super().get_form(form_class)
+        form.fields["bus"].initial = Bus.objects.get(pk=self.kwargs["pk"])
+        return form
+
+    def get_success_url(self):
+        self.object.refresh_from_db()
+        return reverse_lazy("wei:manage_bus", kwargs={"pk": self.object.bus.pk})
 
 
 class WEIRegisterView(ProtectQuerysetMixin, LoginRequiredMixin, CreateView):
@@ -180,4 +244,3 @@ class WEIUpdateRegistrationView(ProtectQuerysetMixin, LoginRequiredMixin, Update
     def get_success_url(self):
         self.object.refresh_from_db()
         return reverse_lazy("wei:wei_detail", kwargs={"pk": self.object.wei.pk})
-