mediagoblin/messages.py

   1 # GNU MediaGoblin -- federated, autonomous media hosting
   2 # Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as published by
   6 # the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 from mediagoblin.tools import common
  18
  19 DEBUG = 'debug'
  20 INFO = 'info'
  21 SUCCESS = 'success'
  22 WARNING = 'warning'
  23 ERROR = 'error'
  24
  25 ADD_MESSAGE_TEST = []
  26
  27
  28 def add_message(request, level, text):
  29     messages = request.session.setdefault('messages', [])
  30     messages.append({'level': level, 'text': text})
  31
  32     if common.TESTS_ENABLED:
  33         ADD_MESSAGE_TEST.append(messages)
  34
  35     request.session.save()
  36
  37
  38 def fetch_messages(request, clear_from_session=True):
  39     messages = request.session.get('messages')
  40     if messages and clear_from_session:
  41         # Save that we removed the messages from the session
  42         request.session['messages'] = []
  43         request.session.save()
  44
  45     return messages
  46
  47
  48 def clear_add_message():
  49     global ADD_MESSAGE_TEST
  50     ADD_MESSAGE_TEST = []