Adding language on TemplateType

connect/api/v2/template_projects/views.py

@@ -13,8 +13,7 @@ class TemplateTypeViewSet(ModelViewSet):
     permission_classes = [IsAdminOrReadOnly]
 
     def get_queryset(self):
-
-        queryset = self.queryset
+        queryset = self.queryset.filter(language=self.request.user.language)
         id = self.request.query_params.get('id', None)
         name = self.request.query_params.get('name', None)
         category = self.request.query_params.get('category', None)

connect/template_projects/migrations/0008_templatetype_language.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.23 on 2024-09-12 17:35
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('template_projects', '0007_auto_20230926_1231'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='templatetype',
+            name='language',
+            field=models.CharField(choices=[('en-us', 'English'), ('pt-br', 'Portuguese'), ('es', 'Spanish')], default='pt-br', max_length=10, verbose_name='language'),
+        ),
+    ]

connect/template_projects/models.py

@@ -2,11 +2,17 @@
 
 from django.db import models
 from django.contrib.postgres.fields import ArrayField
+from django.conf import settings
 from .storage import TemplateTypeImageStorage
 
 
 class TemplateType(models.Model):
-
+    language = models.CharField(
+        "language",
+        max_length=10,
+        choices=settings.LANGUAGES,
+        default='pt-br',
+    )
     level_field = [("low", 1), ("medium", 2), ("high", 3)]
     uuid = models.UUIDField(
         "UUID", default=uuid4.uuid4
@@ -40,7 +46,6 @@ def get_template_code(self, template_name):
 
 
 class TemplateFeature(models.Model):
-
     features_types = [
         ("Flows", "Flows"),
         ("Integrations", "Integrations"),