# HG changeset patch # User Peter Sanchez # Date 1368547544 25200 # Tue May 14 09:05:44 2013 -0700 # Node ID deb84398e0ba86b4d96eaefdc3a3e90ef4a50097 # Parent ae31f4022020042658ef862fa4358db2570aef09 Adding notification to base class. diff --git a/awesomebox/base.py b/awesomebox/base.py --- a/awesomebox/base.py +++ b/awesomebox/base.py @@ -35,18 +35,28 @@ ) return result + def notify_user(self, request, result): + ''' Default method to notify the user of success. Override + to customize + ''' + if 'django.contrib.messages' in settings.INSTALLED_APPS: + from django.contrib import messages + level = messages.SUCCESS if result.success else messages.ERROR + messages.add_message(request, level, result.message) + def cmd_echo(self, data): ''' Example function. Returns back the data passed in. ''' return Result(True, data) +class NullAwesomeBox(BaseAwesomebox): + pass + + def get_awesomebox_class(): ''' Return the class to use for Awesomebox dispatching. ''' - class NullAwesomeBox(BaseAwesomebox): - pass - cls_path = getattr(settings, 'AWESOMEBOX_CLASS', None) cls = NullAwesomeBox if cls_path is None else import_module(cls_path) if not issubclass(cls, BaseAwesomebox): diff --git a/awesomebox/views.py b/awesomebox/views.py --- a/awesomebox/views.py +++ b/awesomebox/views.py @@ -33,7 +33,8 @@ json.dumps(result), mimetype='application/json', ) - # TODO Add message here for user session + + abox.notify_user(request, result) return redirect(self.success_url) return render(request, self.template_name, {'form': form})