Merge pull request #152 from Carifio24/small-class-flag

Add small class flag
cosmicds · Oct 17, 2024 · 3a7a9c1 · 3a7a9c1
2 parents 72ff314 + 6918bcf
commit 3a7a9c1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/models/class.ts b/src/models/class.ts
@@ -12,6 +12,7 @@ export class Class extends Model<InferAttributes<Class>, InferCreationAttributes
   declare asynchronous: CreationOptional<boolean>;
   declare test: CreationOptional<boolean>;
   declare expected_size: CreationOptional<number>;
+  declare small_class: CreationOptional<boolean>;
 }
 
 export function initializeClassModel(sequelize: Sequelize) {
@@ -67,6 +68,9 @@ export function initializeClassModel(sequelize: Sequelize) {
       type: DataTypes.INTEGER.UNSIGNED,
       allowNull: false,
     },
+    small_class: {
+      type: "tinyint(1) GENERATED ALWAYS AS (expected_size < 10) VIRTUAL",
+    },
   }, {
     sequelize,
   });

diff --git a/src/sql/create_class_table.sql b/src/sql/create_class_table.sql
@@ -9,6 +9,7 @@ CREATE TABLE Classes (
     test tinyint(1) NOT NULL DEFAULT 0,
     updated datetime DEFAULT NULL,
     expected_size int(11) UNSIGNED NOT NULL,
+    small_class tinyint(1) GENERATED ALWAYS AS (expected_size < 10) VIRTUAL,
 
     PRIMARY KEY(id),
     INDEX(educator_id),