Prepare database structure

server/models/airport.js

@@ -0,0 +1,28 @@
+'use strict';
+const {
+  Model
+} = require('sequelize');
+module.exports = (sequelize, DataTypes) => {
+  class Airport extends Model {
+    /**
+     * Helper method for defining associations.
+     * This method is not a part of Sequelize lifecycle.
+     * The `models/index` file will call this method automatically.
+     */
+    static associate(models) {
+      // define association here
+    }
+  }
+  Airport.init({
+    name: DataTypes.STRING,
+    iata_code: DataTypes.STRING,
+    icao_code: DataTypes.STRING,
+    lat: DataTypes.FLOAT,
+    lng: DataTypes.FLOAT,
+    country_code: DataTypes.STRING
+  }, {
+    sequelize,
+    modelName: 'Airport',
+  });
+  return Airport;
+};

server/models/flight.js

@@ -0,0 +1,43 @@
+'use strict';
+const {
+  Model
+} = require('sequelize');
+module.exports = (sequelize, DataTypes) => {
+  class Flight extends Model {
+    /**
+     * Helper method for defining associations.
+     * This method is not a part of Sequelize lifecycle.
+     * The `models/index` file will call this method automatically.
+     */
+    static associate(models) {
+      // define association here
+    }
+  }
+  Flight.init({
+    airline_iata: DataTypes.STRING,
+    airline_icao: DataTypes.STRING,
+    flight_number: DataTypes.STRING,
+    flight_iata: DataTypes.STRING,
+    flight_icao: DataTypes.STRING,
+    cs_airline_data: DataTypes.STRING,
+    cs_flight_iata: DataTypes.STRING,
+    cs_flight_number: DataTypes.STRING,
+    dep_iata: DataTypes.STRING,
+    dep_icao: DataTypes.STRING,
+    dep_terminals: DataTypes.JSON,
+    dep_time: DataTypes.TIME,
+    dep_time_utc: DataTypes.TIME,
+    arr_iata: DataTypes.STRING,
+    arr_icao: DataTypes.STRING,
+    arr_terminals: DataTypes.JSON,
+    arr_time: DataTypes.TIME,
+    arr_time_utc: DataTypes.TIME,
+    duration: DataTypes.INTEGER,
+    updated: DataTypes.DATE,
+    days: DataTypes.JSON
+  }, {
+    sequelize,
+    modelName: 'Flight',
+  });
+  return Flight;
+};

server/models/index.js

@@ -0,0 +1,43 @@
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const Sequelize = require('sequelize');
+const process = require('process');
+const basename = path.basename(__filename);
+const env = process.env.NODE_ENV || 'development';
+const config = require(__dirname + '/../config/config.json')[env];
+const db = {};
+
+let sequelize;
+if (config.use_env_variable) {
+  sequelize = new Sequelize(process.env[config.use_env_variable], config);
+} else {
+  sequelize = new Sequelize(config.database, config.username, config.password, config);
+}
+
+fs
+  .readdirSync(__dirname)
+  .filter(file => {
+    return (
+      file.indexOf('.') !== 0 &&
+      file !== basename &&
+      file.slice(-3) === '.js' &&
+      file.indexOf('.test.js') === -1
+    );
+  })
+  .forEach(file => {
+    const model = require(path.join(__dirname, file))(sequelize, Sequelize.DataTypes);
+    db[model.name] = model;
+  });
+
+Object.keys(db).forEach(modelName => {
+  if (db[modelName].associate) {
+    db[modelName].associate(db);
+  }
+});
+
+db.sequelize = sequelize;
+db.Sequelize = Sequelize;
+
+module.exports = db;