Rewrite assign_menu_location().

HGNM · Jun 8, 2016 · 2365f7e · 2365f7e · delucis · Jun 9, 2016
1 parent 0e65fa9
commit 2365f7e
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/functions.php b/functions.php
@@ -25,16 +25,10 @@ function register_my_menu() {
 
 function assign_menu_location() {
   $locations = get_nav_menu_locations();
-  if(!empty($locations)) {
-    foreach($locations as $locationId => $menuValue) {
-      switch($locationId) {
-        case 'primary':
-            $menu = get_term_by('slug', 'main-menu', 'nav_menu');
-        break;
-      }
-      if(isset($menu)) {
-        $locations[$locationId] = $menu->term_id;
-      }
+  if(empty($locations) || $locations['primary'] == 0) {
+    $menu = get_term_by('slug', 'main-menu', 'nav_menu');
+    if(isset($menu)) {
+      $locations['primary'] = $menu->term_id;
     }
     set_theme_mod('nav_menu_locations', $locations);
   }