107 lines
3.0 KiB
PHP
107 lines
3.0 KiB
PHP
<?php
|
|
|
|
namespace Skobkin\Bundle\PointToolsBundle\Command;
|
|
|
|
use Doctrine\ORM\EntityManager;
|
|
use Skobkin\Bundle\PointToolsBundle\Entity\User;
|
|
use Symfony\Component\Console\Command\Command;
|
|
use Symfony\Component\Console\Input\{InputInterface, InputArgument, InputOption};
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
use Symfony\Component\Filesystem\Filesystem;
|
|
|
|
/**
|
|
* Import users from CSV file exported from database by query:
|
|
* COPY (
|
|
* SELECT u.id, u.login, ui.name, to_char(ui.created, 'YYYY-MM-DD_HH24:MI:SS') AS created_at
|
|
* FROM users.logins u
|
|
* LEFT JOIN users.info ui ON (ui.id = u.id)
|
|
* WHERE u.id <> (-1)
|
|
* ) TO '/tmp/point_users.csv' WITH HEADER DELIMITER '|' CSV;
|
|
*/
|
|
class ImportUsersCommand extends Command
|
|
{
|
|
/** @var EntityManager */
|
|
private $em;
|
|
|
|
public function __construct(EntityManager $em)
|
|
{
|
|
$this->em = $em;
|
|
|
|
parent::__construct();
|
|
}
|
|
|
|
protected function configure()
|
|
{
|
|
$this
|
|
->setName('point:import:users')
|
|
->setDescription('Import users from CSV file')
|
|
->addArgument(
|
|
'file',
|
|
InputArgument::REQUIRED,
|
|
'CSV file path'
|
|
)
|
|
->addOption(
|
|
'check-only',
|
|
null,
|
|
InputOption::VALUE_NONE,
|
|
'If set, command will not perform write operations in the database'
|
|
)
|
|
->addOption(
|
|
'no-skip-first',
|
|
null,
|
|
InputOption::VALUE_NONE,
|
|
'Do not skip first line (if no headers in CSV file)'
|
|
)
|
|
;
|
|
}
|
|
|
|
protected function execute(InputInterface $input, OutputInterface $output)
|
|
{
|
|
$fs = new Filesystem();
|
|
|
|
$fileName = $input->getArgument('file');
|
|
|
|
if (!($fs->exists($fileName) && is_readable($fileName))) {
|
|
$output->writeln('File does not exists or not readable.');
|
|
return 1;
|
|
}
|
|
|
|
if (false === ($file = fopen($fileName, 'r'))) {
|
|
$output->writeln('fopen() error');
|
|
return 1;
|
|
}
|
|
|
|
if (!$input->getOption('no-skip-first')) {
|
|
// Reading headers line
|
|
fgets($file);
|
|
}
|
|
|
|
$count = 0;
|
|
|
|
while (false !== ($row = fgetcsv($file, 1000, '|'))) {
|
|
if (count($row) !== 4) {
|
|
continue;
|
|
}
|
|
|
|
$createdAt = \DateTime::createFromFormat('Y-m-d_H:i:s', $row[3]) ?: new \DateTime();
|
|
|
|
$user = new User($row[0], $createdAt, $row[1], $row[2]);
|
|
|
|
if (!$input->getOption('check-only')) {
|
|
$this->em->persist($user);
|
|
$this->em->flush($user);
|
|
$this->em->detach($user);
|
|
}
|
|
|
|
if (OutputInterface::VERBOSITY_VERBOSE === $output->getVerbosity()) {
|
|
$output->writeln('@' . $row[1] . ' added');
|
|
}
|
|
|
|
$count++;
|
|
}
|
|
|
|
$output->writeln($count . ' users imported.');
|
|
|
|
return 0;
|
|
}
|
|
} |