ogcp/forms/action_forms.py

@@ -73,8 +73,7 @@ class SetupForm(FlaskForm):
     ips = HiddenField()
     disk = SelectField(label='Disk', validate_choice=False)
     disk_type = SelectField(label=_l('Type'),
-                            choices=[('EMPTY', 'EMPTY'),
-                                     ('MSDOS', 'MBR'),
+                            choices=[('MSDOS', 'MBR'),
                                      ('GPT', 'GPT')])
     partitions = FieldList(FormField(PartitionForm),
                            min_entries=1,

ogcp/templates/actions/add_center.html

@@ -14,7 +14,7 @@
                   action=url_for('action_center_add'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/add_room.html

@@ -14,7 +14,7 @@
                   action=url_for('action_room_add'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/add_server.html

@@ -12,7 +12,6 @@
 {{ wtf.quick_form(form,
                   action=url_for('server_add_post'),
                   method='post',
-                  button_map={'submit_btn':'primary'},
-                  extra_classes="m-5") }}
+                  button_map={'submit_btn':'primary'}) }}
 
 {% endblock %}

ogcp/templates/actions/center_details.html

@@ -13,6 +13,6 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/center_update.html

@@ -13,6 +13,6 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/client_add.html

@@ -14,6 +14,6 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/client_move.html

@@ -20,6 +20,6 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/delete_center.html

@@ -45,7 +45,7 @@
                   action=url_for('action_center_delete'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/delete_client.html

@@ -21,7 +21,7 @@
                   action=url_for('action_client_delete'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/delete_image.html

@@ -17,7 +17,7 @@
                   action=url_for('action_image_delete'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/delete_repo.html

@@ -11,7 +11,7 @@
 
 <h2 class="mx-5 subhead-heading">{{_('Delete repo')}}</h2>
 
-<form class="form m-5" method="POST">
+<form class="form mx-5" method="POST">
     {{ form.hidden_tag() }}
 
     {{ form.server() }}

ogcp/templates/actions/delete_room.html

@@ -48,7 +48,7 @@
                   action=url_for('action_room_delete'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/delete_server.html

@@ -16,7 +16,7 @@
                   action=url_for('server_delete_post'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/folder_add.html

@@ -13,7 +13,7 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/folder_delete.html

@@ -45,6 +45,6 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'danger'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/folder_update.html

@@ -13,7 +13,7 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/hardware.html

@@ -16,7 +16,7 @@
                   action=url_for('action_hardware'),
                   method='post',
                   button_map={'refresh': 'primary'},
-                  extra_classes='m-5')}}
+                  extra_classes='m-2')}}
 
 <table class="table table-striped">
   <thead class="thead-dark">

ogcp/templates/actions/image_config.html

@@ -8,7 +8,7 @@
 
 <h2 class="mx-5 subhead-heading">{{_('Update image')}} {{ form.name.data }}</h2>
 
-<form  class="form m-5" method="POST" action="{{ url_for('action_image_config') }}">
+<form  class="form mx-5" method="POST" action="{{ url_for('action_image_config') }}">
     {{ form.hidden_tag() }}
 
     {{ form.image_id() }}

ogcp/templates/actions/image_create.html

@@ -12,13 +12,10 @@
 
 {{ macros.cmd_selected_clients(selected_clients) }}
 
-{% set partition_field_id = 'os' %}
-{% include 'partition_warning.html' %}
-
 {{ wtf.quick_form(form,
                   action=url_for('action_image_create'),
                   method='post',
                   button_map={'create': 'primary'},
-                  extra_classes='m-5') }}
+                  extra_classes='mx-5') }}
 
 {% endblock %}

ogcp/templates/actions/image_details.html

@@ -8,7 +8,7 @@
 
 <h2 class="mx-5 subhead-heading">{{_('Image details')}}</h2>
 
-<div class="container m-5">
+<div class="container mx-5">
     <form class="form" method="POST">
         {{ form.hidden_tag() }}
 

ogcp/templates/actions/image_restore.html

@@ -17,9 +17,6 @@
 
 {{ macros.cmd_selected_clients(selected_clients) }}
 
-{% set partition_field_id = 'partition' %}
-{% include 'partition_warning.html' %}
-
 {{ wtf.quick_form(form,
                   action=url_for('action_image_restore'),
                   method='post',

ogcp/templates/actions/image_update.html

@@ -13,9 +13,6 @@
 
 {{ macros.cmd_selected_clients(selected_clients) }}
 
-{% set partition_field_id = 'os' %}
-{% include 'partition_warning.html' %}
-
 <form  class="form mx-5" method="POST" action="{{ url_for('action_image_update') }}">
     {{ form.hidden_tag() }}
 

ogcp/templates/actions/import_clients.html

@@ -14,6 +14,6 @@
                   action=url_for('action_clients_import_post'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/poweroff.html

@@ -21,7 +21,7 @@
                   action=url_for('action_poweroff'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/reboot.html

@@ -21,7 +21,7 @@
                   action=url_for('action_reboot'),
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/repo_details.html

@@ -11,7 +11,7 @@
 
 <h2 class="mx-5 subhead-heading">{{_('Repo details')}}</h2>
 
-<form class="form m-5" method="POST">
+<form class="form mx-5" method="POST">
     {{ form.hidden_tag() }}
 
     {{ form.server() }}

ogcp/templates/actions/room_details.html

@@ -13,6 +13,6 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/room_update.html

@@ -13,7 +13,7 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}
 

ogcp/templates/actions/script_run.html

@@ -20,6 +20,6 @@
 {{ wtf.quick_form(form,
                   method='post',
                   button_map={'submit': 'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/actions/server_update.html

@@ -9,7 +9,7 @@
 
 <h2 class="mx-5 subhead-heading">{{_('Update server')}}</h2>
 
-<form class="form m-5" method="POST">
+<form class="form mx-5" method="POST">
     {{ form.hidden_tag() }}
 
     {{ form.server_addr() }}

ogcp/templates/actions/session.html

@@ -18,13 +18,13 @@
 
 {% if os_groups|length > 0 %}
 
-<p class="mx-5">
+<p>
 {% if os_groups|length > 1 %}
 The selected clients have different installed OS:
 {% endif %}
 </p>
 
-<form class="form-inline m-5" method="POST" id="sessionForm">
+<form class="form-inline" method="POST" id="sessionForm">
     <table class="table table-hover">
         <thead class="thead-light">
             <tr>

ogcp/templates/actions/software.html

@@ -14,6 +14,6 @@
                   action=url_for('action_software'),
                   method='post',
                   button_map={'view': 'primary', 'update': 'primary'},
-                  extra_classes="m-5")}}
+                  extra_classes="mx-5")}}
 
 {% endblock %}

ogcp/templates/auth/delete_user.html

@@ -13,6 +13,6 @@
                   action=url_for('user_delete_post'),
                   method='post',
                   button_map={'submit_btn':'primary'},
-                  extra_classes="m-5") }}
+                  extra_classes="mx-5") }}
 
 {% endblock %}

ogcp/templates/disk_inspector.html

@@ -1,6 +1,6 @@
 {% if selected_disk is defined and setup_data is defined %}
 
-<form class="form-inline m-5" method="POST" id="setupForm">
+<form class="form-inline mx-5" method="POST" id="setupForm">
     {{ disk_form.hidden_tag() }}
     {{ disk_form.ips() }}
 

ogcp/templates/partition_warning.html

@@ -1,30 +0,0 @@
-<div class="mx-5" id="partition-warning" style="display:none; color:red; font-weight: bold;">
-    {{ _('Warning: You have selected a partition from a disk other than 1. This will be considered for storage only.') }}
-</div>
-
-<script>
-function checkPartition() {
-    var partitionSelect = document.getElementById('{{ partition_field_id }}');
-    var selectedValue = partitionSelect.value;
-    var warningDiv = document.getElementById('partition-warning');
-
-    // Extract the disk_id
-    var diskId = parseInt(selectedValue.split(' ')[0]);
-
-    if (diskId !== 1) {
-        warningDiv.style.display = 'block';
-    } else {
-        warningDiv.style.display = 'none';
-    }
-}
-
-document.addEventListener("DOMContentLoaded", function () {
-    var partitionSelect = document.getElementById('{{ partition_field_id }}');
-    if (partitionSelect) {
-        partitionSelect.addEventListener('change', checkPartition);
-        checkPartition();
-    } else {
-        console.error("Element with ID '{{ partition_field_id }}' not found.");
-    }
-});
-</script>

ogcp/views.py

@@ -176,6 +176,9 @@ def client_setup_add_image_names(server, setup_data):
 def get_client_setup(ip):
     server = get_server_from_clients([ip])
 
+    payload = {'clients': [ip]}
+    r = server.post('/refresh', payload)
+
     payload = {'client': [ip]}
     r = server.get('/client/setup', payload)
 
@@ -183,7 +186,7 @@ def get_client_setup(ip):
     res = {}
     for partition in db_partitions:
         if partition['partition'] == 0:
-            partition['code'] = PART_SCHEME_CODES.get(partition['code'], 'EMPTY')
+            partition['code'] = PART_SCHEME_CODES.get(partition['code'], 'MSDOS')
         else:
             partition['code'] = PART_TYPE_CODES.get(partition['code'], 'EMPTY')
 
@@ -774,6 +777,10 @@ def action_setup_show():
 
     setup_data = get_client_setup(base_client)
 
+    if not setup_data:
+        flash(_('Partition information is not available. Boot client in ogLive mode to obtain it'), category='error')
+        return redirect(url_for('commands'))
+
     selected_disk = 1
     common_disk_data = get_common_disk_data(ips)
 
@@ -790,11 +797,8 @@ def action_setup_show():
 
     form.disk.choices = [(disk, disk) for disk in setup_data]
     form.disk.data = selected_disk
-
-    for disk in setup_data:
-        if setup_data[disk][0]['code'] == 'EMPTY':
-            setup_data[disk][0]['code'] = 'GPT'
-    form.disk_type.data = setup_data[selected_disk][0]['code']
+    # If partition table is empty, set MSDOS
+    form.disk_type.data = setup_data[selected_disk][0]['code'] or 1
 
     scopes, _clients = get_scopes(ips)
     return render_template('actions/setup.html',
@@ -1080,6 +1084,8 @@ def action_image_restore():
                 part_id = partition['partition']
                 if part_id == 0:  # This is the disk data, not a partition.
                     continue
+                if disk_id != 1:
+                    continue
                 part_code = partition['code']
                 part_type = PART_TYPE_CODES.get(int(part_code), 'UNKNOWN')
 
@@ -1108,6 +1114,10 @@ def action_image_restore():
 
         reference_patitioning = part_collection.get_partition_setup(0)
 
+        if not reference_patitioning:
+            flash(_(f'No valid partition found'), category='error')
+            return redirect(url_for('commands'))
+
         for disk_id, part_id, part_type, fs_type, part_size in reference_patitioning:
             form.partition.choices.append(
                 (f"{disk_id} {part_id} {part_size} {has_cache}",
@@ -1181,28 +1191,17 @@ def action_software():
         server = get_server_from_clients(ips)
         r = server.get('/client/setup', payload={'client': list(ips)})
 
-        invalid_part_types = get_invalid_image_partition_types()
-
-        for part in r.json()['partitions']:
-            part_id = part['partition']
-            if part_id == 0:
-                continue
-
-            part_type = PART_TYPE_CODES.get(int(part.get('code')), 'UNKNOWN')
-            if  part_type in invalid_part_types:
-                    continue
+        if not r.json()['partitions']:
+            flash(_('Software inventory is not available. Boot client in ogLive mode to obtain it'), category='error')
+            return redirect(url_for('commands'))
 
+        for part in r.json()['partitions'][1:]:
             form.os.choices.append(
                 (f"{part.get('disk')} {part.get('partition')}",
                  f"Disk {part.get('disk')} | Partition {part.get('partition')} "
                  f"| {PART_TYPE_CODES.get(part.get('code'), 'UNKNOWN')} "
                  f"{FS_CODES.get(part.get('filesystem'), 'UNKNOWN')}")
             )
-
-        if not form.os.choices:
-            flash(_(f'No valid partition available'), category='error')
-            return redirect(url_for('commands'))
-
     return render_template('actions/software.html', form=form, scopes=scopes)
 
 @app.route('/action/session', methods=['GET', 'POST'])
@@ -2080,7 +2079,7 @@ def action_run_script():
         for ip in ips:
             r = server.get('/client/setup', payload={'client': [ip]})
 
-            partitions = [p for p in r.json()['partitions'] if p['partition'] != 0]
+            partitions = r.json()['partitions'][1:]
             if not reference_patitioning:
                 reference_patitioning = partitions
             elif reference_patitioning != partitions:
@@ -2373,14 +2372,13 @@ def action_image_create():
 
         invalid_part_types = get_invalid_image_partition_types()
 
-        for part in r.json()['partitions']:
+        for part in r.json()['partitions'][1:]:
             part_type = PART_TYPE_CODES.get(int(part.get('code')), 'UNKNOWN')
 
-            part_id = part['partition']
-            if part_id == 0:  # This is the disk data, not a partition.
+            if part_type in invalid_part_types:
                 continue
 
-            if part_type in invalid_part_types:
+            if part.get('disk') != 1:
                 continue
 
             form.os.choices.append(
@@ -2389,11 +2387,6 @@ def action_image_create():
                  f"| {PART_TYPE_CODES.get(part.get('code'), 'UNKNOWN')} "
                  f"{FS_CODES.get(part.get('filesystem'), 'UNKNOWN')}")
             )
-
-        if not form.os.choices:
-            flash(_(f'No valid partition available'), category='error')
-            return redirect(url_for('commands'))
-
         r = server.get('/client/info', payload={'client': list(ips)})
 
         client_repo_id = r.json()['repo_id']
@@ -2474,17 +2467,19 @@ def action_image_update():
     invalid_part_types = get_invalid_image_partition_types()
 
     part_content = {}
-    for part in r.json()['partitions']:
+    for part in r.json()['partitions'][1:]:
         part_type = PART_TYPE_CODES.get(int(part.get('code')), 'UNKNOWN')
 
-        part_id = part['partition']
-        if part_id == 0:  # This is the disk data, not a partition.
-            continue
-
         if part_type in invalid_part_types:
             continue
 
+        if part.get('disk') != 1:
+                continue
+
         partition_value = f"{part.get('disk')} {part.get('partition')} {part.get('code')}"
+        partition_text = f"Disk {part.get('disk')} | Partition {part.get('partition')} "
+        f"| {PART_TYPE_CODES.get(part.get('code'), 'UNKNOWN')} "
+        f"{FS_CODES.get(part.get('filesystem'), 'UNKNOWN')}"
 
         form.os.choices.append(
             (partition_value,
@@ -2499,10 +2494,6 @@ def action_image_update():
                     part_content[partition_value] = part['image']
                     break
 
-    if not form.os.choices:
-        flash(_(f'No valid partition available'), category='error')
-        return redirect(url_for('commands'))
-
     scopes, _clients = get_scopes(set(ips))
     selected_clients = list(get_selected_clients(scopes['scope']).items())
 
@@ -3484,14 +3475,6 @@ def user_add_post():
         flash(form.errors, category='error')
         return redirect(url_for('users'))
 
-    if re.search(r'\s', form.username.data):
-        flash(_('The username cannot contain spaces'), category='error')
-        return redirect(url_for('users'))
-
-    if re.search(r'\s', form.pwd.data):
-        flash(_('The password cannot contain spaces'), category='error')
-        return redirect(url_for('users'))
-
     if get_user(form.username.data):
         flash(_('This username already exists'), category='error')
         return redirect(url_for('users'))