Changeset 10605 for main/waeup.cas

Timestamp:

10 Sep 2013, 14:24:31 (11 years ago)

Author:

uli

Message:

Add a smarter message box handling.

File:

• main/waeup.cas/trunk/waeup/cas/server.py (modified) (4 diffs)

main/waeup.cas/trunk/waeup/cas/server.py

@@ -4 +4 @@
 import os
 import random
+import re
 import time
 try:
@@ -28 +29 @@
             'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
             '01234567789-')
+
+#: A regular expression that matches a div tag around a MSG_TEXT
+RE_MSG_TAG = re.compile('\<div id="msg".*MSG_TEXT[^<]*</div>', re.DOTALL)
 
 
@@ -233 +237 @@
     # show logged-in screen
     html = get_template('login_successful.html')
-    html = html.replace('MSG_TEXT', msg)
+    html = set_message(msg, html)
     resp = Response(html)
     if not sso:
         resp = set_session_cookie(db, resp)
     return resp
+
+
+def set_message(msg, html):
+    """Insert a message box in html template.
+
+    If the message is empty, not only the string `MSG_TEXT` is removed
+    from `html`, but also any encapsulating ``<div>`` tag with id
+    ``msg``.
+
+    This makes it possible to give message boxes certain additional
+    styles that will not show up if there is no message to display.
+    """
+    if not msg:
+        if not '<div id="msg"' in html:
+            return html.replace('MSG_TEXT', '')
+        return RE_MSG_TAG.sub('', html)
+    return html.replace('MSG_TEXT', msg)
 
 
@@ -325 +346 @@
         html = html.replace('LT_VALUE', lt.ticket)
         html = html.replace('SERVICE_FIELD_VALUE', service_field)
-        html = html.replace('MSG_TEXT', msg)
+        html = set_message(msg, html)
         return Response(html)