vendor/sonata-project/media-bundle/src/Command/AddMassMediaCommand.php line 23

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. /*
  6.  * This file is part of the Sonata Project package.
  7.  *
  8.  * (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
  9.  *
  10.  * For the full copyright and license information, please view the LICENSE
  11.  * file that was distributed with this source code.
  12.  */
  14. namespace Sonata\MediaBundle\Command;
  16. use Symfony\Component\Console\Input\InputInterface;
  17. use Symfony\Component\Console\Input\InputOption;
  18. use Symfony\Component\Console\Output\OutputInterface;
  20. /**
  21.  * @final since sonata-project/media-bundle 3.21.0
  22.  */
  23. class AddMassMediaCommand extends BaseCommand
  24. {
  25.     /**
  26.      * @var string[]
  27.      */
  28.     protected $setters;
  30.     /**
  31.      * {@inheritdoc}
  32.      */
  33.     public function configure()
  34.     {
  35.         $this->setName('sonata:media:add-multiple')
  36.             ->setDescription('Add medias in mass into the database')
  37.             ->setDefinition([
  38.                 new InputOption('file', null, InputOption::VALUE_OPTIONAL, 'The file to parse'),
  39.                 new InputOption('delimiter', null, InputOption::VALUE_OPTIONAL, 'Set the field delimiter (one character only)', ','),
  40.                 new InputOption('enclosure', null, InputOption::VALUE_OPTIONAL, 'Set the field enclosure character (one character only).', '"'),
  41.                 new InputOption('escape', null, InputOption::VALUE_OPTIONAL, 'Set the escape character (one character only). Defaults as a backslash', '\\'),
  42.             ]);
  43.     }
  45.     /**
  46.      * {@inheritdoc}
  47.      */
  48.     public function execute(InputInterface $input, OutputInterface $output)
  49.     {
  50.         $fp = $this->getFilePointer($input, $output);
  51.         $imported = -1;
  53.         while (!feof($fp)) {
  54.             $data = fgetcsv($fp, null, $input->getOption('delimiter'), $input->getOption('enclosure'), $input->getOption('escape'));
  56.             if (-1 === $imported) {
  57.                 $this->setters = $data;
  59.                 ++$imported;
  61.                 continue;
  62.             }
  64.             if (!\is_array($data)) {
  65.                 continue;
  66.             }
  68.             ++$imported;
  70.             $this->insertMedia($data, $output);
  71.             $this->optimize();
  72.         }
  74.         $output->writeln('Done!');
  75.     }
  77.     /**
  78.      * @return resource
  79.      */
  80.     protected function getFilePointer(InputInterface $input, OutputInterface $output)
  81.     {
  82.         if (false !== ftell(STDIN)) {
  83.             return STDIN;
  84.         }
  86.         if (!$input->getOption('file')) {
  87.             throw new \RuntimeException('Please provide a CSV file argument or CSV input stream');
  88.         }
  90.         return fopen($input->getOption('file'), 'r');
  91.     }
  93.     protected function insertMedia(array $data, OutputInterface $output)
  94.     {
  95.         $media = $this->getMediaManager()->create();
  97.         foreach ($this->setters as $pos => $name) {
  98.             $media->{'set'.$name}($data[$pos]);
  99.         }
  101.         try {
  102.             $this->getMediaManager()->save($media);
  103.             $output->writeln(sprintf(' > %s - %s', $media->getId(), $media->getName()));
  104.         } catch (\Exception $e) {
  105.             $output->writeln(sprintf('<error>%s</error> : %s', $e->getMessage(), json_encode($data)));
  106.         }
  107.     }
  109.     protected function optimize()
  110.     {
  111.         if ($this->getContainer()->has('doctrine')) {
  112.             $this->getContainer()->get('doctrine')->getManager()->getUnitOfWork()->clear();
  113.         }
  114.     }
  115. }