- Add a default make to the tenant

- Add a make to the SpecialistMagicLink

eveai_app/views/interaction_forms.py

@@ -8,6 +8,7 @@ from wtforms_sqlalchemy.fields import QuerySelectMultipleField
 
 from common.models.document import Retriever
 from common.models.interaction import EveAITool, Specialist
+from common.models.user import TenantMake
 from common.extensions import cache_manager
 from common.utils.form_assistants import validate_json
 
@@ -140,6 +141,7 @@ class SpecialistMagicLinkForm(FlaskForm):
     description = TextAreaField('Description', validators=[Optional()])
     magic_link_code = StringField('Magic Link Code', validators=[DataRequired(), Length(max=55)], render_kw={'readonly': True})
     specialist_id = SelectField('Specialist', validators=[DataRequired()])
+    tenant_make_id = SelectField('Tenant Make', validators=[Optional()], coerce=int)
     valid_from = DateField('Valid From', id='form-control datepicker', validators=[Optional()])
     valid_to = DateField('Valid To', id='form-control datepicker', validators=[Optional()])
 
@@ -150,9 +152,13 @@ class SpecialistMagicLinkForm(FlaskForm):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         specialists = Specialist.query.all()
-        # Dynamically populate the 'type' field using the constructor
+        # Dynamically populate the specialist field
         self.specialist_id.choices = [(specialist.id, specialist.name) for specialist in specialists]
 
+        # Dynamically populate the tenant_make field with None as first option
+        tenant_makes = TenantMake.query.all()
+        self.tenant_make_id.choices = [(0, 'None')] + [(make.id, make.name) for make in tenant_makes]
+
 
 class EditSpecialistMagicLinkForm(DynamicFormBase):
     name = StringField('Name', validators=[DataRequired(), Length(max=50)])
@@ -161,6 +167,8 @@ class EditSpecialistMagicLinkForm(DynamicFormBase):
                                   render_kw={'readonly': True})
     specialist_id = IntegerField('Specialist', validators=[DataRequired()], render_kw={'readonly': True})
     specialist_name = StringField('Specialist Name', validators=[DataRequired()], render_kw={'readonly': True})
+    tenant_make_id = SelectField('Tenant Make', validators=[Optional()], coerce=int)
+    tenant_make_name = StringField('Tenant Make Name', validators=[Optional()], render_kw={'readonly': True})
     valid_from = DateField('Valid From', id='form-control datepicker', validators=[Optional()])
     valid_to = DateField('Valid To', id='form-control datepicker', validators=[Optional()])
 
@@ -176,5 +184,15 @@ class EditSpecialistMagicLinkForm(DynamicFormBase):
         else:
             self.specialist_name.data = ''
 
+        # Dynamically populate the tenant_make field with None as first option
+        tenant_makes = TenantMake.query.all()
+        self.tenant_make_id.choices = [(0, 'None')] + [(make.id, make.name) for make in tenant_makes]
+
+        # If the form has a tenant_make_id that's not zero, set the tenant_make_name
+        if hasattr(self, 'tenant_make_id') and self.tenant_make_id.data and self.tenant_make_id.data > 0:
+            tenant_make = TenantMake.query.get(self.tenant_make_id.data)
+            if tenant_make:
+                self.tenant_make_name.data = tenant_make.name
+