set custom template

app/api/services/office_templates.py

@@ -13,20 +13,24 @@ class OfficeTemplatesService(BaseService):
     def set_office_custom_template(data):
         logger.info('set_office_custom_template')
         try:
-            # office_users = OfficeUsers.get_office_users_list()
+            with db.session.begin():
-            # office_users_data = []
+                subscrtiption_uid = data.get('subscrtiption_uid', '')
-            # if office_users:
+                custom_id = data.get('custom_id', '')
-            #     for t in office_users:
+
-            #         office_users_data.append({
+                membersSubResult = MembersProducts.get_member_product_by_subscription_uid(subscrtiption_uid)
-            #             'id': t.id,
+                if not membersSubResult:
-            #             'uid': t.uid,
+                    return {
-            #             'username': t.username,
+                        "message": "Invalid Subscription",
-            #             'firstname': t.firstname,
+                        "error_message": "Error - Please select a valid product to view",
-            #             'lastname': t.lastname,
+                        "data": None,
-            #             'acc_level': t.acc_level,
+                        "error": f"Subscription with UID {subscrtiption_uid} does not exist."
-            #             'status': t.status,
+                    }, 400
-            #             'added': t.added.isoformat() if t.added else None
+
-            #         })
+                member_id = membersSubResult.member_id
+                logger.info(f'set_office_custom_template {member_id} {subscrtiption_uid} {custom_id}')
+
+                MembersProducts.set_member_product_custom_template(member_id, subscrtiption_uid, custom_id)
+
 
                 office_users_result = {
                     "set_office_custom_template": [],

app/models/members_products.py

@@ -98,6 +98,15 @@ class MembersProducts(db.Model):
         member_product.product_template = product_template
         return product_template
 
+    @classmethod
+    def set_member_product_custom_template(cls, member_id, uid, custom_template):
+        member_product = cls.query.filter_by(member_id=str(member_id), uid=str(uid)).first()
+        if not member_product:
+            return None
+        logger.info(f'ACTION:::set_member_product_custom_template {member_id} {uid} {custom_template}')
+        member_product.custom_template = custom_template
+        return custom_template
+
     @classmethod
     def set_member_product_external_url(cls, member_id, product_id, external_url):
         member_product = cls.query.filter_by(member_id=str(member_id), product_id=str(product_id)).first()