@@ -2,20 +2,23 @@ from django.conf import settings from userskins.models import SkinPreference import django + class UserskinsMiddleware(object): def __init__(self): never_use_database = getattr(settings,"USERSKINS_NEVER_ACCESS_DATABASE", False) if never_use_database: raise django.core.exceptions.MiddlewareNotUsed + else: + self.default = settings.USERSKINS_DEFAULT + def process_response(self, request, response): - print "wee" if not request.COOKIES.has_key("userskins"): - skin = settings.USERSKINS_DEFAULT + skin = self.default if request.user.is_authenticated(): try: skin = SkinPreference.objects.get(user=request.user).skin - except SkinPreference.ObjectDoesNotExist: + except SkinPreference.DoesNotExist: pass response.set_cookie("userskins", skin) return response userskins/middleware.py d1e98a931aaa8c60b77a493bc4a70ae4da418650 Will Larson lethain@gmail.com http://github.com/lethain/django-userskins/commit/5d1b60a3fe8f2c2564e94fd646f4e112f5803288 5d1b60a3fe8f2c2564e94fd646f4e112f5803288 2008-10-29T18:38:11-07:00 2008-10-29T18:38:11-07:00 Fixed typo of ObjectDoesNotExist versus DoesNotExist for error thrown when user doesn't have a SkinPreference object. e85cf6506f6f08d7497d5df501d481d9267344c8 Will Larson lethain@gmail.com