phpvms/app/Services/ImportExport/AircraftImporter.php
Nabeel S d03a77bd4b
New subfleet not being attached to an airline on import #479 (#505)
* Fix subfleet not being attached to an airline on creation in import #479

* Call airline name with optional() around subfleet

* Minor cleanup
2020-01-16 14:23:23 -05:00

96 lines
2.5 KiB
PHP

<?php
namespace App\Services\ImportExport;
use App\Contracts\ImportExport;
use App\Models\Aircraft;
use App\Models\Airline;
use App\Models\Enums\AircraftState;
use App\Models\Enums\AircraftStatus;
use App\Models\Subfleet;
use App\Support\ICAO;
/**
* Import aircraft
*/
class AircraftImporter extends ImportExport
{
public $assetType = 'aircraft';
/**
* All of the columns that are in the CSV import
* Should match the database fields, for the most part
*/
public static $columns = [
'subfleet' => 'required',
'iata' => 'nullable',
'icao' => 'nullable',
'name' => 'required',
'registration' => 'required',
'hex_code' => 'nullable',
'zfw' => 'nullable|numeric',
'status' => 'nullable',
];
/**
* Find the subfleet specified, or just create it on the fly and attach it to the
* first airline that's been found
*
* @param $type
*
* @return Subfleet|\Illuminate\Database\Eloquent\Model|null|object|static
*/
protected function getSubfleet($type)
{
return Subfleet::firstOrCreate([
'type' => $type,
], [
'name' => $type,
'airline_id' => Airline::where('active', true)->first()->id,
]);
}
/**
* Import a flight, parse out the different rows
*
* @param array $row
* @param int $index
*
* @throws \Exception
*
* @return bool
*/
public function import(array $row, $index): bool
{
$subfleet = $this->getSubfleet($row['subfleet']);
$row['subfleet_id'] = $subfleet->id;
// Generate a hex code
if (!$row['hex_code']) {
$row['hex_code'] = ICAO::createHexCode();
}
// Set a default status
$row['status'] = trim($row['status']);
if (empty($row['status'])) {
$row['status'] = AircraftStatus::ACTIVE;
}
// Just set its state right now as parked
$row['state'] = AircraftState::PARKED;
// Try to add or update
try {
Aircraft::updateOrCreate([
'registration' => $row['registration'],
], $row);
} catch (\Exception $e) {
$this->errorLog('Error in row '.$index.': '.$e->getMessage());
return false;
}
$this->log('Imported '.$row['registration'].' '.$row['name']);
return true;
}
}