[Infinito Theme] Dictionary in icon_hooks

Replace all 'ifs' with a Python dictionary

backend_theme_infinito/icon_hooks.py

@@ -31,128 +31,43 @@ def icons_post_init_hook(cr, registry):
     env = api.Environment(cr, SUPERUSER_ID, {})
     menu_item = env['ir.ui.menu'].search([('parent_id', '=', False)])
 
+    icons_dictionary = {
+        'Contacts': 'contact.png',
+        'Link Tracker': 'link-tracker.png',
+        'Dashboards': 'dashboard.png',
+        'Sales': 'sales.png',
+        'Invoicing': 'invoice.png',
+        'Inventory': 'inventory.png',
+        'Purchase': 'purchase.png',
+        'Calendar': 'calendar.png',
+        'CRM': 'crm.png',
+        'Notes': 'notes.png',
+        'Website': 'website.png',
+        'Point of Sale': 'pos.png',
+        'Manufacturing': 'manufacturing.png',
+        'Repairs': 'repairs.png',
+        'Email Marketing': 'marketing.png',
+        'SMS Marketing': 'sms-marketing.png',
+        'Project': 'project.png',
+        'Surveys': 'surveys.png',
+        'Employees': 'employees.png',
+        'Recruitment': 'recruitment.png',
+        'Attendances': 'attendance.png',
+        'Time Off': 'time-off.png',
+        'Expenses': 'expense.png',
+        'Maintenance': 'maintenance.png',
+        'Live Chat': 'live-chat.png',
+        'Lunch': 'lunch.png',
+        'Fleet': 'fleet.png',
+        'Timesheets': 'timesheets.png',
+        'Events': 'events.png',
+        'eLearning': 'elearning.png',
+        'Members': 'members.png',
+    }
+
     for menu in menu_item:
-        if menu.name == 'Contacts':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'contact.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Link Tracker':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'link-tracker.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Dashboards':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'dashboard.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Sales':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'sales.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Invoicing':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'invoice.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Inventory':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'inventory.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Purchase':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'purchase.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Calendar':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'calendar.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'CRM':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'crm.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Notes':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'notes.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Website':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'website.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Point of Sale':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'pos.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Manufacturing':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'manufacturing.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Repairs':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'repairs.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Email Marketing':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'marketing.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'SMS Marketing':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'sms-marketing.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Project':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'project.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Surveys':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'surveys.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Employees':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'employees.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Recruitment':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'recruitment.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Attendances':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'attendance.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Time Off':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'time-off.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Expenses':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'expense.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Maintenance':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'maintenance.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Live Chat':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'live-chat.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Lunch':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'lunch.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Fleet':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'fleet.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Timesheets':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'timesheets.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Events':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'events.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'eLearning':
-            img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'elearning.png')
-            menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})
-        if menu.name == 'Members':
+        filename = icons_dictionary.get(menu.name)
+        if filename:
             img_path = get_module_resource(
-                'backend_theme_infinito', 'static', 'src', 'img', 'icons', 'members.png')
+                'backend_theme_infinito', 'static', 'src', 'img', 'icons', filename)
             menu.write({'web_icon_data': base64.b64encode(open(img_path, "rb").read())})