Bulk interface update

cyder/base/views.py

@@ -145,6 +145,7 @@ def cy_view(request, template, pk=None, obj_type=None):
     Klass, FormKlass = get_klasses(obj_type)
     obj = get_object_or_404(Klass, pk=pk) if pk else None
     form = None
+    batchInterfaceForm = None
     if request.method == 'POST':
         object_table = None
         page_obj = None
@@ -190,6 +191,11 @@ def cy_view(request, template, pk=None, obj_type=None):
         page_obj = make_paginator(request, do_sort(request, object_list), 50)
         object_table = tablefy(page_obj, request=request)
 
+        if obj_type in ['static_interface', 'dynamic_interface']:
+            from cyder.cydhcp.interface.forms import BatchInterfaceForm
+            batchInterfaceForm = BatchInterfaceForm()
+            batchInterfaceForm.make_usable(request)
+
     if isinstance(form, UsabilityFormMixin):
         form.make_usable(request)
 
@@ -201,6 +207,7 @@ def cy_view(request, template, pk=None, obj_type=None):
         'obj_type': obj_type,
         'pretty_obj_type': Klass.pretty_type,
         'pk': pk,
+        'batchInterfaceForm': batchInterfaceForm,
     })
 
 

cyder/cydhcp/interface/forms.py

@@ -0,0 +1,11 @@
+from django import forms
+
+from cyder.cydhcp.range.models import Range
+from cyder.base.mixins import UsabilityFormMixin
+
+
+class BatchInterfaceForm(forms.Form, UsabilityFormMixin):
+    range_type = forms.ChoiceField(
+        widget=forms.RadioSelect,
+        choices=(('static', 'Static'), ('dynamic', 'Dynamic')))
+    range = forms.ModelChoiceField(queryset=Range.objects.all())

cyder/cydhcp/interface/urls.py

@@ -1,8 +1,11 @@
 from django.conf.urls.defaults import patterns, url
-
 from cyder.cydhcp.interface.views import is_last_interface
+from cyder.cydhcp.interface.views import batch_update, batch_update_get_ranges
 
 urlpatterns = patterns(
     '',
     url(r'^last_interface/', is_last_interface, name='is_last_interface'),
+    url(r'^batch_update/', batch_update, name='batch_update'),
+    url(r'^batch_update_get_ranges/', batch_update_get_ranges,
+        name='batch_update_get_ranges'),
 )

cyder/cydhcp/interface/views.py

@@ -1,10 +1,12 @@
 from django.db.models.loading import get_model
-from django.shortcuts import get_object_or_404
+from django.core.exceptions import ValidationError
+from django.shortcuts import get_object_or_404, redirect
 from django.http import HttpResponse
 
 from cyder.cydhcp.interface.dynamic_intr.models import DynamicInterface
 from cyder.cydhcp.interface.static_intr.models import StaticInterface
-
+from cyder.cydhcp.range.range_usage import range_usage
+from cyder.cydhcp.network.utils import get_ranges
 import json
 
 
@@ -19,3 +21,127 @@ def is_last_interface(request):
         last_interface = True
 
     return HttpResponse(json.dumps({'last_interface': last_interface}))
+
+
+def batch_update_get_ranges(request):
+    if not request.POST:
+        return redirect(request.META.get('HTTP_REFERER', ''))
+    ranges = []
+    ctnr = request.session['ctnr']
+    range_type = request.POST.get('range_type', None)
+    if range_type:
+        range_type = [range_type[:2]]
+        range_qs = get_ranges([], ctnr, range_type, all_ranges=True)
+    else:
+        range_qs = get_ranges([], ctnr, all_ranges=True)
+
+    for n, rng in enumerate(range_qs):
+        ranges.append([rng.id, rng.__str__()])
+
+    return HttpResponse(json.dumps({'ranges': ranges}))
+
+
+def batch_update_same_type(rng, ip, intr, intr_type):
+    if 'static' in intr_type:
+        intr.ip_str = str(ip)
+        intr.ip_type = rng.ip_type
+    else:
+        intr.range = rng
+
+    return intr
+
+
+
+def batch_update_different_type(rng, ip, intr, intr_type):
+    new_intr_type = 'staticdynamicinterface'.replace(
+        intr_type, '')
+    NewInterface = get_model('cyder', new_intr_type)
+    kwargs = {'mac': intr.mac, 'dhcp_enabled': intr.dhcp_enabled,
+              'ctnr': intr.ctnr, 'workgroup': intr.workgroup,
+              'system': intr.system}
+    if 'static' in new_intr_type:
+        label = '{0}-{1}'.format(intr.system.name, intr.mac)
+        kwargs.update({'ip_str': str(ip), 'ip_type': rng.ip_type,
+                       'dns_enabled': True, 'domain': rng.domain,
+                       'label': label})
+    else:
+        kwargs.update({'range': rng})
+
+    return NewInterface(**kwargs)
+
+
+def batch_update(request):
+    if not request.POST:
+        return redirect(request.META.get('HTTP_REFERER', ''))
+
+    for field in ['range', 'range_type', 'interfaces']:
+        if not request.POST.get(field, None):
+            field = ' '.join(field.split('_'))
+            return HttpResponse(json.dumps({
+                'error': 'Please select {0}'.format(field)}))
+
+    Range = get_model('cyder', 'range')
+    rng_qs = Range.objects.filter(id=request.POST['range'])
+    if rng_qs.exists():
+        rng = rng_qs.get()
+    else:
+        return HttpResponse(json.dumps({
+            'error': 'That range does not exist'}))
+
+    success = True
+    ip = None
+    start_lower = None
+    intr_type = ('').join(request.POST['interface_type'].split()[:-1])
+    new_intrs = []
+    same_type = intr_type[:2] == rng.range_type
+    Interface = get_model('cyder', intr_type + 'interface')
+    interfaces = request.POST['interfaces'].split(',')
+    interface_qs = Interface.objects.filter(pk__in=interfaces)
+
+    if rng.range_type == 'st':
+        used_ips, _ = range_usage(
+            rng.start_lower, rng.end_lower, rng.ip_type)
+        if (len(interfaces) >
+                (((rng.end_lower - rng.start_lower) + 1) - len(used_ips))):
+            return HttpResponse(json.dumps({
+                'error': 'Range does not have enough space for selected '
+                'interfaces'}))
+
+    for intr in interface_qs:
+        if rng.range_type == 'st':
+            if ip:
+                start_lower = ip._ip + 1
+
+            ip = rng.get_next_ip(start_lower=start_lower)
+
+        if same_type:
+            new_intr = batch_update_same_type(rng, ip, intr, intr_type)
+
+        else:
+            new_intr = batch_update_different_type(rng, ip, intr, intr_type)
+
+        new_intrs.append(new_intr)
+
+    if same_type:
+        for intr in new_intrs:
+            intr.save()
+    else:
+        for intr in new_intrs:
+            try:
+                intr.save()
+            except ValidationError as e:
+                success = False
+                break
+
+        if success:
+            interface_qs.delete()
+            rng.save()
+        else:
+            for intr in new_intrs:
+                intr.delete()
+
+            rng.save()
+            return HttpResponse(json.dumps({
+                'error': 'Batch update was unsuccessful: {0}'.format(str(e))}))
+
+    return HttpResponse(json.dumps({'success': True}))

cyder/cydhcp/range/models.py

@@ -371,19 +371,22 @@ def get_usage(self):
 
         return usage
 
-    def get_next_ip(self):
+    def get_next_ip(self, start_lower=None):
         """Finds the most appropriate IP address within a range. If it can't
         find an IP it returns None. If it finds an IP it returns an IPv4Address
         object.
 
             :returns: ipaddr.IPv4Address
         """
-        if self.network.ip_type != '4':
+        if not start_lower:
+            start_lower = self.start_lower
+
+        if self.network and self.network.ip_type != '4':
             return None
         elif self.ip_type != '4':
             return None
 
-        ip = find_free_ip(self.start_lower, self.end_lower, ip_type='4')
+        ip = find_free_ip(start_lower, self.end_lower, ip_type='4')
         if ip:
             return ip
         else:

cyder/templates/base/list.html

@@ -41,6 +41,14 @@
           </a>
         </div>
       </div>
+      <div class="batch-update">
+        {% if pretty_obj_type in ["static interface", "dynamic interface"] %}
+          <form id="enable-batch-update">
+            <input type="checkbox" name="batch" id="enable-batch-check">
+            <label for="enable-batch-check">Enter batch update mode</label>
+          </form>
+        {% endif %}
+      </div>
     {% endif %}
   {% endif %}
 

cyder/templates/cydhcp/cydhcp_view.html

@@ -28,6 +28,9 @@
         Create {{ pretty_obj_type }}
       </a>
     {% endif %}
+    {% if batchInterfaceForm %}
+    <a id="batch_update_btn" style="display:none" class="btn submit c">Update Interfaces</a>
+    {% endif %}
   {% endif %}
 {% endblock %}
 
@@ -55,6 +58,21 @@ <h2 id="form-title">{{ 'Updating ' if obj else 'Creating ' + pretty_obj_type }}<
       </form>
     </div>
   {% endif %}
+  {% if batchInterfaceForm %}
+    <div class="hidden" id="batch-form">
+      <form class="form" method="POST">
+        <h2 id="form-title">Batch update for selected interfaces</h2>
+        <div class="inner-form" id="batch-hidden-inner-form">
+          {{ batchInterfaceForm.as_p() }}
+        </div>
+        <div class="form-btns">
+          <hr>
+          <a class="btn c" id="batch-submit">Update</a>
+          <a class="btn cancel c" id="batch-cancel">Cancel</a>
+        </div>
+      </form>
+    </div>
+  {% endif %}
   {% if obj_type == 'range' %}
     {{ js('rangeform') }}
   {% endif %}

media/css/tables.scss

@@ -34,6 +34,7 @@ table.table {
         top: 2px;
     }
     td, th {
+        vertical-align: middle;
         padding: 5px 0;
     }
     thead {
@@ -123,6 +124,20 @@ table {
     }
 }
 .editable-grid {
+    clear: both;
+    float: right;
+    margin-right: 17px;
+}
+.batch-update {
+    clear: both;
     float: right;
-    margin-right: 1.5%;
+    margin-right: 10px;
+}
+.batch-update-checkbox {
+    width: 15px;
+    height: 15px;
+}
+#selectAll {
+    margin-left: 5px;
+    vertical-align: middle;
 }