From 79abcd460bdde7e6de4e441b615fd19bda04f566 Mon Sep 17 00:00:00 2001
From: catborise <catborise@yahoo.com>
Date: Fri, 15 Nov 2019 11:29:16 +0300
Subject: [PATCH] Add Qos functions and Edit network with xml function

---
 networks/templates/add_inbound_qos.html  | 41 +++++++++++++++++
 networks/templates/add_outbound_qos.html | 40 +++++++++++++++++
 vrtManager/network.py                    | 57 ++++++++++++++++++++++++
 3 files changed, 138 insertions(+)
 create mode 100644 networks/templates/add_inbound_qos.html
 create mode 100644 networks/templates/add_outbound_qos.html

diff --git a/networks/templates/add_inbound_qos.html b/networks/templates/add_inbound_qos.html
new file mode 100644
index 0000000..135b1c8
--- /dev/null
+++ b/networks/templates/add_inbound_qos.html
@@ -0,0 +1,41 @@
+{% load i18n %}
+{% if request.user.is_superuser %}
+    <a href="#AddInboundQos" type="button" class="btn btn-success pull-right" data-toggle="modal">
+        <span class="glyphicon glyphicon-plus" aria-hidden="true"></span>
+    </a>
+
+    <!-- Modal pool -->
+    <div class="modal fade" id="AddInboundQos" tabindex="-1" role="dialog" aria-labelledby="AddInboundQosLabel" aria-hidden="true">
+        <div class="modal-dialog">
+            <div class="modal-content">
+                <div class="modal-header">
+                    <button type="button" class="close" data-dismiss="modal" aria-hidden="true">&times;</button>
+                    <h4 class="modal-title">{% trans "Add Inbound Qos for Network" %}</h4>
+                </div>
+                <form class="form-horizontal" method="post" action="" role="form" novalidate>{% csrf_token %}
+                    <div class="modal-body">
+                        <div class="form-group col-sm-4 ">
+                            <label for="qos_inbound_av">{% trans "Average" %}:</label>
+                            <input id="qos_inbound_av" class="form-control" name="qos_inbound_average" value="{{ att.average }}"/>
+                        </div>
+                        <div class="form-group col-sm-4">
+                            <label for="qos_inbound_peak">{% trans "Peak" %}:</label>
+                            <input id="qos_inbound_peak" class="form-control" name="qos_inbound_peak" value="{{ att.peak }}"/>
+                        </div>
+                        <div class="form-group col-sm-4">
+                            <label for="qos_inbound_burst">{% trans "Burst" %}:</label>
+                            <input id="qos_inbound_burst" class="form-control" name="qos_inbound_burst" value="{{ att.burst }}"/></p>
+                        </div>
+                        
+                    </div> <!-- /.modal-content -->
+                    <div class="modal-footer">
+                        <input name="qos_direction" value="inbound" hidden/>
+                        <div class="col-sm-6">
+                            <button class="btn btn-primary btn-block" name="set_qos">{% trans 'Save' %}</button>
+                        </div>
+                    </div>
+                </form>
+            </div>
+        </div> <!-- /.modal-dialog -->
+    </div> <!-- /.modal -->
+{% endif %}
\ No newline at end of file
diff --git a/networks/templates/add_outbound_qos.html b/networks/templates/add_outbound_qos.html
new file mode 100644
index 0000000..4adf19d
--- /dev/null
+++ b/networks/templates/add_outbound_qos.html
@@ -0,0 +1,40 @@
+{% load i18n %}
+{% if request.user.is_superuser %}
+    <a href="#AddInboundQos" type="button" class="btn btn-success pull-right" data-toggle="modal">
+        <span class="glyphicon glyphicon-plus" aria-hidden="true"></span>
+    </a>
+
+    <!-- Modal pool -->
+    <div class="modal fade" id="AddInboundQos" tabindex="-1" role="dialog" aria-labelledby="AddInboundQosLabel" aria-hidden="true">
+        <div class="modal-dialog">
+            <div class="modal-content">
+                <div class="modal-header">
+                    <button type="button" class="close" data-dismiss="modal" aria-hidden="true">&times;</button>
+                    <h4 class="modal-title">{% trans "Add Inbound Qos for Network" %}</h4>
+                </div>
+                <form class="form-horizontal" method="post" action="" role="form" novalidate>{% csrf_token %}
+                    <div class="modal-body">
+                        <div class="form-group col-sm-4 ">
+                            <label for="qos_inbound_av">{% trans "Average" %}:</label>
+                            <input id="qos_inbound_av" class="form-control" name="qos_inbound_average" value="{{ att.average }}"/>
+                        </div>
+                        <div class="form-group col-sm-4">
+                            <label for="qos_inbound_peak">{% trans "Peak" %}:</label>
+                            <input id="qos_inbound_peak" class="form-control" name="qos_inbound_peak" value="{{ att.peak }}"/>
+                        </div>
+                        <div class="form-group col-sm-4">
+                            <label for="qos_inbound_burst">{% trans "Burst" %}:</label>
+                            <input id="qos_inbound_burst" class="form-control" name="qos_inbound_burst" value="{{ att.burst }}"/></p>
+                        </div>
+                    </div> <!-- /.modal-content -->
+                    <div class="modal-footer">
+                        <input name="qos_direction" value="inbound" hidden/>
+                        <div class="col-sm-6">
+                            <button class="btn btn-primary btn-block" name="set_qos">{% trans 'Save' %}</button>
+                        </div>
+                    </div>
+                </form>
+            </div>
+        </div> <!-- /.modal-dialog -->
+    </div> <!-- /.modal -->
+{% endif %}
\ No newline at end of file
diff --git a/vrtManager/network.py b/vrtManager/network.py
index ebe54af..87333da 100644
--- a/vrtManager/network.py
+++ b/vrtManager/network.py
@@ -290,3 +290,60 @@ class wvmNetwork(wvmConnect):
                             parent_index,
                             VIR_NETWORK_UPDATE_AFFECT_LIVE | VIR_NETWORK_UPDATE_AFFECT_CONFIG)
 
+    def get_qos(self):
+        qos_values = dict()
+        tree = etree.fromstring(self._XMLDesc(0))
+        qos = tree.xpath("/network/bandwidth")
+        if qos:
+            qos = qos[0]
+
+            in_qos = qos.find('inbound')
+            if in_qos is not None:
+                in_av = in_qos.get('average')
+                in_peak = in_qos.get('peak')
+                in_burst = in_qos.get('burst')
+                qos_values['inbound'] = {'average': in_av, 'peak': in_peak, 'burst': in_burst}
+
+            out_qos = qos.find('outbound')
+            if out_qos is not None:
+                out_av = out_qos.get('average')
+                out_peak = out_qos.get('peak')
+                out_burst = out_qos.get('burst')
+                qos_values['outbound'] = {'average': out_av, 'peak': out_peak, 'burst': out_burst}
+        return qos_values
+
+    def set_qos(self, direction, average, peak, burst):
+        if direction == "inbound":
+            xml = "<inbound average='{}' peak='{}' burst='{}'/>".format(average, peak, burst)
+        elif direction == "outbound":
+            xml = "<outbound average='{}' peak='{}' burst='{}'/>".format(average, peak, burst)
+        else:
+            raise Exception('Direction must be inbound or outbound')
+
+        tree = etree.fromstring(self._XMLDesc(0))
+
+        band = tree.xpath("/network/bandwidth")
+        if len(band) == 0:
+            xml = "<bandwidth>" + xml + "</bandwidth>"
+            tree.append(etree.fromstring(xml))
+        else:
+            direct = band[0].find(direction)
+            if direct is not None:
+                parent = direct.getparent()
+                parent.remove(direct)
+                parent.append(etree.fromstring(xml))
+            else:
+                band[0].append(etree.fromstring(xml))
+        new_xml = etree.tostring(tree)
+        self.wvm.networkDefineXML(new_xml)
+
+    def unset_qos(self, direction):
+        tree = etree.fromstring(self._XMLDesc(0))
+        for direct in tree.xpath("/network/bandwidth/{}".format(direction)):
+            parent = direct.getparent()
+            parent.remove(direct)
+
+        self.wvm.networkDefineXML(etree.tostring(tree))
+
+    def edit_network(self, new_xml):
+        self.wvm.networkDefineXML(new_xml)