From 421f17d5caab376d3f115d1eb3ced923fb383e89 Mon Sep 17 00:00:00 2001
From: Info-IIG <infosupport.infragroupe@ldc.fr>
Date: Wed, 29 Jun 2022 15:08:07 +0200
Subject: [PATCH] Added VM DRBD Status to appsettings

---
 .../migrations/0007_auto_20220629_1213.py     | 29 +++++++++++++++++++
 .../templates/allinstances_index_grouped.html |  6 ++++
 .../allinstances_index_nongrouped.html        |  4 +++
 instances/views.py                            |  5 ++--
 4 files changed, 42 insertions(+), 2 deletions(-)
 create mode 100644 appsettings/migrations/0007_auto_20220629_1213.py

diff --git a/appsettings/migrations/0007_auto_20220629_1213.py b/appsettings/migrations/0007_auto_20220629_1213.py
new file mode 100644
index 0000000..e9f2ed5
--- /dev/null
+++ b/appsettings/migrations/0007_auto_20220629_1213.py
@@ -0,0 +1,29 @@
+# Generated by Django 3.2.12 on 2022-06-29 12:13
+
+from django.db import migrations
+from django.utils.translation import gettext_lazy as _
+
+
+def add_default_settings(apps, schema_editor):
+    setting = apps.get_model("appsettings", "AppSettings")
+    db_alias = schema_editor.connection.alias
+    setting.objects.using(db_alias).bulk_create([
+        setting(31, _("VM DRBD Status"), "VM_DRBD_STATUS", "False", "True,False", _("Show VM DRBD Status")),
+    ])
+
+
+def del_default_settings(apps, schema_editor):
+    setting = apps.get_model("appsettings", "AppSettings")
+    db_alias = schema_editor.connection.alias
+    setting.objects.using(db_alias).filter(key="VM_DRBD_STATUS").delete()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('appsettings', '0006_alter_appsettings_id'),
+    ]
+
+    operations = [
+	    migrations.RunPython(add_default_settings, del_default_settings),
+    ]
diff --git a/instances/templates/allinstances_index_grouped.html b/instances/templates/allinstances_index_grouped.html
index a6c3bc6..bdeda51 100644
--- a/instances/templates/allinstances_index_grouped.html
+++ b/instances/templates/allinstances_index_grouped.html
@@ -7,7 +7,9 @@
             <th scope="col">{% trans "Name" %}<br>{% trans "Description" %}</th>
             <th scope="col" class="d-none d-sm-table-cell">{% trans "User"%}</th>
             <th scope="col">{% trans "Status" %}</th>
+	    {% if app_settings.VM_DRBD_STATUS == 'True' %}
 	    <th scope="col">{% trans "Role/Disk" %}</th>
+	    {% endif %}
             <th scope="col" class="d-none d-sm-table-cell">{% trans "VCPU" %}</th>
             <th scope="col" class="d-none d-sm-table-cell">{% trans "Memory" %}</th>
             <th scope="col" style="width:200px;" data-sortable="false">{% trans "Actions" %} & {% trans "Mem Usage" %}</th>
@@ -28,7 +30,9 @@
                     <td>
                         <span class="text-success">{% trans "Connected" %}</span>
                     </td>
+		    {% if app_settings.VM_DRBD_STATUS == 'True' %}
 		    <td class="d-none d-sm-table-cell"></td>
+		    {% endif %}
                     <td class="d-none d-sm-table-cell text-center">{{ compute.cpu_count }}</td>
                     <td class="d-none d-sm-table-cell text-right">{{ compute.ram_size|filesizeformat }}</td>
                     <td>
@@ -64,9 +68,11 @@
                                 <span class="text-warning">{% trans "Suspended" %}</span>
                             {% endif %}
                         </td>
+			{% if app_settings.VM_DRBD_STATUS == 'True' %}
 			<td>
 				{% if instance.drbd == "Primary/OK" or instance.drbd == "Secondary/OK" %}<span class="text-success">{% else %}<span class="text-danger">{% endif %}{{ instance.drbd }}</span>
 			</td>
+			{% endif %}
                         <td>{{ instance.proxy.instance.info.3 }}</td>
                         <td>{{ instance.cur_memory }} MB</td>
                         <td class="text-nowrap">
diff --git a/instances/templates/allinstances_index_nongrouped.html b/instances/templates/allinstances_index_nongrouped.html
index 114b6b8..a836fa7 100644
--- a/instances/templates/allinstances_index_nongrouped.html
+++ b/instances/templates/allinstances_index_nongrouped.html
@@ -7,7 +7,9 @@
                 <th scope="col">{% trans 'Host' %}<br>{% trans 'User' %}</th>
             {% endif %}
             <th scope="col">{% trans 'Status' %}</th>
+	    {% if app_settings.VM_DRBD_STATUS == 'True' %}
 	    <th scope="col">{% trans 'Role/Disk' %}</th>
+	    {% endif %}
             <th scope="col">{% trans 'VCPU' %}</th>
             <th scope="col">{% trans 'Memory' %}</th>
             <th scope="col" data-sortable="false">{% trans 'Actions' %}</th>
@@ -45,9 +47,11 @@
                         {% if instance.proxy.instance.info.0 == 3 %}<span
                             class="text-warning">{% trans "Suspended" %}</span>{% endif %}
                     </td>
+		    {% if app_settings.VM_DRBD_STATUS == 'True' %}
 		    <td>
 			    {% if instance.drbd == "Primary/OK" or instance.drbd == "Secondary/OK" %}<span class="text-success">{% else %}<span class="text-danger">{% endif %}{{ instance.drbd }}</span>
 		    </td>
+		    {% endif %}
                     <td>{{ instance.proxy.instance.info.3 }}</td>
                     <td>{{ instance.cur_memory }} MB</td>
                     <td class="text-nowrap">
diff --git a/instances/views.py b/instances/views.py
index ff8702c..449fc7b 100644
--- a/instances/views.py
+++ b/instances/views.py
@@ -128,8 +128,9 @@ def instance(request, pk):
     storages_host = sorted(instance.proxy.get_storages(True))
     net_models_host = instance.proxy.get_network_models()
 
-    instance.drbd = drbd_status(request, pk)
-    instance.save()
+    if app_settings.VM_DRBD_STATUS == 'True':
+        instance.drbd = drbd_status(request, pk)
+        instance.save()
 
     return render(request, "instance.html", locals())