TEST: Add test datetime fields

ThiefMaster · ThiefMaster · commit dc466e43ab42 · 2025-04-07T12:04:37.000+02:00
diff --git a/indico/modules/announcement/controllers.py b/indico/modules/announcement/controllers.py
@@ -5,22 +5,27 @@
 # modify it under the terms of the MIT License; see the
 # LICENSE file for more details.
 
+from datetime import datetime
+
 from flask import flash, redirect
 
 from indico.modules.admin import RHAdminBase
 from indico.modules.announcement import announcement_settings
 from indico.modules.announcement.forms import AnnouncementForm
 from indico.modules.announcement.views import WPAnnouncement
-from indico.util.i18n import _
+from indico.util.date_time import server_to_utc
+# from indico.util.i18n import _
 from indico.web.flask.util import url_for
 from indico.web.forms.base import FormDefaults
 
 
 class RHAnnouncement(RHAdminBase):
     def _process(self):
-        form = AnnouncementForm(obj=FormDefaults(**announcement_settings.get_all()))
+        form = AnnouncementForm(obj=FormDefaults(**announcement_settings.get_all(),
+                                                 test_start_dt=server_to_utc(datetime(2025, 4, 8, 13, 37))))
         if form.validate_on_submit():
-            announcement_settings.set_multi(form.data)
-            flash(_('Settings have been saved'), 'success')
+            # announcement_settings.set_multi(form.data)
+            # flash(_('Settings have been saved'), 'success')
+            flash(f'{form.test_start_dt.data} - {form.test_end_dt.data}', 'info')
             return redirect(url_for('announcement.manage'))
         return WPAnnouncement.render_template('settings.html', 'announcement', form=form)
diff --git a/indico/modules/announcement/forms.py b/indico/modules/announcement/forms.py
@@ -5,16 +5,28 @@
 # modify it under the terms of the MIT License; see the
 # LICENSE file for more details.
 
+from datetime import datetime
+
 from wtforms.fields import BooleanField, TextAreaField
-from wtforms.validators import DataRequired
+from wtforms.validators import DataRequired, Optional
 
+from indico.util.date_time import server_to_utc
 from indico.util.i18n import _
 from indico.web.forms.base import IndicoForm
-from indico.web.forms.validators import UsedIf
+from indico.web.forms.fields.datetime import IndicoDateTimeField
+from indico.web.forms.validators import DateTimeRange, LinkedDateTime, UsedIf
 from indico.web.forms.widgets import SwitchWidget
 
 
 class AnnouncementForm(IndicoForm):
     enabled = BooleanField(_('Enabled'), widget=SwitchWidget())
     message = TextAreaField(_('Message'), [UsedIf(lambda form, _: form.enabled.data), DataRequired()],
                             description=_('You may use Markdown and basic HTML elements for formatting.'))
+
+    test_start_dt = IndicoDateTimeField('Start', [DataRequired(),
+                                                  DateTimeRange(earliest=server_to_utc(datetime(2025, 4, 5)),
+                                                               latest=server_to_utc(datetime(2025, 5, 20)))])
+    test_end_dt = IndicoDateTimeField('End', [Optional(), LinkedDateTime('test_start_dt'),
+                                              DateTimeRange(earliest=None,
+                                                            latest=server_to_utc(datetime(2025, 6, 20, 15)))],
+                                       allow_clear=True)
