Return representive state data for module migrate_rgw_pools

Signed-off-by: Tobias Wolf <[email protected]>

Author: NotTheEvilOne

src/rookify/modules/migrate_rgw_pools/main.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+from collections import OrderedDict
 from typing import Any, Dict
 from ..exception import ModuleException
 from ..machine import Machine
@@ -70,6 +71,25 @@ def execute(self) -> None:
         for zone_name, zone_data in zones.items():
             self._migrate_zone(zone_name, zone_data)
 
+    def get_readable_key_value_state(self) -> Dict[str, str]:
+        migrated_pools = self.machine.get_execution_state_data(
+            "MigrateRgwPoolsHandler", "migrated_pools", default_value=[]
+        )
+
+        zones = self.machine.get_preflight_state("MigrateRgwPoolsHandler").zones
+
+        kv_state_data = OrderedDict()
+
+        for zone_data in zones.values():
+            for osd_pool in zone_data["osd_pools"].values():
+                key_name = "ceph RGW pool {0}".format(osd_pool["pool_name"])
+                kv_state_data[key_name] = self._get_readable_json_dump(osd_pool)
+
+                key_name = "ceph RGW pool {0} is created".format(osd_pool["pool_name"])
+                kv_state_data[key_name] = osd_pool["pool_name"] in migrated_pools
+
+        return kv_state_data
+
     def _migrate_zone(self, zone_name: str, zone_data: Dict[str, Any]) -> None:
         migrated_zones = self.machine.get_execution_state_data(
             "MigrateRgwPoolsHandler", "migrated_zones", default_value=[]