Address pstodulk's review comments

Introduce LeappComponents enum
oamg · Dec 22, 2023 · a4c1210 · a4c1210
1 parent 6d1a08c
commit a4c1210
Showing 1 changed file with 27 additions and 17 deletions.
diff --git a/repos/system_upgrade/common/libraries/rpms.py b/repos/system_upgrade/common/libraries/rpms.py
@@ -2,25 +2,35 @@
 from leapp.libraries.common.config.version import get_source_major_version
 from leapp.models import InstalledRPM
 
+
+class LeappComponents(object):
+    FRAMEWORK = 'framework'
+    REPOSITORY = 'repository'
+    COCKPIT = 'cockpit'
+    TOOLS = 'tools'
+
+
 LEAPP_PACKAGES_MAP = {
-        'framework': {'7': {'pkgs': ['leapp', 'python2-leapp'],
-                            'deps': ['leapp-deps']},
-                      '8': {'pkgs': ['leapp', 'python3-leapp'],
-                            'deps': ['leapp-deps']}
-                      },
-        'repository': {'7': {'pkgs': ['leapp-upgrade-el7toel8'],
-                             'deps': ['leapp-upgrade-el7toel8-deps']},
-                       '8': {'pkgs': ['leapp-upgrade-el8toel9'],
-                             'deps': ['leapp-upgrade-el8toel9-deps']}
-                       },
-        'cockpit': {'7': {'pkgs': ['cockpit-leapp']},
-                    '8': {'pkgs': ['cockpit-leapp']}
-                    },
-        'tools': {'7': {'pkgs': ['snactor']},
-                  '8': {'pkgs': ['snactor']}
-                  }
+        LeappComponents.FRAMEWORK: {'7': {'pkgs': ['leapp', 'python2-leapp'],
+                                          'deps': ['leapp-deps']},
+                                    '8': {'pkgs': ['leapp', 'python3-leapp'],
+                                          'deps': ['leapp-deps']}
+                                    },
+        LeappComponents.REPOSITORY: {'7': {'pkgs': ['leapp-upgrade-el7toel8'],
+                                           'deps': ['leapp-upgrade-el7toel8-deps']},
+                                     '8': {'pkgs': ['leapp-upgrade-el8toel9'],
+                                           'deps': ['leapp-upgrade-el8toel9-deps']}
+                                     },
+        LeappComponents.COCKPIT: {'7': {'pkgs': ['cockpit-leapp']},
+                                  '8': {'pkgs': ['cockpit-leapp']}
+                                  },
+        LeappComponents.TOOLS: {'7': {'pkgs': ['snactor']},
+                                '8': {'pkgs': ['snactor']}
+                                }
         }
-LEAPP_DEFAULT_COMPONENTS = ['framework', 'repository', 'tools']
+LEAPP_DEFAULT_COMPONENTS = [LeappComponents.FRAMEWORK,
+                            LeappComponents.REPOSITORY,
+                            LeappComponents.TOOLS]
 
 
 def get_installed_rpms():