Add Template_Categories table

app/dao/template_categories_dao.py

@@ -1,4 +1,5 @@
 import uuid
+from datetime import datetime
 
 from flask import current_app
 
@@ -69,6 +70,7 @@ def dao_delete_template_category_by_id(template_category_id, cascade=False):
                     else template_category.email_process_type
                 )
                 template.category = dao_get_template_category_by_id(default_category_id)
+                template.updated_at = datetime.utcnow
                 db.session.add(template)
 
         db.session.delete(template_category)

app/models.py

@@ -1044,6 +1044,8 @@ class TemplateCategory(BaseModel):
     sms_process_type = db.Column(db.String(200), nullable=False)
     email_process_type = db.Column(db.String(200), nullable=False)
     hidden = db.Column(db.Boolean, nullable=False, default=False)
+    created_at = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
+    updated_at = db.Column(db.DateTime, onupdate=datetime.datetime.utcnow)
 
     def serialize(self):
         return {
@@ -1055,6 +1057,8 @@ def serialize(self):
             "sms_process_type": self.sms_process_type,
             "email_process_type": self.email_process_type,
             "hidden": self.hidden,
+            "created_at": self.created_at,
+            "updated_at": self.updated_at,
         }
 
     @classmethod

migrations/versions/0454_add_template_category.py

@@ -27,23 +27,25 @@ def upgrade():
         sa.Column("sms_process_type", sa.String(length=255), nullable=False),
         sa.Column("email_process_type", sa.String(length=255), nullable=False),
         sa.Column("hidden", sa.Boolean(), nullable=False),
+        sa.Column("created_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=True),
         sa.UniqueConstraint("name_en"),
         sa.UniqueConstraint("name_fr"),
     )
 
     # Insert the generic low, medium, and high categories
     op.execute(
-        "INSERT INTO template_categories (id, name_en, name_fr, sms_process_type, email_process_type, hidden) VALUES ('{}', 'Low Category (Bulk)', 'Catégorie Basse (En Vrac)', 'low', 'low', true)".format(
-            current_app.config["DEFAULT_TEMPLATE_CATEGORY_LOW"]
+        "INSERT INTO template_categories (id, name_en, name_fr, sms_process_type, email_process_type, hidden, created_at) VALUES ('{}', 'Low Category (Bulk)', 'Catégorie Basse (En Vrac)', 'low', 'low', true, now())".format(
+            current_app.config["DEFAULT_TEMPLATE_CATEGORY_LOW"],
         )
     )
     op.execute(
-        "INSERT INTO template_categories (id, name_en, name_fr, sms_process_type, email_process_type, hidden) VALUES ('{}', 'Medium Category (Normal)', 'Catégorie Moyenne (Normale)', 'low', 'low', true)".format(
+        "INSERT INTO template_categories (id, name_en, name_fr, sms_process_type, email_process_type, hidden, created_at) VALUES ('{}', 'Medium Category (Normal)', 'Catégorie Moyenne (Normale)', 'low', 'low', true, now())".format(
             current_app.config["DEFAULT_TEMPLATE_CATEGORY_MEDIUM"]
         )
     )
     op.execute(
-        "INSERT INTO template_categories (id, name_en, name_fr, sms_process_type, email_process_type, hidden) VALUES ('{}', 'High Category (Priority)', 'Catégorie Haute (Priorité)', 'low', 'low', true)".format(
+        "INSERT INTO template_categories (id, name_en, name_fr, sms_process_type, email_process_type, hidden, created_at) VALUES ('{}', 'High Category (Priority)', 'Catégorie Haute (Priorité)', 'low', 'low', true, now())".format(
             current_app.config["DEFAULT_TEMPLATE_CATEGORY_HIGH"]
         )
     )

tests/app/conftest.py

@@ -365,6 +365,8 @@ def create_sample_template(
         data["postage"] = "second"
     if category:
         data["category"] = category
+    else:
+        data.update({"template_category_id": current_app.config["DEFAULT_TEMPLATE_CATEGORY_LOW"]})
     template = Template(**data)
     dao_create_template(template)