export improvements, props blepoxp, see #10317

git-svn-id: http://svn.automattic.com/wordpress/trunk@14444 1a063a9b-81f0-0310-95a4-ce76da25c4cd
This commit is contained in:
wpmuguru 2010-05-04 17:31:36 +00:00
parent 9fb099891f
commit cea6004286
2 changed files with 390 additions and 384 deletions

View File

@ -18,40 +18,38 @@ $title = __('Export');
if ( isset( $_GET['download'] ) ) {
$author = isset($_GET['author']) ? $_GET['author'] : 'all';
$category = isset($_GET['category']) ? $_GET['category'] : 'all';
$taxonomy = array();
foreach ( get_taxonomies( array( 'show_ui' => true ) ) as $tax )
$taxonomy[ $tax ] = ! empty( $_GET['taxonomy'][ $tax ] ) ? $_GET['taxonomy'][ $tax ] : 'all';
$post_type = isset($_GET['post_type']) ? stripslashes_deep($_GET['post_type']) : 'all';
$status = isset($_GET['status']) ? stripslashes_deep($_GET['status']) : 'all';
$mm_start = isset($_GET['mm_start']) ? $_GET['mm_start'] : 'all';
$mm_end = isset($_GET['mm_end']) ? $_GET['mm_end'] : 'all';
$aa_start = isset($_GET['aa_start']) ? intval($_GET['aa_start']) : 0;
$aa_end = isset($_GET['aa_end']) ? intval($_GET['aa_end']) : 0;
if($mm_start != 'all' && $aa_start > 0) {
$start_date = sprintf( "%04d-%02d-%02d", $aa_start, $mm_start, 1 );
if( $mm_start != 'all' ) {
$start_date = sprintf( "%04d-%02d-%02d", substr( $mm_start, 0, 4 ), substr( $mm_start, 5, 2 ), 1 );
} else {
$start_date = 'all';
}
if($mm_end != 'all' && $aa_end > 0) {
if($mm_end == 12) {
$mm_end = 1;
$aa_end++;
} else {
$mm_end++;
}
$end_date = sprintf( "%04d-%02d-%02d", $aa_end, $mm_end, 1 );
if( $mm_end != 'all' ) {
$end_date = sprintf( "%04d-%02d-%02d", substr( $mm_end, 0, 4 ), substr( $mm_end, 5, 2 ), 1 );
} else {
$end_date = 'all';
}
export_wp( $author, $category, $post_type, $status, $start_date, $end_date );
export_wp( array( 'author' => $author, 'taxonomy' => $taxonomy, 'post_type' => $post_type, 'post_status' => $status, 'start_date' => $start_date, 'end_date' => $end_date ) );
die();
}
require_once ('admin-header.php');
$months = "";
for ( $i = 1; $i < 13; $i++ ) {
$months .= "\t\t\t<option value=\"" . zeroise($i, 2) . '">' .
$wp_locale->get_month_abbrev( $wp_locale->get_month( $i ) ) . "</option>\n";
} ?>
$dateoptions = '';
if ( $monthyears = $wpdb->get_results( "SELECT YEAR(post_date) AS `year`, MONTH(post_date) AS `month`, count(ID) as posts FROM $wpdb->posts WHERE post_type = 'post' AND post_status = 'publish' GROUP BY YEAR(post_date), MONTH(post_date) ORDER BY post_date DESC " ) ) {
foreach ( $monthyears as $monthyear ) {
$dateoptions .= "\t<option value=\"" . $monthyear->year . '-' . zeroise( $monthyear->month, 2 ) . '">' . $wp_locale->get_month( $monthyear->month ) . ' ' . $monthyear->year . "</option>\n";
}
}
?>
<div class="wrap">
<?php screen_icon(); ?>
@ -66,19 +64,16 @@ for ( $i = 1; $i < 13; $i++ ) {
<table class="form-table">
<tr>
<th><label for="mm_start"><?php _e('Restrict Date'); ?></label></th>
<td><strong><?php _e('Start:'); ?></strong> <?php _e('Month'); ?>&nbsp;
<td><strong><?php _e('Start:'); ?></strong>
<select name="mm_start" id="mm_start">
<option value="all" selected="selected"><?php _e('All Dates'); ?></option>
<?php echo $months; ?>
</select>&nbsp;<?php _e('Year'); ?>&nbsp;
<input type="text" id="aa_start" name="aa_start" value="" size="4" maxlength="5" />
</td>
<td><strong><?php _e('End:'); ?></strong> <?php _e('Month'); ?>&nbsp;
<?php echo ($dateoptions); ?>
</select> <br/>
<strong><?php _e('End:'); ?></strong>
<select name="mm_end" id="mm_end">
<option value="all" selected="selected"><?php _e('All Dates'); ?></option>
<?php echo $months; ?>
</select>&nbsp;<?php _e('Year'); ?>&nbsp;
<input type="text" id="aa_end" name="aa_end" value="" size="4" maxlength="5" />
<?php echo $dateoptions; ?>
</select>
</td>
</tr>
<tr>
@ -87,27 +82,23 @@ for ( $i = 1; $i < 13; $i++ ) {
<select name="author" id="author">
<option value="all" selected="selected"><?php _e('All Authors'); ?></option>
<?php
$authors = $wpdb->get_results( "SELECT DISTINCT u.id, u.display_name FROM $wpdb->users u INNER JOIN $wpdb->posts p ON u.id = p.post_author ORDER BY u.display_name" );
foreach ( (array) $authors as $author )
$authors = $wpdb->get_results( "SELECT DISTINCT u.id, u.display_name FROM $wpdb->users u INNER JOIN $wpdb->posts p WHERE u.id = p.post_author ORDER BY u.display_name" );
foreach ( (array) $authors as $author ) {
echo "<option value='{$author->id}'>{$author->display_name}</option>\n";
}
?>
</select>
</td>
</tr>
<tr>
<th><label for="category"><?php _e('Restrict Category'); ?></label></th>
<th><?php _e('Restrict Taxonomies'); ?></th>
<td>
<select name="category" id="category">
<option value="all" selected="selected"><?php _e('All Categories'); ?></option>
<?php
$categories = (array) get_categories('get=all');
if($categories) {
foreach ( $categories as $cat ) {
echo "<option value='{$cat->term_taxonomy_id}'>{$cat->name}</option>\n";
}
<?php foreach ( get_taxonomies( array( 'show_ui' => true ), 'objects' ) as $tax_obj ) {
$term_dropdown = wp_dropdown_categories( array( 'taxonomy' => $tax_obj->name, 'hide_if_empty' => true, 'show_option_all' => __( 'All Terms' ), 'name' => 'taxonomy[' . $tax_obj->name . ']', 'id' => 'taxonomy-' . $tax_obj->name, 'class' => '', 'echo' => false ) );
if ( $term_dropdown )
echo '<label for="taxonomy-' . $tax_obj->name . '">' . $tax_obj->label . '</label>: ' . $term_dropdown . '<br/>';
}
?>
</select>
</td>
</tr>
<tr>
@ -115,8 +106,9 @@ if($categories) {
<td>
<select name="post_type" id="post_type">
<option value="all" selected="selected"><?php _e('All Content'); ?></option>
<option value="page"><?php _e('Pages'); ?></option>
<option value="post"><?php _e('Posts'); ?></option>
<?php foreach ( get_post_types( array( 'public' => true, 'can_export' => true ), 'objects' ) as $post_type_obj ) { ?>
<option value="<?php echo $post_type_obj->name; ?>"><?php echo $post_type_obj->label; ?></option>
<?php } ?>
</select>
</td>
</tr>
@ -125,10 +117,9 @@ if($categories) {
<td>
<select name="status" id="status">
<option value="all" selected="selected"><?php _e('All Statuses'); ?></option>
<option value="draft"><?php _e('Draft'); ?></option>
<option value="private"><?php _e('Privately published'); ?></option>
<option value="publish"><?php _e('Published'); ?></option>
<option value="future"><?php _e('Scheduled'); ?></option>
<?php foreach ( get_post_stati( array( 'internal' => false ), 'objects' ) as $post_status_obj ) { ?>
<option value="<?php echo $post_status_obj->name; ?>"><?php echo $post_status_obj->label; ?></option>
<?php } ?>
</select>
</td>
</tr>

View File

@ -21,45 +21,55 @@ define('WXR_VERSION', '1.0');
*
* @since unknown
*
* @param unknown_type $author
* @param unknown_type $args
*/
function export_wp($author='', $category='', $post_type='', $status='', $start_date='', $end_date='') {
function export_wp( $args = array() ) {
global $wpdb, $post_ids, $post, $wp_taxonomies;
if ( ! is_array( $args ) )
$args = array( 'author' => $args );
$defaults = array( 'author' => null, 'taxonomy' => null, 'post_type' => null, 'post_status' => null, 'start_date' => null, 'end_date' => null );
$args = wp_parse_args( $args, $defaults );
extract($args);
do_action('export_wp');
if(strlen($start_date) > 4 && strlen($end_date) > 4) {
if( strlen( $start_date ) > 4 && strlen( $end_date ) > 4 )
$filename = 'wordpress.' . $start_date . '.' . $end_date . '.xml';
} else {
else
$filename = 'wordpress.' . date( 'Y-m-d' ) . '.xml';
}
header( 'Content-Description: File Transfer' );
header("Content-Disposition: attachment; filename=$filename");
header( 'Content-Disposition: attachment; filename=' . $filename );
header( 'Content-Type: text/xml; charset=' . get_option( 'blog_charset' ), true );
if ( $post_type and $post_type != 'all' ) {
if ( $post_type && $post_type != 'all' )
$where = $wpdb->prepare("WHERE post_type = %s ", $post_type);
} else {
else
$where = "WHERE post_type != 'revision' ";
}
if ( $author and $author != 'all' ) {
if ( $author && $author != 'all' ) {
$author_id = (int) $author;
$where .= $wpdb->prepare( "AND post_author = %d ", $author_id );
}
if ( $start_date and $start_date != 'all' ) {
if ( $start_date && $start_date != 'all' )
$where .= $wpdb->prepare( "AND post_date >= %s ", $start_date );
}
if ( $end_date and $end_date != 'all' ) {
if ( $end_date && $end_date != 'all' )
$where .= $wpdb->prepare( "AND post_date < %s ", $end_date );
if ( $taxonomy && is_array( $taxonomy ) ) {
foreach ( $taxonomy as $term_id ) {
if ( $term_id != 'all' )
$where .= $wpdb->prepare( "AND ID IN (SELECT object_id FROM $wpdb->term_relationships WHERE term_taxonomy_id = %d) ", $term_id );
}
if ( $category and $category != 'all' ) {
$taxomony_id = (int) $category;
$where .= $wpdb->prepare("AND ID IN (SELECT object_id FROM {$wpdb->term_relationships} " . "WHERE term_taxonomy_id = %d) ", $taxomony_id);
}
if ( $status and $status != 'all' ) {
if ( $post_status && $post_status != 'all' )
$where .= $wpdb->prepare( "AND post_status = %s", $status );
}
// grab a snapshot of post IDs, just in case it changes during the export
$post_ids = $wpdb->get_col( "SELECT ID FROM $wpdb->posts $where ORDER BY post_date_gmt ASC" );
@ -85,8 +95,9 @@ function wxr_missing_parents($categories) {
if ( ! is_array( $categories ) || empty( $categories ) )
return array();
foreach ( $categories as $category )
foreach ( $categories as $category ){
$parents[$category->term_id] = $category->parent;
}
$parents = array_unique( array_diff( $parents, array_keys( $parents ) ) );
@ -108,12 +119,11 @@ while ( $parents = wxr_missing_parents($categories) ) {
$pass = 0;
$passes = 1000 + count( $categories );
while ( ( $cat = array_shift( $categories ) ) && ++$pass < $passes ) {
if ( $cat->parent == 0 || isset($cats[$cat->parent]) ) {
if ( $cat->parent == 0 || isset( $cats[$cat->parent] ) )
$cats[$cat->term_id] = $cat;
} else {
else
$categories[] = $cat;
}
}
unset( $categories );
/**
@ -128,7 +138,6 @@ function wxr_cdata($str) {
$str = utf8_encode( $str );
// $str = ent2ncr(esc_html($str));
$str = "<![CDATA[$str" . ( ( substr( $str, -1 ) == ']' ) ? ' ' : '') . "]]>";
return $str;
@ -145,14 +154,12 @@ function wxr_site_url() {
global $current_site;
// mu: the base url
if ( isset($current_site->domain) ) {
if ( isset( $current_site->domain ) )
return 'http://' . $current_site->domain . $current_site->path;
}
// wp: the blog url
else {
else
return get_bloginfo_rss( 'url' );
}
}
/**
* {@internal Missing Short Description}}
@ -296,7 +303,7 @@ echo '<?xml version="1.0" encoding="' . get_bloginfo('charset') . '"?' . ">\n";
<channel>
<title><?php bloginfo_rss( 'name' ); ?></title>
<link><?php bloginfo_rss( 'url' ) ?></link>
<description><?php bloginfo_rss("description") ?></description>
<description><?php bloginfo_rss( 'description' ) ?></description>
<pubDate><?php echo mysql2date( 'D, d M Y H:i:s +0000', get_lastpostmodified( 'GMT' ), false ); ?></pubDate>
<generator>http://wordpress.org/?v=<?php bloginfo_rss( 'version' ); ?></generator>
<language><?php echo get_option( 'rss_language' ); ?></language>
@ -312,14 +319,19 @@ echo '<?xml version="1.0" encoding="' . get_bloginfo('charset') . '"?' . ">\n";
<?php if ( $terms ) : foreach ( $terms as $t ) : ?>
<wp:term><wp:term_taxonomy><?php echo $t->taxonomy; ?></wp:term_taxonomy><wp:term_slug><?php echo $t->slug; ?></wp:term_slug><wp:term_parent><?php echo $t->parent ? $custom_taxonomies[$t->parent]->name : ''; ?></wp:term_parent><?php wxr_term_name( $t ); ?><?php wxr_term_description( $t ); ?></wp:term>
<?php endforeach; endif; ?>
<?php do_action( 'rss2_head' ); ?>
<?php if ( $post_ids ) {
global $wp_query;
$wp_query->in_the_loop = true; // Fake being in the loop.
// fetch 20 posts at a time rather than loading the entire table into memory
while ( $next_posts = array_splice( $post_ids, 0, 20 ) ) {
$where = "WHERE ID IN (" . join( ',', $next_posts ) . ")";
$posts = $wpdb->get_results( "SELECT * FROM $wpdb->posts $where ORDER BY post_date_gmt ASC" );
// Begin Loop
foreach ($posts as $post) {
setup_postdata( $post );
@ -385,7 +397,10 @@ if ( $comments ) { foreach ( $comments as $c ) { ?>
</wp:comment>
<?php } } ?>
</item>
<?php } } } ?>
<?php
}
}
} ?>
</channel>
</rss>
<?php