Cache lookups.

Fixes philipn#3.

Author: philipn

block_ip/middleware.py

@@ -1,5 +1,6 @@
 from django.http import HttpResponseForbidden
 from django.conf import settings
+from django.core.cache import cache
 
 from models import BlockIP
 
@@ -22,7 +23,11 @@ def process_request(self, request):
         ip = get_ip(request)
         # TODO: Look into something more optimized for large numbers
         # of blocks. https://github.com/jimfunk/django-postgresql-netfields
-        deny_ips = [i.get_network() for i in BlockIP.objects.all()]
+        block_ips = cache.get('blockip:list')
+        if block_ips is None:
+            block_ips = BlockIP.objects.all()
+            cache.set('blockip:list', block_ips)
+        deny_ips = [i.get_network() for i in block_ips]
 
         for net in deny_ips:
             if ip in net:

block_ip/models.py

@@ -2,6 +2,8 @@
 
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
+from django.core.cache import cache
+from django.db.models.signals import post_save, post_delete
 
 
 class BlockIP(models.Model):
@@ -17,3 +19,11 @@ def get_network(self):
     class Meta:
         verbose_name = _('IPs & masks to ban')
         verbose_name_plural = _('IPs & masks to ban')
+
+
+def _clear_cache(sender, instance, **kwargs):
+    cache.set('blockip:list', BlockIP.objects.all())
+
+
+post_save.connect(_clear_cache, sender=BlockIP)
+post_delete.connect(_clear_cache, sender=BlockIP)