makeTestEdits.php

Go to the documentation of this file.

<?php
// @codeCoverageIgnoreStart
require_once __DIR__ . '/Maintenance.php';
// @codeCoverageIgnoreEnd
 
use MediaWiki\Content\ContentHandler;
use MediaWiki\Maintenance\Maintenance;
use MediaWiki\Title\Title;
use MediaWiki\User\User;
 
class MakeTestEdits extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Make test edits for a user' );
        $this->addOption( 'user', 'User name', true, true );
        $this->addOption( 'count', 'Number of edits', true, true );
        $this->addOption( 'namespace', 'Namespace number', false, true );
        $this->addOption( 'watchlist', 'Add edited pages to user watchlist', false, false );
        $this->setBatchSize( 100 );
    }
 
    public function execute() {
        $user = User::newFromName( $this->getOption( 'user' ) );
        if ( !$user->isRegistered() ) {
            $this->fatalError( "No such user exists." );
        }
 
        $count = (int)$this->getOption( 'count' );
        $namespace = (int)$this->getOption( 'namespace', 0 );
        $batchSize = $this->getBatchSize();
        $services = $this->getServiceContainer();
        $wikiPageFactory = $services->getWikiPageFactory();
        $watchedItemStore = $services->getWatchedItemStore();
 
        $titleBatches = $this->newBatchIterator(
            static function () use ( $namespace, $count ) {
                for ( $i = 0; $i < $count; ++$i ) {
                    yield Title::makeTitleSafe( $namespace, "Page " . wfRandomString( 2 ) );
                }
            }
        );
 
        $watchlist = $this->getOption( 'watchlist' );
        foreach ( $titleBatches as $titleBatch ) {
            $editedTitles = $watchlist ? [] : null;
            $this->beginTransactionRound( __METHOD__ );
            foreach ( $titleBatch as $title ) {
                $page = $wikiPageFactory->newFromTitle( $title );
                $content = ContentHandler::makeContent( wfRandomString(), $title );
                $summary = "Change " . wfRandomString( 6 );
 
                $page->doUserEditContent( $content, $user, $summary );
 
                // Collect titles for watchlist if requested
                if ( $watchlist ) {
                    $editedTitles[] = $title;
                }
 
                $this->output( "Edited $title\n" );
            }
            if ( $editedTitles ) {
                $watchedItemStore->addWatchBatchForUser( $user, $editedTitles );
            }
            $this->commitTransactionRound( __METHOD__ );
        }
 
        $this->output( "Done\n" );
    }
}
 
// @codeCoverageIgnoreStart
$maintClass = MakeTestEdits::class;
require_once RUN_MAINTENANCE_IF_MAIN;
// @codeCoverageIgnoreEnd