Yuri Kuznetsov
112 lines
3.7 KiB
PHP
112 lines
3.7 KiB
PHP
<?php
|
|
/************************************************************************
|
|
* This file is part of EspoCRM.
|
|
*
|
|
* EspoCRM - Open Source CRM application.
|
|
* Copyright (C) 2014-2022 Yurii Kuznietsov, Taras Machyshyn, Oleksii Avramenko
|
|
* Website: https://www.espocrm.com
|
|
*
|
|
* EspoCRM 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* EspoCRM 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 EspoCRM. If not, see http://www.gnu.org/licenses/.
|
|
*
|
|
* The interactive user interfaces in modified source and object code versions
|
|
* of this program must display Appropriate Legal Notices, as required under
|
|
* Section 5 of the GNU General Public License version 3.
|
|
*
|
|
* In accordance with Section 7(b) of the GNU General Public License version 3,
|
|
* these Appropriate Legal Notices must retain the display of the "EspoCRM" word.
|
|
************************************************************************/
|
|
|
|
namespace Espo\Classes\ConsoleCommands;
|
|
|
|
use Espo\Core\Console\Command;
|
|
use Espo\Core\Console\Command\Params;
|
|
use Espo\Core\Console\Exceptions\ArgumentNotSpecified;
|
|
use Espo\Core\Console\Exceptions\InvalidArgument;
|
|
use Espo\Core\Console\IO;
|
|
use Espo\Core\Exceptions\Error;
|
|
use Espo\Core\FieldProcessing\NextNumber\BeforeSaveProcessor;
|
|
use Espo\Core\ORM\Entity as CoreEntity;
|
|
use Espo\Core\ORM\Repository\SaveOption;
|
|
use Espo\ORM\EntityManager;
|
|
use Espo\ORM\Query\Part\Order;
|
|
|
|
class PopulateNumbers implements Command
|
|
{
|
|
private BeforeSaveProcessor $beforeSaveProcessor;
|
|
private EntityManager $entityManager;
|
|
|
|
public function __construct(
|
|
BeforeSaveProcessor $beforeSaveProcessor,
|
|
EntityManager $entityManager
|
|
) {
|
|
$this->beforeSaveProcessor = $beforeSaveProcessor;
|
|
$this->entityManager = $entityManager;
|
|
}
|
|
|
|
/**
|
|
* @throws Error
|
|
*/
|
|
public function run(Params $params, IO $io): void
|
|
{
|
|
$entityType = $params->getArgument(0);
|
|
$field = $params->getArgument(1);
|
|
|
|
$orderBy = $params->getOption('orderBy') ?? 'createdAt';
|
|
$order = $params->getOption('order') ?? Order::ASC;
|
|
|
|
if (!$entityType) {
|
|
throw new ArgumentNotSpecified("No entity type argument.");
|
|
}
|
|
|
|
if (!$field) {
|
|
throw new ArgumentNotSpecified("No field argument.");
|
|
}
|
|
|
|
$fieldType = $this->entityManager
|
|
->getDefs()
|
|
->getEntity($entityType)
|
|
->getField($field)
|
|
->getType();
|
|
|
|
if ($fieldType !== 'number') {
|
|
throw new InvalidArgument("Field `{$field}` is not of `number` type.");
|
|
}
|
|
|
|
$collection = $this->entityManager
|
|
->getRDBRepository($entityType)
|
|
->where([
|
|
$field => null,
|
|
])
|
|
->order($orderBy, $order)
|
|
->sth()
|
|
->find();
|
|
|
|
foreach ($collection as $i => $entity) {
|
|
if (!$entity instanceof CoreEntity) {
|
|
throw new Error();
|
|
}
|
|
|
|
$this->beforeSaveProcessor->processPopulate($entity, $field);
|
|
$this->entityManager->saveEntity($entity, [SaveOption::IMPORT => true]);
|
|
|
|
if ($i % 1000 === 0) {
|
|
$io->write('.');
|
|
}
|
|
}
|
|
|
|
$io->writeLine('');
|
|
$io->writeLine('Done.');
|
|
}
|
|
}
|