From a1bb1b3c2a0631ad41262ef92f6cce02c3d376bf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicolas=20L=C5=93uillet?= <nicolas@loeuillet.org>
Date: Wed, 30 Sep 2015 14:22:38 +0200
Subject: [PATCH] userId is now set in parameters

---
 src/Wallabag/CoreBundle/Command/ImportCommand.php | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/Wallabag/CoreBundle/Command/ImportCommand.php b/src/Wallabag/CoreBundle/Command/ImportCommand.php
index f27aeff7f..4a174202a 100644
--- a/src/Wallabag/CoreBundle/Command/ImportCommand.php
+++ b/src/Wallabag/CoreBundle/Command/ImportCommand.php
@@ -3,6 +3,7 @@
 namespace Wallabag\CoreBundle\Command;
 
 use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
+use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
 use Symfony\Component\Console\Helper\ProgressBar;
@@ -15,7 +16,12 @@ class ImportCommand extends ContainerAwareCommand
     {
         $this
             ->setName('import:json')
-            ->setDescription('Import entries from JSON file');
+            ->setDescription('Import entries from JSON file')
+            ->addArgument(
+                'userId',
+                InputArgument::REQUIRED,
+                'user ID to populate'
+            );
     }
 
     protected function execute(InputInterface $input, OutputInterface $output)
@@ -32,6 +38,8 @@ class ImportCommand extends ContainerAwareCommand
 
     protected function import(InputInterface $input, OutputInterface $output)
     {
+        $userId = $input->getArgument('userId');
+
         // Getting php array of data from CSV
         $data = $this->get($input, $output);
 
@@ -48,7 +56,7 @@ class ImportCommand extends ContainerAwareCommand
         $progress->start();
 
         $user = $em->getRepository('WallabagCoreBundle:User')
-            ->findOneById(1);
+            ->findOneById($userId);
 
         foreach ($data as $object) {
             $array = (array) $object;