How to set WordPress Menus to Theme Locations

menus-to-locations.php

<?php

$locations = get_theme_mod( 'nav_menu_locations' ); // registered menu locations in theme
$menus = wp_get_nav_menus(); // registered menus
 
if($menus) {
	foreach($menus as $menu) { // assign menus to theme locations
		if( $menu->name == 'Main' ) {
			$locations['main_navigation'] = $menu->term_id;
		} else if( $menu->name == '404' ) {
			$locations['404_pages'] = $menu->term_id;
		} else if( $menu->name == 'Top' ) {
			$locations['top_navigation'] = $menu->term_id;
		}
	}
}
 
set_theme_mod('nav_menu_locations', $locations); // set menus to locations