From d0dfd3a4c397c2f93dbe42eae0c52bd36e3fa877 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lera=20Elvo=C3=A9?= <yagich@poto.cafe>
Date: Thu, 27 Nov 2025 19:56:17 +0300
Subject: [PATCH] re-parse MOTDs if required on init

---
 app/__init__.py   | 17 ++++++++++++++++-
 app/constants.py  |  4 ++++
 app/routes/mod.py |  4 ++--
 3 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/app/__init__.py b/app/__init__.py
index f9dbf1b..b74226b 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -1,6 +1,6 @@
 from flask import Flask, session, request
 from dotenv import load_dotenv
-from .models import Avatars, Users, PostHistory, Posts
+from .models import Avatars, Users, PostHistory, Posts, MOTD
 from .auth import digest
 from .routes.users import is_logged_in, get_active_user, get_prefers_theme
 from .routes.threads import get_post_url
@@ -51,6 +51,7 @@ def create_deleted_user():
 def reparse_babycode():
     print('Re-parsing babycode, this may take a while...')
     from .db import db
+    from .constants import MOTD_BANNED_TAGS
     post_histories = PostHistory.findall([
         ('markup_language', '=', 'babycode'),
         ('format_version', 'IS NOT', BABYCODE_VERSION)
@@ -80,6 +81,20 @@ def reparse_babycode():
                 })
         print(f'Re-parsed {len(users_with_sigs)} user sigs.')
 
+    stale_motds = MOTD.findall([
+        ['markup_language', '=', 'babycode'],
+        ['format_version', 'IS NOT', BABYCODE_VERSION]
+    ])
+    if stale_motds:
+        print('Re-parsing MOTDs...')
+        with db.transaction():
+            for motd in stale_motds:
+                motd.update({
+                    'body_rendered': babycode_to_html(motd['body_original_markup'], banned_tags=MOTD_BANNED_TAGS),
+                    'format_version': BABYCODE_VERSION,
+                })
+        print('Re-parsing MOTDs done.')
+
     print('Re-parsing done.')
 
 def create_app():
diff --git a/app/constants.py b/app/constants.py
index da7c41f..495d4d3 100644
--- a/app/constants.py
+++ b/app/constants.py
@@ -47,6 +47,10 @@ REACTION_EMOJI = [
     'scissors',
 ]
 
+MOTD_BANNED_TAGS = [
+    'img', 'spoiler',
+]
+
 def permission_level_string(perm):
     return PermissionLevelString[PermissionLevel(int(perm))]
 
diff --git a/app/routes/mod.py b/app/routes/mod.py
index 0971ce2..8b3534d 100644
--- a/app/routes/mod.py
+++ b/app/routes/mod.py
@@ -4,7 +4,7 @@ from flask import (
     )
 from .users import get_active_user, is_logged_in
 from ..models import Users, PasswordResetLinks, MOTD
-from ..constants import InfoboxKind
+from ..constants import InfoboxKind, MOTD_BANNED_TAGS
 from ..lib.babycode import babycode_to_html, BABYCODE_VERSION
 from ..db import db
 import secrets
@@ -73,7 +73,7 @@ def motd_editor_form():
     data = {
             'title': title,
             'body_original_markup': orig_body,
-            'body_rendered': babycode_to_html(orig_body, banned_tags=['img', 'spoiler']),
+            'body_rendered': babycode_to_html(orig_body, banned_tags=MOTD_BANNED_TAGS),
             'format_version': BABYCODE_VERSION,
             'edited_at': int(time.time()),
     }