From 52d12303ea30c2c31ed152024825ee8735be748d Mon Sep 17 00:00:00 2001 From: azaozz Date: Mon, 9 Feb 2009 19:19:38 +0000 Subject: [PATCH] Add hook "after_db_upgrade" and flush rewrite rules on the next page load after successful upgrade git-svn-id: http://svn.automattic.com/wordpress/trunk@10530 1a063a9b-81f0-0310-95a4-ce76da25c4cd --- wp-admin/admin.php | 12 +++++++++++- wp-admin/includes/upgrade.php | 4 +--- 2 files changed, 12 insertions(+), 4 deletions(-) diff --git a/wp-admin/admin.php b/wp-admin/admin.php index fc6f70780..c88a1d4b7 100644 --- a/wp-admin/admin.php +++ b/wp-admin/admin.php @@ -18,7 +18,17 @@ if ( defined('ABSPATH') ) else require_once('../wp-load.php'); -if ( get_option('db_version') != $wp_db_version ) { +if ( get_option('db_version') == 'db_upgraded' ) { + $wp_rewrite->flush_rules(); + update_option('db_version', $wp_db_version); + + /** + * Runs on the next page load after successful upgrade + * + * @since 2.8 + */ + do_action('after_db_upgrade'); +} elseif ( get_option('db_version') != $wp_db_version ) { wp_redirect(admin_url('upgrade.php?_wp_http_referer=' . urlencode(stripslashes($_SERVER['REQUEST_URI'])))); exit; } diff --git a/wp-admin/includes/upgrade.php b/wp-admin/includes/upgrade.php index f41422b57..922ac5964 100644 --- a/wp-admin/includes/upgrade.php +++ b/wp-admin/includes/upgrade.php @@ -278,9 +278,7 @@ function upgrade_all() { maybe_disable_automattic_widgets(); - $wp_rewrite->flush_rules(); - - update_option('db_version', $wp_db_version); + update_option('db_version', 'db_upgraded'); } /**