322 lines
9.4 KiB
PHP
322 lines
9.4 KiB
PHP
<?php
|
|
|
|
namespace App\Command;
|
|
|
|
use Symfony\Component\Console\Command\Command;
|
|
use Symfony\Component\Console\Input\InputArgument;
|
|
use Symfony\Component\Console\Input\InputInterface;
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
|
|
use Doctrine\ORM\EntityManagerInterface;
|
|
|
|
use CrEOF\Spatial\PHP\Types\Geometry\Point;
|
|
use DateTime;
|
|
|
|
use App\Entity\VehicleManufacturer;
|
|
use App\Entity\Vehicle;
|
|
use App\Entity\Battery;
|
|
use App\Entity\BatteryManufacturer;
|
|
use App\Entity\BatteryModel;
|
|
use App\Entity\BatterySize;
|
|
|
|
class ImportVehicleCompatibilityCommand extends Command
|
|
{
|
|
protected $em;
|
|
protected $vmfg_index;
|
|
protected $v_index;
|
|
protected $bmodel_index;
|
|
protected $bsize_index;
|
|
protected $b_index;
|
|
|
|
public function __construct(EntityManagerInterface $em)
|
|
{
|
|
$this->em = $em;
|
|
$this->vmfg_index = [];
|
|
$this->v_index = [];
|
|
$this->b_index = [];
|
|
|
|
parent::__construct();
|
|
}
|
|
|
|
protected function configure()
|
|
{
|
|
$this->setName('vehicle:import_old')
|
|
->setDescription('Import a CSV file with vehicles and batteries.')
|
|
->setHelp('Creates vehicles and batteries based on imported CSV.')
|
|
->addArgument('file', InputArgument::REQUIRED, 'Path to the CSV file.');
|
|
}
|
|
|
|
protected function populateVehicleIndex()
|
|
{
|
|
$vs = $this->em->getRepository(Vehicle::class)->findAll();
|
|
|
|
$this->v_index = [];
|
|
$this->vmfg_index = [];
|
|
foreach ($vs as $v)
|
|
{
|
|
$mfg_name = $v->getManufacturer()->getName();
|
|
$this->vmfg_index[$mfg_name] = $v->getManufacturer();
|
|
if (!isset($this->v_index[$mfg_name]))
|
|
$this->v_index[$mfg_name] = [];
|
|
|
|
$this->v_index[$mfg_name][$v->getMake() . '|' . $v->getModelYearFrom() . '|' . $v->getModelYearTo()] = $v;
|
|
}
|
|
}
|
|
|
|
protected function populateBatteryIndex()
|
|
{
|
|
$bs = $this->em->getRepository(Battery::class)->findAll();
|
|
|
|
$this->b_index = [];
|
|
$this->bmodel_index = [];
|
|
$this->bsize_index = [];
|
|
foreach ($bs as $b)
|
|
{
|
|
$this->bsize_index[$b->getSize()->getName()] = $b->getSize();
|
|
$this->bmodel_index[$b->getModel()->getName()] = $b->getModel();
|
|
$this->b_index[$b->getSize()->getName() . '|' . $b->getModel()->getName()] = $b;
|
|
}
|
|
}
|
|
|
|
protected function handleBatteryData(&$batteries, &$b_models, $batt_field, $row_num)
|
|
{
|
|
// split battery into parts
|
|
$batt = trim($batt_field);
|
|
if ($batt == 'AGM')
|
|
{
|
|
$b_model = 'AGM';
|
|
$b_size = 'AGM';
|
|
}
|
|
else
|
|
{
|
|
$res = preg_match_all("/(.*)(GOLD|EXCEL|ENDURO|TRUCKMASTER|EFB)(.*)/", $batt, $matches);
|
|
|
|
if ($res)
|
|
{
|
|
$b_size = trim($matches[1][0]);
|
|
$b_model = trim(trim($matches[2][0]) . ' ' . trim($matches[3][0]));
|
|
}
|
|
else
|
|
{
|
|
$b_size = $batt;
|
|
$b_model = '';
|
|
}
|
|
|
|
// print_r($b_size);
|
|
// print_r($b_model);
|
|
}
|
|
|
|
// link battery to row num for vehicle matching later
|
|
if (!isset($batteries[$b_size]))
|
|
$batteries[$b_size] = [];
|
|
if (!isset($batteries[$b_size][$b_model]))
|
|
$batteries[$b_size][$b_model] = [];
|
|
|
|
$batteries[$b_size][$b_model][] = $row_num;
|
|
$b_models[$b_model] = null;
|
|
}
|
|
|
|
protected function execute(InputInterface $input, OutputInterface $output)
|
|
{
|
|
$csv_file = $input->getArgument('file');
|
|
$this->populateVehicleIndex();
|
|
$this->populateBatteryIndex();
|
|
|
|
/*
|
|
CSV column order:
|
|
0 - brandh
|
|
1 - make
|
|
2 - model
|
|
3 - recommended battery
|
|
4 - alternative battery
|
|
5 - ignore
|
|
*/
|
|
|
|
// attempt to open file
|
|
try
|
|
{
|
|
$fh = fopen($csv_file, "r");
|
|
}
|
|
catch (Exception $e)
|
|
{
|
|
throw new Exception('The file "' . $csv_file . '" could be read.');
|
|
}
|
|
|
|
// get entity manager
|
|
$em = $this->em;
|
|
|
|
$vbrands = [];
|
|
$batteries = [];
|
|
$bmodels = [];
|
|
|
|
// loop through rows
|
|
$row_num = 1;
|
|
while (($fields = fgetcsv($fh)) !== false)
|
|
{
|
|
$output->writeln("Parsing row " . $row_num . "...");
|
|
|
|
// ignore first row
|
|
if ($row_num == 1)
|
|
{
|
|
$row_num++;
|
|
continue;
|
|
}
|
|
|
|
// parse it
|
|
$brand = trim($fields[0]);
|
|
if (!isset($vbrands[$brand]))
|
|
{
|
|
// build array
|
|
$vbrands[$brand] = [
|
|
'vehicles' => [],
|
|
];
|
|
}
|
|
|
|
$make = trim($fields[1]);
|
|
|
|
$model = trim($fields[2]);
|
|
if (empty($model))
|
|
$model = 'NONE';
|
|
|
|
$vbrands[$brand]['vehicles'][$row_num] = [
|
|
'make' => $make,
|
|
'model' => $model,
|
|
];
|
|
|
|
// battery
|
|
$this->handleBatteryData($batteries, $b_models, $fields[3], $row_num);
|
|
$this->handleBatteryData($batteries, $b_models, $fields[4], $row_num);
|
|
|
|
|
|
$row_num++;
|
|
}
|
|
// print_r($b_models);
|
|
// print_r($batteries);
|
|
|
|
// save to db
|
|
|
|
// vehicles
|
|
$vehicle_index = [];
|
|
foreach ($vbrands as $brand_name => $vbrand)
|
|
{
|
|
// get manufacturer or make one
|
|
if (!isset($this->vmfg_index[$brand_name]))
|
|
{
|
|
// manufacturer
|
|
$mfg = new VehicleManufacturer();
|
|
$mfg->setName($brand_name);
|
|
$em->persist($mfg);
|
|
}
|
|
else
|
|
{
|
|
$mfg = $this->vmfg_index[$brand_name];
|
|
}
|
|
|
|
// vehicles
|
|
foreach ($vbrand['vehicles'] as $row_num => $vdata)
|
|
{
|
|
$model = $vdata['model'];
|
|
if ($model == 'NONE')
|
|
{
|
|
$m_year_from = 0;
|
|
$m_year_to = 0;
|
|
}
|
|
else
|
|
{
|
|
$ex_model = explode('-', $model);
|
|
$m_year_from = trim($ex_model[0]);
|
|
|
|
if (isset($ex_model[1]))
|
|
$m_year_to = trim($ex_model[1]);
|
|
else
|
|
$m_year_to = 0;
|
|
}
|
|
|
|
if (!isset($this->v_index[$brand_name][$vdata['make'] . '|' . intval($m_year_from) . '|' . intval($m_year_to)]))
|
|
{
|
|
echo "NO MATCH ";
|
|
echo $vdata['make'] . '|' . intval($m_year_from) . '|' . intval($m_year_to) + 0;
|
|
echo "\n";
|
|
// vehicle
|
|
$vehicle = new Vehicle();
|
|
$vehicle->setManufacturer($mfg)
|
|
->setMake($vdata['make'])
|
|
->setModelYearFrom($m_year_from)
|
|
->setModelYearTo($m_year_to);
|
|
$em->persist($vehicle);
|
|
}
|
|
else
|
|
{
|
|
$vehicle = $this->v_index[$brand_name][$vdata['make'] . '|' . intval($m_year_from) . '|' . intval($m_year_to)];
|
|
}
|
|
|
|
// update our data for battery linking
|
|
$vehicle_index[$row_num] = $vehicle;
|
|
}
|
|
|
|
$em->flush();
|
|
}
|
|
|
|
// battery models
|
|
foreach ($b_models as $b_model => $throw_away)
|
|
{
|
|
if (!isset($this->bmodel_index[$b_model]))
|
|
{
|
|
$bmodel = new BatteryModel();
|
|
$bmodel->setName($b_model);
|
|
$em->persist($bmodel);
|
|
}
|
|
else
|
|
$bmodel = $this->bmodel_index[$b_model];
|
|
|
|
$b_models[$b_model] = $bmodel;
|
|
}
|
|
|
|
// batteries
|
|
$bmfg = $em->getRepository(BatteryManufacturer::class)->find(1);
|
|
foreach ($batteries as $b_size => $batt_models)
|
|
{
|
|
if (!isset($this->bsize_index[$b_size]))
|
|
{
|
|
$bsize = new BatterySize();
|
|
$bsize->setName($b_size);
|
|
$em->persist($bsize);
|
|
}
|
|
else
|
|
$bsize = $this->bsize_index[$b_size];
|
|
|
|
foreach ($batt_models as $b_model => $vehicle_rows)
|
|
{
|
|
$bmodel = $b_models[$b_model];
|
|
|
|
if (!isset($this->b_index[$bsize->getName() . '|' . $bmodel->getName()]))
|
|
{
|
|
$battery = new Battery();
|
|
$battery->setManufacturer($bmfg)
|
|
->setModel($bmodel)
|
|
->setSize($bsize)
|
|
->setProductCode('')
|
|
->setSAPCode('')
|
|
->setWarrantyPrivate(21)
|
|
->setWarrantyCommercial(6)
|
|
->setSellingPrice(5000);
|
|
|
|
$em->persist($battery);
|
|
}
|
|
else
|
|
$battery = $this->b_index[$bsize->getName() . '|' . $bmodel->getName()];
|
|
|
|
// link to vehicles
|
|
foreach ($vehicle_rows as $row_num)
|
|
{
|
|
$vehicle = $vehicle_index[$row_num];
|
|
$battery->addVehicle($vehicle);
|
|
}
|
|
}
|
|
}
|
|
|
|
$em->flush();
|
|
|
|
return 0;
|
|
}
|
|
}
|