capsul-flask/capsulflask/consistency.py

import pprint
from flask import current_app
from capsulflask.db import get_model

# {
#   "capsul-123abc45": {
#     "id": "capsul-123abc45",
#     "public_ipv4": "123.123.123.123",
#     "public_ipv6": "::::",
#     "host": "baikal",
#     "network_name": "public1",
#     "virtual_bridge_name": "virbr1",
#     "desired_state": "running"
#   },
#   { ... },
#   ...
# }
def get_all_vms_from_db() -> dict:
  db_hosts = get_model().list_hosts_with_networks(None)
  db_vms_by_host_and_network = get_model().non_deleted_vms_by_host_and_network(None)

  db_vms_by_id = dict()

  for kv in db_hosts.items():
    host_id = kv[0]
    value = kv[1]
    for network in value['networks']:
      if host_id in db_vms_by_host_and_network and network['network_name'] in db_vms_by_host_and_network[host_id]:
        for vm in db_vms_by_host_and_network[host_id][network['network_name']]:
          vm['network_name'] = network['network_name']
          vm['virtual_bridge_name'] = network['virtual_bridge_name']
          vm['host'] = host_id
          db_vms_by_id[vm['id']] = vm
 
  # for vm in db_vms:
  #   if vm["id"] not in db_vms_by_id:
  #     # TODO 
  #     raise Exception("non_deleted_vms_by_host_and_network did not return a vm that was returned by all_vm_ids_with_desired_state") 
  #   else:
  #     db_vms_by_id[vm["id"]]["state"] = vm["desired_state"] 

  return db_vms_by_id


# this returns the same shape of object as get_all_vms_from_db except it has 'state' instead of 'desired_state'
def get_all_vms_from_hosts(db_vms_by_id: dict) -> dict:
  virt_vms_by_id = current_app.config["HUB_MODEL"].get_all_by_id()
  current_app.logger.info(f"\n*******************2:\n{pprint.pformat(virt_vms_by_id)}\n\n\n\n")
  db_hosts = get_model().list_hosts_with_networks(None)

  for vm_id, virt_vm in virt_vms_by_id.items():
    if vm_id in db_vms_by_id:
      if 'host' not in virt_vm:
        virt_vm['host'] = db_vms_by_id[vm_id]['host']
      if 'network_name' not in virt_vm:
        virt_vm['network_name'] = db_vms_by_id[vm_id]['network_name']
    else:
      if 'host' not in virt_vm:
        virt_vm['host'] = "unknown"
      if 'network_name' not in virt_vm:
        virt_vm['network_name'] = "unknown"

    if virt_vm['host'] in db_hosts and virt_vm['network_name'] in db_hosts[virt_vm['host']]:
      virt_vm['virtual_bridge_name'] = db_hosts[virt_vm['host']][virt_vm['network_name']]['virtual_bridge_name']
    else:
      virt_vm['virtual_bridge_name'] = "unknown"

  return virt_vms_by_id

def ensure_vms_and_db_are_synced():
  
  
  pass
  # # Now creating the capsuls running status ui
  # # 



  # for vm in db_vms:
  #   db_ids_dict[vm['id']] = vm['desired_state']
  
  # for vm in virt_vms:
  #   virt_ids_dict[vm['id']] = vm['desired_state']

  # errors = list()

  # for id in db_ids_dict:
  #   if id not in virt_ids_dict:
  #     errors.append(f"{id} is in the database but not in the virtualization model")
  #   elif db_ids_dict[id] != virt_ids_dict[id]:
  #     errors.append(f"{id} has the desired state {db_ids_dict[id]} in the database but current state {virt_ids_dict[id]} in the virtualization model")

  # for id in virt_ids_dict:
  #   if id not in db_ids_dict:
  #     errors.append(f"{id} is in the virtualization model but not in the database")

  # if len(errors) > 0:
  #   email_addresses_raw = current_app.config['ADMIN_EMAIL_ADDRESSES'].split(",")
  #   email_addresses = list(filter(lambda x: len(x) > 6, map(lambda x: x.strip(), email_addresses_raw ) ))

  #   current_app.logger.info(f"cron_task: sending inconsistency warning email to {','.join(email_addresses)}:")
  #   for error in errors:
  #     current_app.logger.info(f"cron_task:      {error}.")
      
  #   current_app.config["FLASK_MAIL_INSTANCE"].send(
  #       Message(
  #         "Capsul Consistency Check Failed",
  #         sender=current_app.config["MAIL_DEFAULT_SENDER"],
  #         body="\n".join(errors),
  #         recipients=email_addresses
  #       )
  #   )