Adding apps_icons with same logic of apps_label_title

clint74 · clint74 · commit 81e049f2d5cf · 2014-04-17T10:36:01.000-03:00
diff --git a/xadmin/views/base.py b/xadmin/views/base.py
@@ -298,6 +298,7 @@ class CommAdminView(BaseAdminView):
     global_models_icon = {}
     default_model_icon = None
     apps_label_title = {}
+    apps_icons = {}
 
     def get_site_menu(self):
         return None
@@ -321,6 +322,7 @@ def get_url(menu, had_urls):
             if getattr(model_admin, 'hidden_menu', False):
                 continue
             app_label = model._meta.app_label
+            app_icon = None
             model_dict = {
                 'title': unicode(capfirst(model._meta.verbose_name_plural)),
                 'url': self.get_model_url(model, "changelist"),
@@ -349,16 +351,22 @@ def get_url(menu, had_urls):
                                 app_title = getattr(mod, 'verbose_name')
                             elif 'app_title' in dir(mod):
                                 app_title = getattr(mod, 'app_title')
+                #find app icon
+                if app_label.lower() in self.apps_icons:
+                    app_icon = self.apps_icons[app_label.lower()]
 
                 nav_menu[app_key] = {
                     'title': app_title,
                     'menus': [model_dict],
                 }
 
             app_menu = nav_menu[app_key]
-            if ('first_icon' not in app_menu or
+            if app_icon:
+                app_menu['first_icon'] = app_icon
+            elif ('first_icon' not in app_menu or
                     app_menu['first_icon'] == self.default_model_icon) and model_dict.get('icon'):
                 app_menu['first_icon'] = model_dict['icon']
+
             if 'first_url' not in app_menu and model_dict.get('url'):
                 app_menu['first_url'] = model_dict['url']
 
