X-Git-Url: https://scripts.mit.edu/gitweb/autoinstallsdev/mediawiki.git/blobdiff_plain/19e297c21b10b1b8a3acad5e73fc71dcb35db44a..6932310fd58ebef145fa01eb76edf7150284d8ea:/maintenance/initUserPreference.php

diff --git a/maintenance/initUserPreference.php b/maintenance/initUserPreference.php
new file mode 100644
index 00000000..f4da570f
--- /dev/null
+++ b/maintenance/initUserPreference.php
@@ -0,0 +1,84 @@
+<?php
+/**
+ * Initialize a user preference based on the value
+ * of another preference.
+ *
+ * @ingroup Maintenance
+ */
+
+require_once __DIR__ . '/Maintenance.php';
+
+/**
+ * Maintenance script that initializes a user preference
+ * based on the value of another preference.
+ *
+ * @ingroup Maintenance
+ */
+class InitUserPreference extends Maintenance {
+	public function __construct() {
+		parent::__construct();
+		$this->addOption(
+			'target',
+			'Name of the user preference to initialize',
+			true,
+			true,
+			't'
+		);
+		$this->addOption(
+			'source',
+			'Name of the user preference to take the value from',
+			true,
+			true,
+			's'
+		);
+		$this->setBatchSize( 300 );
+	}
+
+	public function execute() {
+		$target = $this->getOption( 'target' );
+		$source = $this->getOption( 'source' );
+		$this->output( "Initializing '$target' based on the value of '$source'\n" );
+
+		$dbr = $this->getDB( DB_REPLICA );
+		$dbw = $this->getDB( DB_MASTER );
+
+		$iterator = new BatchRowIterator(
+			$dbr,
+			'user_properties',
+			[ 'up_user', 'up_property' ],
+			$this->mBatchSize
+		);
+		$iterator->setFetchColumns( [ 'up_user', 'up_value' ] );
+		$iterator->addConditions( [
+			'up_property' => $source,
+			'up_value IS NOT NULL',
+			'up_value != 0',
+		] );
+
+		$processed = 0;
+		foreach ( $iterator as $batch ) {
+			foreach ( $batch as $row ) {
+				$values = [
+					'up_user' => $row->up_user,
+					'up_property' => $target,
+					'up_value' => $row->up_value,
+				];
+				$dbw->upsert(
+					'user_properties',
+					$values,
+					[ 'up_user', 'up_property' ],
+					$values,
+					__METHOD__
+				);
+
+				$processed += $dbw->affectedRows();
+			}
+		}
+
+		$this->output( "Processed $processed user(s)\n" );
+		$this->output( "Finished!\n" );
+	}
+}
+
+$maintClass = 'InitUserPreference'; // Tells it to run the class
+require_once RUN_MAINTENANCE_IF_MAIN;