Set constant order of the entries in the main menu

Author: KirillMysnik

srcds/addons/source-python/plugins/admin/core/frontends/menus.py

@@ -8,6 +8,7 @@
 
 # Source.Python Admin
 from ..clients import clients
+from ..config import config
 from ..helpers import format_player_name
 from ..strings import strings_common
 
@@ -144,8 +145,10 @@ def select(self, client):
 
         client.send_popup(self.popup)
 
-# Main section
-main_menu = MenuSection(None, strings_common['title main'], 'admin')
+    def load_order(self, order):
+        self.order = order[:]
+        self.sort(key=lambda entry_: (self.order.index(entry_.id)
+                                      if entry_.id in self.order else -1))
 
 
 class MenuCommand(MenuEntry):
@@ -542,3 +545,12 @@ def _get_player_id(self, player):
 
     def _iter(self):
         yield from PlayerIter(self.base_filter)
+
+
+# =============================================================================
+# >> MAIN MENU
+# =============================================================================
+main_menu = MenuSection(None, strings_common['title main'], 'admin')
+main_menu.load_order(list(map(
+    lambda x: x.strip(), config['menus']['order'].split(',')
+)))

srcds/addons/source-python/plugins/admin/plugins/included/admin_comm_management/admin_comm_management.py

@@ -21,7 +21,7 @@
 # >> MENU ENTRIES
 # =============================================================================
 menu_section = main_menu.add_entry(MenuSection(
-    main_menu, plugin_strings['section_title main']))
+    main_menu, plugin_strings['section_title main'], 'comm_management'))
 
 menu_section_chat = menu_section.add_entry(MenuSection(
     menu_section, plugin_strings['section_title chat']))
@@ -59,6 +59,7 @@
     menu_section_voice,
     plugin_strings['popup_title unblock_voice']))
 
+
 # =============================================================================
 # >> SYNCHRONOUS DATABASE OPERATIONS
 # =============================================================================

srcds/addons/source-python/plugins/admin/plugins/included/admin_kick_ban/admin_kick_ban.py

@@ -70,7 +70,7 @@ class _KickPage(PlayerBasedFeaturePage):
 # >> MENU ENTRIES
 # =============================================================================
 menu_section = main_menu.add_entry(MenuSection(
-    main_menu, plugin_strings['section_title main']))
+    main_menu, plugin_strings['section_title main'], 'kick_ban'))
 
 menu_section_steamid = menu_section.add_entry(MenuSection(
     menu_section, plugin_strings['section_title steamid_bans']))

srcds/addons/source-python/plugins/admin/plugins/included/admin_life_management/admin_life_management.py

@@ -112,7 +112,7 @@ def on_error(self, error):
 # >> MENU ENTRIES
 # =============================================================================
 menu_section = main_menu.add_entry(MenuSection(
-    main_menu, plugin_strings['section_title']))
+    main_menu, plugin_strings['section_title'], 'life_management'))
 
 slay_menu_command = menu_section.add_entry(PlayerBasedMenuCommand(
     slay_feature,

srcds/addons/source-python/plugins/admin/plugins/included/admin_team_management/admin_team_management.py

@@ -100,7 +100,7 @@ class _DelayedMoveToCTFeature(_DelayedMoveToTeamFeature):
 # >> MENU ENTRIES
 # =============================================================================
 menu_section = main_menu.add_entry(MenuSection(
-    main_menu, plugin_strings['section_title']))
+    main_menu, plugin_strings['section_title'], 'team_management'))
 
 move_to_spec_menu_command = menu_section.add_entry(PlayerBasedMenuCommand(
     move_to_spec_feature,

srcds/addons/source-python/plugins/admin/plugins/included/admin_tracking/admin_tracking.py

@@ -720,7 +720,7 @@ def message_func(message):
 # >> MENU ENTRIES
 # =============================================================================
 menu_section = main_menu.add_entry(MenuSection(
-    main_menu, plugin_strings['section_title']))
+    main_menu, plugin_strings['section_title'], 'tracking'))
 
 track_menu_command = menu_section.add_entry(_TrackPopupMenuCommand(
     track_popup_feature,

srcds/cfg/source-python/admin/config.ini

@@ -1,3 +1,6 @@
 [database]
 uri=sqlite:///{admin_data_path}/spa.db
 prefix=spa_
+
+[menus]
+order=kick_ban,tracking,life_management,comm_management,team_management