feat: relationships

Chia-Network · Dec 8, 2021 · 0cd24ce · 0cd24ce
1 parent 6ba5853
commit 0cd24ce
Show file tree

Hide file tree

Showing 8 changed files with 32 additions and 10 deletions.
diff --git a/src/models/co-benefits/co-benefits.model.js b/src/models/co-benefits/co-benefits.model.js
@@ -2,6 +2,7 @@
 import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
+import { Project } from '../projects';
 
 class CoBenefit extends Model {
   /**
@@ -10,7 +11,7 @@ class CoBenefit extends Model {
    * The `models/index` file will call this method automatically.
    */
   static associate(models) {
-    // define association here
+    CoBenefit.belongsTo(Project);
   }
 }
 

diff --git a/src/models/locations/locations.model.js b/src/models/locations/locations.model.js
@@ -2,6 +2,7 @@
 import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
+import { Project } from '../projects';
 
 class ProjectLocation extends Model {
   /**
@@ -10,7 +11,7 @@ class ProjectLocation extends Model {
    * The `models/index` file will call this method automatically.
    */
   static associate(models) {
-    // define association here
+    ProjectLocation.belongsTo(Project);
   }
 }
 

diff --git a/src/models/projects/projects.model.js b/src/models/projects/projects.model.js
@@ -3,14 +3,26 @@ import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
 
+import { RelatedProject } from './../related-projects';
+import { Vintage } from './../vintages';
+import { Qualification } from './../qualifications';
+import { ProjectLocation } from './../locations';
+import { Rating } from './../ratings';
+import { CoBenefit } from './../co-benefits';
+
 class Project 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
+    Project.hasMany(RelatedProject);
+    Project.hasMany(Vintage);
+    Project.hasMany(Qualification);
+    Project.hasMany(Rating);
+    Project.hasMany(CoBenefit);
+    Project.hasMany(ProjectLocation);
   }
 }
 
@@ -46,7 +58,7 @@ Project.init(
     createdAt: Sequelize.DATE,
     updatedAt: Sequelize.DATE,
   },
-  { sequelize, modelName: 'Projects' },
+  { sequelize, modelName: 'Projects', foreignKey: 'projectId' },
 );
 
 export { Project };
diff --git a/src/models/qualifications/qualifications.model.js b/src/models/qualifications/qualifications.model.js
@@ -2,6 +2,8 @@
 import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
+import { Project } from '../projects';
+import { Unit } from '../units';
 
 class Qualification extends Model {
   /**
@@ -10,7 +12,8 @@ class Qualification extends Model {
    * The `models/index` file will call this method automatically.
    */
   static associate(models) {
-    // define association here
+    Qualification.belongsTo(Project);
+    Qualification.hasMany(Unit);
   }
 }
 
@@ -55,7 +58,7 @@ Qualification.init(
       type: Sequelize.DATE,
     },
   },
-  { sequelize, modelName: 'Qualifications' },
+  { sequelize, modelName: 'Qualifications', foreignKey: 'qualificationId' },
 );
 
 export { Qualification };
diff --git a/src/models/ratings/ratings.model.js b/src/models/ratings/ratings.model.js
@@ -2,6 +2,7 @@
 import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
+import { Project } from '../projects/index';
 
 class Rating extends Model {
   /**
@@ -10,7 +11,7 @@ class Rating extends Model {
    * The `models/index` file will call this method automatically.
    */
   static associate(models) {
-    // define association here
+    Rating.belongsTo(Project);
   }
 }
 

diff --git a/src/models/related-projects/related-projects.model.js b/src/models/related-projects/related-projects.model.js
@@ -3,14 +3,16 @@ import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
 
+import { Project } from '../projects';
+
 class RelatedProject 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
+    RelatedProject.belongsTo(Project);
   }
 }
 

diff --git a/src/models/units/units.model.js b/src/models/units/units.model.js
@@ -2,6 +2,7 @@
 import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
+import { Project } from '../projects/index';
 
 class Unit extends Model {
   /**
@@ -10,7 +11,7 @@ class Unit extends Model {
    * The `models/index` file will call this method automatically.
    */
   static associate(models) {
-    // define association here
+    Unit.belongsTo(Project);
   }
 }
 

diff --git a/src/models/vintages/vintages.model.js b/src/models/vintages/vintages.model.js
@@ -2,6 +2,7 @@
 import Sequelize from 'sequelize';
 const { Model } = Sequelize;
 import { sequelize } from '../database';
+import { Project } from '../projects/index';
 
 class Vintage extends Model {
   /**
@@ -10,7 +11,7 @@ class Vintage extends Model {
    * The `models/index` file will call this method automatically.
    */
   static associate(models) {
-    // define association here
+    Vintage.belongsTo(Project);
   }
 }