Faster Way To Add Late Starts

core/admin.py

@@ -24,6 +24,7 @@
     TermAdminForm,
     UserAdminForm,
     UserCreationAdminForm,
+    LateStartEventForm
 )
 from .models import Comment, StaffMember
 from .utils.actions import (
@@ -48,6 +49,11 @@
     PostTypeFilter,
 )
 
+from django.template.response import TemplateResponse
+from django.shortcuts import redirect
+from django.urls import path
+from datetime import datetime, time
+
 User = get_user_model()
 
 # Register your models here.
@@ -546,13 +552,59 @@ def formfield_for_manytomany(self, db_field, request, **kwargs):
                 kwargs["queryset"] = models.Tag.objects.all().order_by("name")
         return super().formfield_for_manytomany(db_field, request, **kwargs)
 
-
 class EventAdmin(CustomTimeMixin, admin.ModelAdmin):
     list_display = ["name", "organization", "start_date", "end_date"]
     list_filter = [OrganizationListFilter]
     ordering = ["-start_date", "-end_date"]
     search_fields = ["name"]
 
+    def get_urls(self):  
+        return [
+            path(
+                "createLateStart/",
+                self.admin_site.admin_view(self.late_start_view)
+            ),
+            *super().get_urls(),
+        ]
+
+    def late_start_view(self, request):
+
+        url = request.get_full_path()
+
+        context = dict(
+            self.admin_site.each_context(request),
+            form=LateStartEventForm,
+            url=url,
+            title='Add Late Start',
+            media=LateStartEventForm().media
+        )
+
+        if request.method == 'POST':
+            form = LateStartEventForm(request.POST)
+            if form.is_valid():
+                start_date_value = form.cleaned_data.get('start_date')
+                start_date = datetime.combine(start_date_value, time(hour=10))
+                end_date = datetime.combine(start_date_value, time(hour=10, second=1))
+
+                organization = form.cleaned_data.get('organization')
+
+                data = {
+                    'name': 'Late Start',
+                    'term': models.Term.get_current(start_date),
+                    'organization': organization,
+                    'schedule_format': 'late-start',
+                    'start_date': start_date,
+                    'end_date': end_date
+                }
+
+                models.Event.objects.create(**data)
+                return redirect("/admin/core/event")
+            else:
+                context['form'] = form 
+                return TemplateResponse(request, "admin/custom_form.html", context)
+        else:
+            return TemplateResponse(request, "admin/custom_form.html", context)
+
     def get_queryset(self, request):
         qs = super().get_queryset(request)
         if request.user.is_superuser:

core/forms.py

@@ -5,6 +5,7 @@
 from django.contrib.auth.forms import (
     AdminUserCreationForm as ContribAdminUserCreationForm,
 )
+from django.contrib.admin.widgets import AdminDateWidget
 from django.utils import timezone
 from django_select2 import forms as s2forms
 from martor.widgets import AdminMartorWidget
@@ -13,6 +14,7 @@
 from core.views.mixins import CaseInsensitiveUsernameMixin
 
 
+
 class MetropolisSignupForm(SignupForm, CaseInsensitiveUsernameMixin):
     first_name = forms.CharField(
         max_length=30,
@@ -295,3 +297,26 @@ class UserAdminForm(CaseInsensitiveUsernameMixin, ContribUserChangeForm):
 
 class UserCreationAdminForm(CaseInsensitiveUsernameMixin, ContribAdminUserCreationForm):
     pass
+
+class LateStartEventForm(forms.Form):
+    start_date = forms.DateField(widget=AdminDateWidget())
+    organization = forms.ModelChoiceField(
+        queryset=models.Organization.objects.all(),
+        required=True
+    )
+
+    def __init__(self, *args, **kwargs):
+        super(LateStartEventForm, self).__init__(*args, **kwargs)
+
+        try:
+            self.fields["organization"].initial = models.Organization.objects.get(name='SAC')
+        except models.Organization.DoesNotExist:
+            pass 
+
+
+    def clean(self):
+        cleaned_data = super().clean()
+        if cleaned_data.get('start_date') != None and models.Term.get_current(cleaned_data['start_date']) == None:
+            raise forms.ValidationError(
+                {'start_date': 'No Term Found For Date'}
+            )

core/models/course.py

@@ -266,7 +266,7 @@ def get_events(cls, user=None):
         return events
 
     def clean(self):
-        if self.start_date > self.end_date:
+        if self.start_date != None and self.end_date != None and self.start_date > self.end_date:
             raise ValidationError(
                 {
                     "start_date": _("Start date must be before end date"),

templates/admin/custom_form.html

@@ -0,0 +1,49 @@
+{% extends "admin/base_site.html" %}
+{% load i18n admin_urls static admin_modify %}
+
+{% block extrahead %}
+    {{ block.super }}
+    <script src="{% url 'admin:jsi18n' %}"></script>
+    <script src="{% static "admin/js/core.js" %}"></script>
+    {{ media }}
+{% endblock %}
+
+{% block extrastyle %}
+    {{ block.super }}
+    <link rel="stylesheet" type="text/css" href="{% static "admin/css/forms.css" %}">
+{% endblock %}
+
+{% if not is_popup %}
+    {% block breadcrumbs %}<div class="breadcrumbs" style="height:0px;padding:0px"></div>{% endblock %}
+{% endif %}
+
+{% block coltype %}colM{% endblock %}
+{% block content %}
+    <form action="{{ url }}" method="post">
+        {% csrf_token %}
+        {% if form.errors %}
+            <p class="errornote">
+                {% if form.errors|length == 1 %}
+                    {% translate "Please correct the error below." %}
+                {% else %}
+                    {% translate "Please correct the errors below." %}
+                {% endif %}
+            </p>
+        {% endif %}
+        <fieldset class="module aligned ">
+            {% for field in form %}
+                <div class="form-row field-name">
+                    {{ field.errors }}
+                    <div>
+                        <div class="flex-container">
+                            {{ field.label_tag }} {{ field }}
+                        </div>
+                    </div>
+                </div>
+            {% endfor %}
+        </fieldset>
+        <div class="submit-row">
+            <input class="default" type="submit" value="Save">
+        </div>
+    </form>
+{% endblock %}