Skip to content

Commit

Permalink
Account ICAO for subfleet, airline/location if columns exist #628
Browse files Browse the repository at this point in the history
  • Loading branch information
nabeelio committed Mar 9, 2020
1 parent 80b20a8 commit eb0777c
Showing 1 changed file with 48 additions and 18 deletions.
66 changes: 48 additions & 18 deletions modules/Importer/Services/Importers/AircraftImporter.php
Original file line number Diff line number Diff line change
Expand Up @@ -11,32 +11,57 @@ class AircraftImporter extends BaseImporter
{
protected $table = 'aircraft';

/**
* CONSTANTS
*/
public const SUBFLEET_NAME = 'Imported Aircraft';

public function run($start = 0)
{
$this->comment('--- AIRCRAFT IMPORT ---');

$subfleet = $this->getSubfleet();
$fields = [
'id',
'icao',
'name',
'fullname',
'registration',
'enabled',
];

// See if there is an airline column
$columns = $this->db->getColumns($this->table);
if (in_array('airline', $columns)) {
$fields[] = 'airline';
}

$this->info('Subfleet ID is '.$subfleet->id);
if (in_array('location', $columns)) {
$fields[] = 'location';
}

$count = 0;
$rows = $this->db->readRows($this->table, $this->idField, $start);
$rows = $this->db->readRows($this->table, $this->idField, $start, $fields);
foreach ($rows as $row) {

$airline_id = null;
if (!empty($row->airline)) {
$airline_id = $this->idMapper->getMapping('airlines', $row->airline_id);
}

$subfleet = $this->getSubfleet($row->icao, $airline_id);
$this->info('Subfleet ID is '.$subfleet->id);

$where = [
'name' => $row->fullname,
'registration' => $row->registration,
];

$aircraft = Aircraft::firstOrCreate($where, [
$cols = [
'icao' => $row->icao,
'name' => $row->fullname,
'subfleet_id' => $subfleet->id,
'active' => $row->enabled,
]);
];

if (!empty($row->location)) {
$cols['airport_id'] = $row->location;
}

$aircraft = Aircraft::firstOrCreate($where, $cols);

$this->idMapper->addMapping('aircraft', $row->id, $aircraft->id);

Expand All @@ -51,16 +76,21 @@ public function run($start = 0)
/**
* Return the subfleet
*
* @param string $icao ICAO of the subfleet
* @param int $airline_id
*
* @return mixed
*/
protected function getSubfleet()
protected function getSubfleet($icao, $airline_id = null)
{
$airline = Airline::first();
$subfleet = Subfleet::firstOrCreate([
'airline_id' => $airline->id,
'name' => self::SUBFLEET_NAME,
], ['type' => 'PHPVMS']);
if ($airline_id === null) {
$airline = Airline::first();
$airline_id = $airline->id;
}

return $subfleet;
return Subfleet::firstOrCreate([
'airline_id' => $airline_id,
'name' => $icao,
], ['type' => $icao]);
}
}

0 comments on commit eb0777c

Please sign in to comment.