1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
<?php
/**
* Copyright (C) 2005 Brion Vibber <brion@pobox.com>
* http://www.mediawiki.org/
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
* http://www.gnu.org/copyleft/gpl.html
*
* @package MediaWiki
* @subpackage Maintenance
*/
$optionsWithArgs = array( 'report' );
require_once( 'commandLine.inc' );
require_once( 'SpecialImport.php' );
class BackupReader {
var $reportingInterval = 100;
var $reporting = true;
var $pageCount = 0;
var $revCount = 0;
var $dryRun = false;
function BackupReader() {
$this->stderr = fopen( "php://stderr", "wt" );
}
function reportPage( $page ) {
$this->pageCount++;
}
function handleRevision( $rev ) {
$title = $rev->getTitle();
$display = $title->getPrefixedText();
$timestamp = $rev->getTimestamp();
#echo "$display $timestamp\n";
$this->revCount++;
$this->report();
if( !$this->dryRun ) {
call_user_func( $this->importCallback, $rev );
}
}
function report( $final = false ) {
if( $final xor ( $this->pageCount % $this->reportingInterval == 0 ) ) {
$this->showReport();
}
}
function showReport() {
if( $this->reporting ) {
$delta = wfTime() - $this->startTime;
if( $delta ) {
$rate = $this->pageCount / $delta;
$revrate = $this->revCount / $delta;
} else {
$rate = '-';
$revrate = '-';
}
$this->progress( "$this->pageCount ($rate pages/sec $revrate revs/sec)" );
}
}
function progress( $string ) {
fwrite( $this->stderr, $string . "\n" );
}
function importFromFile( $filename ) {
if( preg_match( '/\.gz$/', $filename ) ) {
$filename = 'compress.zlib://' . $filename;
}
$file = fopen( $filename, 'rt' );
$this->importFromHandle( $file );
}
function importFromStdin() {
$file = fopen( 'php://stdin', 'rt' );
$this->importFromHandle( $file );
}
function importFromHandle( $handle ) {
$this->startTime = wfTime();
$source = new ImportStreamSource( $handle );
$importer = new WikiImporter( $source );
$importer->setPageCallback( array( &$this, 'reportPage' ) );
$this->importCallback = $importer->setRevisionCallback(
array( &$this, 'handleRevision' ) );
$importer->doImport();
}
}
$reader = new BackupReader();
if( isset( $options['quiet'] ) ) {
$reader->reporting = false;
}
if( isset( $options['report'] ) ) {
$reader->reportingInterval = IntVal( $options['report'] );
}
if( isset( $options['dry-run'] ) ) {
$reader->dryRun = true;
}
if( isset( $args[0] ) ) {
$reader->importFromFile( $args[0] );
} else {
$reader->importFromStdin();
}
?>
|
