Changeset 73db3090c34e51a2508cdc048d3ca2f4cf3da288 for pxe/views.py

Timestamp:

05/12/10 14:10:28 (2 years ago)

Author:

Frederic Lepied <frederic.lepied@…>

Children:

36e3187115d175e2667c28737fdaccbd1fed1723

Parents:

a37e6686ba85de9cf2e143a6485dd63bceff9404

git-committer:

Frederic Lepied <frederic.lepied@…> (05/12/10 14:10:28)

Message:

allow to use IP addresses and sub-adresses instead of mac addresses

Files:

• pxe/views.py (modified) (2 diffs)

pxe/views.py

@@ -13 +13 @@
 from pxe.models import *
 
+def get_system(request, mac):
+    try:
+        return System.objects.get(macaddress__mac=simplify_mac(mac))
+    except System.DoesNotExist:
+        pass
+
+    addr = request.META['REMOTE_ADDR']
+    l = map(lambda x: '%02x' % int(x), addr.split('.'))
+    for i in range(len(l), 1, -1):
+        try:
+            return System.objects.get(macaddress__mac=''.join(l[0:i]))
+        except System.DoesNotExist:
+            pass
+    raise Http404
+
 def localboot1(request):
     return localboot(request, get_mac(request))
     
 def localboot(request, mac):
-    system = get_object_or_404(System, macaddress__mac=simplify_mac(mac))
+    system = get_system(request, mac)
     set_next_boot(system, settings.PXE_LOCAL)
     return HttpResponse("Next boot set to local", mimetype="text/plain")
@@ -25 +40 @@
     
 def profile(request, mac):
-    system = get_object_or_404(System, macaddress__mac=simplify_mac(mac))
+    system = get_system(request, mac)
     log = Log.objects.filter(system=system).order_by('-date')[0]
     return HttpResponse(log.boot_name.name, mimetype="text/plain")