convheb/create_list_for_annex.py

from typing import List
from cocluto.inventory import Inventory
from cocluto.SimpaDbUtil import SqlDatabaseReader, SqlFile
from urllib.parse import urlparse
import re



def get_md_table_separator_line(columns_width):
    sep_line = '|'
    for col_width in columns_width:
        sep_line += '-' * col_width
        sep_line += '|'
    return sep_line


def render_table_md(columns_data, labels):
    assert len(labels) == len(columns_data)
    num_cols = len(columns_data)
    num_rows = len(columns_data[0])
    cols_width = [0 for col in range(num_cols)]
    for icol in range(num_cols):
        cols_width[icol] = max(cols_width[icol], len(labels[icol]))
        for irow in range(num_rows):
            cols_width[icol] = max(cols_width[icol], len(columns_data[icol][irow]))
    sep_line = get_md_table_separator_line([col_width + 2 for col_width in cols_width])
    line_format = "|" + ''.join([" %%%ds |" % col_width for col_width in cols_width])
    print(sep_line)
    print(line_format % tuple(labels))
    print(sep_line)
    for irow in range(num_rows):
        row = []
        for icol in range(num_cols):
            row.append(columns_data[icol][irow])

        print(line_format % tuple(row))
    print(sep_line)


def render_table_wiki(columns_data, labels):
    assert len(labels) == len(columns_data)
    num_cols = len(columns_data)
    num_rows = len(columns_data[0])
    cols_width = [0 for col in range(num_cols)]
    for icol in range(num_cols):
        cols_width[icol] = max(cols_width[icol], len(labels[icol]))
        for irow in range(num_rows):
            cols_width[icol] = max(cols_width[icol], len(columns_data[icol][irow]))
    cols_width = [col_width + 1 for col_width in cols_width]
    header_line_format = "^" + ''.join([" %%%ds ^" % col_width for col_width in cols_width])
    line_format = "|" + ''.join([" %%%ds |" % col_width for col_width in cols_width])
    print(header_line_format % tuple(labels))
    for irow in range(num_rows):
        row = []
        for icol in range(num_cols):
            row.append(columns_data[icol][irow])

        print(line_format % tuple(row))


def print_machine_table(machines_fqdn: List[str], inventory: Inventory):

    machines_name = []
    machines_spec_id = []
    machines_serial_number = []
    for machine_fqdn in machines_fqdn:
        machine_name = machine_fqdn.split('.')[0]
        machine_spec_id = inventory.machine_name_to_machine_spec_id(machine_name)
        serial_number = inventory.get_machine_serial_number(machine_name)

        machines_name.append(machine_name)
        machines_spec_id.append(machine_spec_id)
        machines_serial_number.append(serial_number)
    name_label = 'Nom'
    model_label = 'Modèle'
    serial_label = 'Numéro de Série'

    render_table_md([machines_name, machines_spec_id, machines_serial_number], [name_label, model_label, serial_label])
    render_table_wiki([machines_name, machines_spec_id, machines_serial_number], [name_label, model_label, serial_label])


#  get the list of machines hosted at the dsi
def get_hosted_machines_fqdn(inventory: Inventory) -> List[str]:
    machines_fqdn = []
    domain_name = 'ipr.univ-rennes1.fr'
    dsi_rack_ids = [
        'rack_dsi_h7',
        'rack_dsi_i7'
    ]
    conditions = [('rack_id == "%s"' % rack_id) for rack_id in dsi_rack_ids]
    sql_query = 'select machine_id from rackable_machine_to_location where %s;' % ' or '.join(conditions)
    rows = inventory.query(sql_query)
    for (machine_id, ) in rows:
        match = re.match(r'(?P<cluster_name>[a-z]+)(?P<from>[0-9]+)_(?P<to>[0-9]+)', machine_id)
        if match:
            # handle containers such as physix72_75
            from_index = int(match['from'])
            to_index = int(match['to'])
            cluster_name = match['cluster_name']
            # print(from_index, to_index)
            for machine_index in range(from_index, to_index + 1):
                machine_id = '%s%d' % (cluster_name, machine_index)
                if machine_id == 'physix75':
                    machine_id = 'alambix75'  # hack because the inventory expects all machines of a container such as physix72_75 to belong to the same cluster, and it's not the case here
                machines_fqdn.append('%s.%s' % (machine_id, domain_name))
        else:
            if re.match('^switch', machine_id) is None:  # ignore switches, as they belong to the dsi
                machines_fqdn.append('%s.%s' % (machine_id, domain_name))
    return sorted(machines_fqdn)


def main():
    itinv_db_url = "file:///home/graffy/work/simpaweb/itinv.git/itinv.sql"
    url_parts = urlparse(itinv_db_url)
    sql_source = SqlFile(url_parts.path)
    sql_reader = SqlDatabaseReader(sql_source)
    inventory = Inventory(sql_reader)

    machines_fqdn = get_hosted_machines_fqdn(inventory)
    print('ipr machines hosted at the dsi (%d):' % len(machines_fqdn))

    print_machine_table(machines_fqdn, inventory)


if __name__ == '__main__':
    main()