FGFS-4.1/phpvms
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Account ICAO for subfleet, airline/location if columns exist #628 (#632)

Browse Source
This commit is contained in:
Nabeel S 2020-03-09 13:40:10 -04:00 committed by GitHub
parent 80b20a8b25
commit 412919fe21
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 47 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
modules/Importer/Services/Importers/AircraftImporter.php
View File

@ -11,32 +11,56 @@ 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',
];
$this->info('Subfleet ID is '.$subfleet->id);
// See if there is an airline column
$columns = $this->db->getColumns($this->table);
if (in_array('airline', $columns)) {
$fields[] = 'airline';
}
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);
@ -51,16 +75,21 @@ class AircraftImporter extends BaseImporter
/**
* 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]);
}
}